nex/college-bot-v2
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Activity Actions

Make truths_counter command more accurate

Browse source
This commit is contained in:
Nexus 2024-04-15 18:32:17 +01:00
parent 2130c993b0
commit 5829faba46
Signed by: nex
GPG key ID: 0FA334385D0B689F
1 changed files with 17 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
src/cogs/quote_quota.py
View file

@ -186,8 +186,19 @@ class QuoteQuota(commands.Cog):
@commands.slash_command(name="truths")
async def truths_counter(self, ctx: discord.ApplicationContext):
"""Counts the number of times the word 'truth' has been said in the quotes channel."""
def is_truth(msg: discord.Message) -> bool:
if msg.author.id == 1101439218334576742:
if msg.created_at.timestamp() <= 1713202258.991351:
# Pre 6:30pm 15th April, embeds were not tagged for detection.
return any((_te.type == "rich" for _te in msg.embeds))
for __t_e in msg.embeds:
if __t_e.type == "rich" and __t_e.colour.value == 0x5448EE:
return True
return False
channel = discord.utils.get(ctx.guild.text_channels, name="spam")
BOT_ID = 1101439218334576742
if not channel:
return await ctx.respond(":x: Cannot find spam channel.")
@ -208,9 +219,7 @@ class QuoteQuota(commands.Cog):
count_all_time = 0
async for message in channel.history(limit=None, after=now - timedelta(hours=1), before=now):
if message.author.id == BOT_ID:
if not any([x.type == "rich" for x in message.embeds]):
continue
if is_truth(message):
count_hour += 1
per_minute = count_hour / 60
@ -218,20 +227,16 @@ class QuoteQuota(commands.Cog):
embed.description = "counting truths for the last 24 hours"
await ctx.edit(embed=embed)
async for message in channel.history(limit=None, after=now - timedelta(hours=1), before=now):
if message.author.id == BOT_ID:
if not any([x.type == "rich" for x in message.embeds]):
continue
count_hour += 1
async for message in channel.history(limit=None, after=now - timedelta(days=1), before=now):
if is_truth(message):
count_day += 1
per_hour = count_day / 24
embed.add_field(name="Last 24 hours", value=f"{count_day:,} ({per_hour:,.1f}/hour)", inline=False)
embed.description = "counting truths for the last 7 days"
await ctx.edit(embed=embed)
async for message in channel.history(limit=None, after=now - timedelta(days=7), before=now):
if message.author.id == BOT_ID:
if not any([x.type == "rich" for x in message.embeds]):
continue
if is_truth(message):
count_week += 1
per_day = count_week / 7
embed.add_field(name="Last 7 days", value=f"{count_week:,} ({per_day:,.1f}/day)", inline=False)