Other => Presence

This commit is contained in:
klemek
2021-01-11 16:35:33 +01:00
parent bfe279095d
commit ae9cc69ed2
6 changed files with 40 additions and 37 deletions
+1 -1
View File
@@ -1,4 +1,4 @@
from .emote import Emote, get_emote_dict from .emote import Emote, get_emote_dict
from .frequency import Frequency from .frequency import Frequency
from .composition import Composition from .composition import Composition
from .other import Other from .presence import Presence
@@ -5,7 +5,7 @@ from collections import defaultdict
from utils import mention, channel_mention, plural from utils import mention, channel_mention, plural
class Other: class Presence:
def __init__(self): def __init__(self):
self.most_used_reaction = "" self.most_used_reaction = ""
self.most_used_reaction_count = 0 self.most_used_reaction_count = 0
+5 -5
View File
@@ -7,7 +7,7 @@ from scanners import (
FullScanner, FullScanner,
FrequencyScanner, FrequencyScanner,
CompositionScanner, CompositionScanner,
OtherScanner, PresenceScanner,
) )
logging.basicConfig( logging.basicConfig(
@@ -26,10 +26,10 @@ bot.log_calls = True
bot.client.bot = bot # TODO place in miniscord bot.client.bot = bot # TODO place in miniscord
bot.register_command( bot.register_command(
"other", "pres(ence)?",
lambda *args: OtherScanner().compute(*args), lambda *args: PresenceScanner().compute(*args),
"other: other data analysis", "pres: presence analysis",
OtherScanner.help(), PresenceScanner.help(),
) )
bot.register_command( bot.register_command(
"emotes", "emotes",
+1 -1
View File
@@ -1,5 +1,5 @@
from .emotes_scanner import EmotesScanner from .emotes_scanner import EmotesScanner
from .frequency_scanner import FrequencyScanner from .frequency_scanner import FrequencyScanner
from .composition_scanner import CompositionScanner from .composition_scanner import CompositionScanner
from .other_scanner import OtherScanner from .presence_scanner import PresenceScanner
from .full_scanner import FullScanner from .full_scanner import FullScanner
+7 -7
View File
@@ -6,8 +6,8 @@ import discord
# Custom libs # Custom libs
from .scanner import Scanner from .scanner import Scanner
from . import FrequencyScanner, CompositionScanner, OtherScanner, EmotesScanner from . import FrequencyScanner, CompositionScanner, PresenceScanner, EmotesScanner
from data_types import Frequency, Composition, Other, Emote, get_emote_dict from data_types import Frequency, Composition, Presence, Emote, get_emote_dict
from logs import ChannelLogs, MessageLog from logs import ChannelLogs, MessageLog
@@ -32,7 +32,7 @@ class FullScanner(Scanner):
guild = message.channel.guild guild = message.channel.guild
self.freq = Frequency() self.freq = Frequency()
self.comp = Composition() self.comp = Composition()
self.other = Other() self.pres = Presence()
# Create emotes dict from custom emojis of the guild # Create emotes dict from custom emojis of the guild
# Create emotes dict from custom emojis of the guild # Create emotes dict from custom emojis of the guild
self.emotes = get_emote_dict(message.channel.guild) self.emotes = get_emote_dict(message.channel.guild)
@@ -41,7 +41,7 @@ class FullScanner(Scanner):
def compute_message(self, channel: ChannelLogs, message: MessageLog): def compute_message(self, channel: ChannelLogs, message: MessageLog):
FrequencyScanner.analyse_message(message, self.freq, self.raw_members) FrequencyScanner.analyse_message(message, self.freq, self.raw_members)
CompositionScanner.analyse_message(message, self.comp, self.raw_members) CompositionScanner.analyse_message(message, self.comp, self.raw_members)
OtherScanner.analyse_message(channel, message, self.other, self.raw_members) PresenceScanner.analyse_message(channel, message, self.pres, self.raw_members)
EmotesScanner.analyse_message( EmotesScanner.analyse_message(
message, self.emotes, self.raw_members, all_emojis=True message, self.emotes, self.raw_members, all_emojis=True
) )
@@ -52,14 +52,14 @@ class FullScanner(Scanner):
def get_results(self, intro: str) -> List[str]: def get_results(self, intro: str) -> List[str]:
FrequencyScanner.compute_results(self.freq) FrequencyScanner.compute_results(self.freq)
CompositionScanner.compute_results(self.comp, self.emotes) CompositionScanner.compute_results(self.comp, self.emotes)
OtherScanner.compute_results(self.other, self.emotes) PresenceScanner.compute_results(self.pres, self.emotes)
res = [intro] res = [intro]
res += ["__Frequency__:"] res += ["__Frequency__:"]
res += self.freq.to_string() res += self.freq.to_string()
res += ["__Composition__:"] res += ["__Composition__:"]
res += self.comp.to_string() res += self.comp.to_string()
res += ["__Other__:"] res += ["__Presence__:"]
res += self.other.to_string( res += self.pres.to_string(
show_top_channel=len(self.channels) > 1, show_top_channel=len(self.channels) > 1,
show_mentioned=(len(self.members) > 0), show_mentioned=(len(self.members) > 0),
) )
@@ -6,29 +6,29 @@ import discord
from .scanner import Scanner from .scanner import Scanner
from . import EmotesScanner from . import EmotesScanner
from data_types import Other, Emote, get_emote_dict from data_types import Presence, Emote, get_emote_dict
from logs import ChannelLogs, MessageLog from logs import ChannelLogs, MessageLog
class OtherScanner(Scanner): class PresenceScanner(Scanner):
@staticmethod @staticmethod
def help() -> str: def help() -> str:
return "```\n" return "```\n"
+"%other : Show other statistics\n" +"%pres : Show presence statistics\n"
+"arguments:\n" +"arguments:\n"
+"* @member/me : filter for one or more member\n" +"* @member/me : filter for one or more member\n"
+"* #channel/here : filter for one or more channel\n" +"* #channel/here : filter for one or more channel\n"
+"Example: %other #mychannel1 @user\n" +"Example: %pres #mychannel1 @user\n"
+"```" +"```"
def __init__(self): def __init__(self):
super().__init__( super().__init__(
help=OtherScanner.help(), help=PresenceScanner.help(),
intro_context="Other data", intro_context="Presence",
) )
async def init(self, message: discord.Message, *args: str) -> bool: async def init(self, message: discord.Message, *args: str) -> bool:
self.other = Other() self.pres = Presence()
# Create emotes dict from custom emojis of the guild # Create emotes dict from custom emojis of the guild
self.emotes = get_emote_dict(message.channel.guild) self.emotes = get_emote_dict(message.channel.guild)
return True return True
@@ -37,40 +37,43 @@ class OtherScanner(Scanner):
EmotesScanner.analyse_message( EmotesScanner.analyse_message(
message, self.emotes, self.raw_members, all_emojis=True message, self.emotes, self.raw_members, all_emojis=True
) )
return OtherScanner.analyse_message( return PresenceScanner.analyse_message(
channel, message, self.other, self.raw_members channel, message, self.pres, self.raw_members
) )
def get_results(self, intro: str) -> List[str]: def get_results(self, intro: str) -> List[str]:
OtherScanner.compute_results(self.other, self.emotes) PresenceScanner.compute_results(self.pres, self.emotes)
res = [intro] res = [intro]
res += self.other.to_string( res += self.pres.to_string(
show_top_channel=len(self.channels) > 1, show_top_channel=(len(self.channels) > 1),
show_mentioned=(len(self.members) > 0), show_mentioned=(len(self.members) > 0),
) )
return res return res
@staticmethod @staticmethod
def analyse_message( def analyse_message(
channel: ChannelLogs, message: MessageLog, other: Other, raw_members: List[int] channel: ChannelLogs,
message: MessageLog,
pres: Presence,
raw_members: List[int],
) -> bool: ) -> bool:
impacted = False impacted = False
# If author is included in the selection (empty list is all) # If author is included in the selection (empty list is all)
if not message.bot and (len(raw_members) == 0 or message.author in raw_members): if not message.bot and (len(raw_members) == 0 or message.author in raw_members):
impacted = True impacted = True
other.channel_usage[channel.id] += 1 pres.channel_usage[channel.id] += 1
for mention in message.mentions: for mention in message.mentions:
if mention in raw_members: if mention in raw_members:
other.mentions[mention] += 1 pres.mentions[mention] += 1
return impacted return impacted
@staticmethod @staticmethod
def compute_results(other: Other, emotes: Dict[str, Emote]): def compute_results(pres: Presence, emotes: Dict[str, Emote]):
# calculate total reactions # calculate total reactions
other.used_reaction_total = sum([emote.reactions for emote in emotes.values()]) pres.used_reaction_total = sum([emote.reactions for emote in emotes.values()])
if other.used_reaction_total > 0: if pres.used_reaction_total > 0:
# calculate most used reaction # calculate most used reaction
other.most_used_reaction = sorted( pres.most_used_reaction = sorted(emotes, key=lambda k: emotes[k].reactions)[
emotes, key=lambda k: emotes[k].reactions -1
)[-1] ]
other.most_used_reaction_count = emotes[other.most_used_reaction].reactions pres.most_used_reaction_count = emotes[pres.most_used_reaction].reactions