new scanner %chan
This commit is contained in:
@@ -8,7 +8,7 @@ import discord
|
||||
from logs import ChannelLogs, MessageLog
|
||||
from .scanner import Scanner
|
||||
from data_types import Counter
|
||||
from utils import COMMON_HELP_ARGS, plural, precise, mention, alt_mention
|
||||
from utils import COMMON_HELP_ARGS, mention, channel_mention
|
||||
|
||||
|
||||
class MessagesScanner(Scanner):
|
||||
@@ -16,7 +16,7 @@ class MessagesScanner(Scanner):
|
||||
def help() -> str:
|
||||
return (
|
||||
"```\n"
|
||||
+ "%msg: Rank users mentions by their messages\n"
|
||||
+ "%msg: Rank users by their messages\n"
|
||||
+ "arguments:\n"
|
||||
+ COMMON_HELP_ARGS
|
||||
+ "* <n> - top <n>, default is 10\n"
|
||||
@@ -39,21 +39,25 @@ class MessagesScanner(Scanner):
|
||||
for arg in args:
|
||||
if arg.isdigit():
|
||||
self.top = int(arg)
|
||||
self.all_mentions = "all" in args
|
||||
self.all_messages = "all" in args
|
||||
# Create mentions dict
|
||||
self.messages = defaultdict(Counter)
|
||||
return True
|
||||
|
||||
def compute_message(self, channel: ChannelLogs, message: MessageLog):
|
||||
return MessagesScanner.analyse_message(
|
||||
message, self.messages, self.raw_members, all_mentions=self.all_mentions
|
||||
channel.id,
|
||||
message,
|
||||
self.messages,
|
||||
self.raw_members,
|
||||
all_messages=self.all_messages,
|
||||
)
|
||||
|
||||
def get_results(self, intro: str) -> List[str]:
|
||||
names = [name for name in self.messages]
|
||||
names.sort(key=lambda name: self.messages[name].score(), reverse=True)
|
||||
names = names[: self.top]
|
||||
usage_count = sum([message.usages for message in self.messages.values()])
|
||||
usage_count = Counter.total(self.messages)
|
||||
res = [intro]
|
||||
res += [
|
||||
self.messages[name].to_string(
|
||||
@@ -61,6 +65,7 @@ class MessagesScanner(Scanner):
|
||||
mention(name),
|
||||
total_usage=usage_count,
|
||||
counted="message",
|
||||
transform=lambda id: f" in {channel_mention(id)}",
|
||||
)
|
||||
for name in names
|
||||
]
|
||||
@@ -68,14 +73,15 @@ class MessagesScanner(Scanner):
|
||||
|
||||
@staticmethod
|
||||
def analyse_message(
|
||||
channel_id: int,
|
||||
message: MessageLog,
|
||||
messages: Dict[str, Counter],
|
||||
raw_members: List[int],
|
||||
*,
|
||||
all_mentions: bool,
|
||||
all_messages: bool,
|
||||
) -> bool:
|
||||
impacted = False
|
||||
if not message.bot or all_mentions:
|
||||
if not message.bot or all_messages:
|
||||
impacted = True
|
||||
messages[message.author].update_use(1, message.created_at)
|
||||
messages[message.author].update_use(1, message.created_at, channel_id)
|
||||
return impacted
|
||||
|
||||
Reference in New Issue
Block a user