klemek
83 lines
2.4 KiB
Python
83 lines
2.4 KiB
Python
from typing import Dict, List
|
|
from collections import defaultdict
|
|
import discord
|
|
|
|
|
|
# Custom libs
|
|
|
|
from logs import ChannelLogs, MessageLog
|
|
from .scanner import Scanner
|
|
from data_types import Counter
|
|
from utils import COMMON_HELP_ARGS, mention, channel_mention
|
|
|
|
|
|
class ReactionsScanner(Scanner):
|
|
@staticmethod
|
|
def help() -> str:
|
|
return (
|
|
"```\n"
|
|
+ "%react: Rank users by their reactions\n"
|
|
+ "arguments:\n"
|
|
+ COMMON_HELP_ARGS
|
|
+ "* <n> - top <n>, default is 10\n"
|
|
+ "Example: %react 10 #channel\n"
|
|
+ "```"
|
|
)
|
|
|
|
def __init__(self):
|
|
super().__init__(
|
|
has_digit_args=True,
|
|
help=ReactionsScanner.help(),
|
|
intro_context="Reactions",
|
|
)
|
|
|
|
async def init(self, message: discord.Message, *args: str) -> bool:
|
|
# get max emotes to view
|
|
self.top = 10
|
|
for arg in args:
|
|
if arg.isdigit():
|
|
self.top = int(arg)
|
|
# Create mentions dict
|
|
self.messages = defaultdict(Counter)
|
|
return True
|
|
|
|
def compute_message(self, channel: ChannelLogs, message: MessageLog):
|
|
return ReactionsScanner.analyse_message(
|
|
channel.id,
|
|
message,
|
|
self.messages,
|
|
self.raw_members,
|
|
)
|
|
|
|
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 = Counter.total(self.messages)
|
|
res = [intro]
|
|
res += [
|
|
self.messages[name].to_string(
|
|
names.index(name),
|
|
mention(name),
|
|
total_usage=usage_count,
|
|
counted="reaction",
|
|
transform=lambda id: f" in {channel_mention(id)}",
|
|
)
|
|
for name in names
|
|
]
|
|
return res
|
|
|
|
@staticmethod
|
|
def analyse_message(
|
|
channel_id: int,
|
|
message: MessageLog,
|
|
messages: Dict[str, Counter],
|
|
raw_members: List[int],
|
|
) -> bool:
|
|
impacted = True
|
|
for reaction in message.reactions:
|
|
for member_id in message.reactions[reaction]:
|
|
if len(raw_members) == 0 or member_id in raw_members:
|
|
messages[member_id].update_use(1, message.created_at, channel_id)
|
|
return impacted
|