emojis sorting

This commit is contained in:
klemek
2021-01-13 15:12:28 +01:00
parent ecf6cfe821
commit e106ed4e41
2 changed files with 57 additions and 23 deletions
+27 -12
View File
@@ -5,7 +5,7 @@ import discord
# Custom libs
from utils import mention, plural, from_now, top_key
from utils import mention, plural, from_now, top_key, percent
class Emote:
@@ -32,12 +32,16 @@ class Emote:
def used(self) -> bool:
return self.usages > 0 or self.reactions > 0
def score(self) -> float:
def score(self, *, usage_weight: int = 1, react_weight: int = 1) -> float:
# Score is compose of usages + reactions
# When 2 emotes have the same score,
# the days since last use is stored in the digits
# (more recent first)
return self.usages + self.reactions + 1 / (100000 * (self.use_days() + 1))
return (
self.usages * usage_weight
+ self.reactions * react_weight
+ 1 / (100000 * (self.use_days() + 1))
)
def life_days(self) -> int:
return (datetime.today() - self.emoji.created_at).days
@@ -52,7 +56,16 @@ class Emote:
def get_top_member(self) -> int:
return top_key(self.members)
def to_string(self, i: int, name: str, show_life: bool, show_members: bool) -> str:
def to_string(
self,
i: int,
name: str,
*,
total_usage: int,
total_react: int,
show_life: bool,
show_members: bool,
) -> str:
# place
output = ""
if i == 0:
@@ -65,17 +78,19 @@ class Emote:
output += f"**#{i + 1}**"
output += f" {name} - "
if not self.used():
output += "never used "
output += "never used"
else:
output += f"{plural(self.usages, 'time')} "
if self.reactions >= 1:
output += f"and {plural(self.reactions, 'reaction')} "
if show_life and not self.default:
output += f"(in {plural(self.life_days(), 'day')}) "
if self.used():
output += f"(last used {from_now(self.last_used)})"
if self.usages > 0:
output += f"{plural(self.usages, 'time')} ({percent(self.usages/total_usage)})"
if self.usages > 0 and self.reactions > 0:
output += " and "
if self.reactions >= 1:
output += f"{plural(self.reactions, 'reaction')} ({percent(self.reactions/total_react)})"
output += f" (last used {from_now(self.last_used)})"
if show_members:
output += f" (mostly by {mention(self.get_top_member())}: {self.members[self.get_top_member()]})"
if show_life and not self.default:
output += f" (in {plural(self.life_days(), 'day')})"
return output
+30 -11
View File
@@ -22,6 +22,7 @@ class EmotesScanner(Scanner):
+ "* <n> : top <n> emojis, default is 20\n"
+ "* all : list all common emojis in addition to this guild's\n"
+ "* members : show top member for each emojis\n"
+ "* sort:usage/reaction : other sorting methods\n"
+ "Example: %emojis 10 all #mychannel1 #mychannel2 @user\n"
+ "```"
)
@@ -29,7 +30,7 @@ class EmotesScanner(Scanner):
def __init__(self):
super().__init__(
has_digit_args=True,
valid_args=["all", "members"],
valid_args=["all", "members", "sort:usage", "sort:reaction"],
help=EmotesScanner.help(),
intro_context="Emoji usage",
)
@@ -51,6 +52,11 @@ class EmotesScanner(Scanner):
)
# Create emotes dict from custom emojis of the guild
self.emotes = get_emote_dict(guild)
self.sort = None
if "sort:usage" in args:
self.sort = "usage"
elif "sort:reaction" in args:
self.sort = "reaction"
return True
def compute_message(self, channel: ChannelLogs, message: MessageLog):
@@ -60,23 +66,36 @@ class EmotesScanner(Scanner):
def get_results(self, intro: str) -> List[str]:
names = [name for name in self.emotes]
names.sort(key=lambda name: self.emotes[name].score(), reverse=True)
names.sort(
key=lambda name: self.emotes[name].score(
usage_weight=(0 if self.sort == "reaction" else 1),
react_weight=(0 if self.sort == "usage" else 1),
),
reverse=True,
)
names = names[: self.top]
res = [intro]
allow_unused = self.full and len(self.members) == 0
res += [
self.emotes[name].to_string(
names.index(name), name, False, self.show_members
)
for name in names
if allow_unused or self.emotes[name].used()
]
# Get the total of all emotes used
usage_count = 0
reaction_count = 0
for name in self.emotes:
usage_count += self.emotes[name].usages
reaction_count += self.emotes[name].reactions
res = [intro]
allow_unused = self.full and len(self.members) == 0
if self.sort is not None:
res += [f"(Sorted by {self.sort})"]
res += [
self.emotes[name].to_string(
names.index(name),
name,
total_usage=usage_count,
total_react=reaction_count,
show_life=False,
show_members=self.show_members,
)
for name in names
if allow_unused or self.emotes[name].used()
]
res += [
f"Total: {plural(usage_count,'time')} ({precise(usage_count/self.msg_count)}/msg)"
]