Klemek
97 lines
3.1 KiB
Python
97 lines
3.1 KiB
Python
from typing import List
|
|
import discord
|
|
|
|
|
|
# Custom libs
|
|
|
|
from .scanner import Scanner
|
|
from data_types import Presence
|
|
from logs import ChannelLogs, MessageLog
|
|
from utils import COMMON_HELP_ARGS
|
|
|
|
|
|
class PresenceScanner(Scanner):
|
|
@staticmethod
|
|
def help() -> str:
|
|
return (
|
|
"```\n"
|
|
+ "%pres: Show presence statistics\n"
|
|
+ "arguments:\n"
|
|
+ COMMON_HELP_ARGS
|
|
+ "* all/everyone - include bots\n"
|
|
+ "Example: %pres #mychannel1 @user\n"
|
|
+ "```"
|
|
)
|
|
|
|
def __init__(self):
|
|
super().__init__(
|
|
valid_args=["all", "everyone"],
|
|
help=PresenceScanner.help(),
|
|
intro_context="Presence",
|
|
)
|
|
|
|
async def init(self, message: discord.Message, *args: str) -> bool:
|
|
self.pres = Presence()
|
|
self.member_specific = len(self.members) > 0
|
|
self.all_messages = "all" in args or "everyone" in args
|
|
return True
|
|
|
|
def compute_message(self, channel: ChannelLogs, message: MessageLog):
|
|
return PresenceScanner.analyse_message(
|
|
channel,
|
|
message,
|
|
self.pres,
|
|
self.raw_members,
|
|
all_messages=self.all_messages,
|
|
)
|
|
|
|
def get_results(self, intro: str) -> List[str]:
|
|
res = [intro]
|
|
res += self.pres.to_string(
|
|
self.msg_count,
|
|
self.total_msg,
|
|
chan_count=len(self.channels) if not self.full else None,
|
|
show_top_channel=(len(self.channels) > 1),
|
|
member_specific=self.member_specific,
|
|
)
|
|
return res
|
|
|
|
@staticmethod
|
|
def analyse_message(
|
|
channel: ChannelLogs,
|
|
message: MessageLog,
|
|
pres: Presence,
|
|
raw_members: List[int],
|
|
*,
|
|
all_messages: bool,
|
|
) -> bool:
|
|
impacted = False
|
|
# If author is included in the selection (empty list is all)
|
|
if (
|
|
(not message.bot or all_messages)
|
|
and len(raw_members) == 0
|
|
or message.author in raw_members
|
|
):
|
|
impacted = True
|
|
pres.channel_usage[channel.id] += 1
|
|
for mention in message.mentions:
|
|
pres.mention_others[mention] += 1
|
|
pres.messages[message.author] += 1
|
|
pres.channel_total[channel.id] += 1
|
|
pres.mention_count[message.author] += len(message.mentions)
|
|
if len(raw_members) > 0:
|
|
for mention in message.mentions:
|
|
if mention in raw_members:
|
|
pres.mentions[message.author] += 1
|
|
for reaction in message.reactions:
|
|
for member_id in message.reactions[reaction]:
|
|
pres.used_reaction[member_id] += 1
|
|
if member_id in raw_members:
|
|
pres.reactions[reaction] += 1
|
|
else:
|
|
for reaction in message.reactions:
|
|
pres.reactions[reaction] += len(message.reactions[reaction])
|
|
for member_id in message.reactions[reaction]:
|
|
pres.used_reaction[member_id] += 1
|
|
return impacted
|