Croc/croccybot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

rewrite: Move GUI cards into main repo.

Browse Source
This commit is contained in:
Conatum
2022-12-23 07:02:26 +02:00
parent f328324747
commit 1cef97cdd4
20 changed files with 6110 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

278
src/modules/pending-rewrite/gui-commands/stats.py Normal file
View File

@@ -0,0 +1,278 @@
import asyncio
import time
from datetime import datetime, timedelta
from cmdClient.checks import in_guild
from utils.lib import utc_now
from data import tables
from data.conditions import LEQ
from core import Lion
from LionContext import LionContext as Context
from modules.study.tracking.data import session_history
from modules.stats.achievements import get_achievements_for
from ...cards import StatsCard, ProfileCard
from ...utils import get_avatar_key, image_as_file
from ..module import module, ratelimit
async def get_stats_card_for(ctx: Context, target):
lion = Lion.fetch(ctx.guild.id, target.id)
history = session_history.select_where(
guildid=ctx.guild.id,
userid=target.id,
select_columns=(
"start_time",
"(start_time + duration * interval '1 second') AS end_time"
),
_extra="ORDER BY start_time DESC"
)
# Current economy balance (accounting for current session)
workout_total = lion.data.workout_count
# Leaderboard ranks
exclude = set(m.id for m in ctx.guild_settings.unranked_roles.members)
exclude.update(ctx.client.user_blacklist())
exclude.update(ctx.client.objects['ignored_members'][ctx.guild.id])
if target.bot or target.id in exclude:
time_rank = None
coin_rank = None
else:
time_rank, coin_rank = tables.lions.queries.get_member_rank(ctx.guild.id, target.id, list(exclude or [0]))
# Study time
# First get the all/month/week/day timestamps
day_start = lion.day_start
month_start = day_start.replace(day=1)
period_timestamps = (
datetime(1970, 1, 1),
month_start,
day_start - timedelta(days=day_start.weekday()),
day_start
)
study_times = [0, 0, 0, 0]
for i, timestamp in enumerate(period_timestamps):
study_time = tables.session_history.queries.study_time_since(ctx.guild.id, target.id, timestamp)
if not study_time:
# So we don't make unecessary database calls
break
study_times[i] = study_time
# Streak data for the study run view
streaks = []
streak = 0
streak_end = None
date = day_start
daydiff = timedelta(days=1)
if 'sessions' in ctx.client.objects and lion.session:
day_attended = True
streak_end = day_start.day
else:
day_attended = None
periods = [(row['start_time'], row['end_time']) for row in history]
i = 0
while i < len(periods):
row = periods[i]
i += 1
if row[1] > date:
# They attended this day
day_attended = True
if streak_end is None:
streak_end = (date - month_start).days + 1
continue
elif day_attended is None:
# Didn't attend today, but don't break streak
day_attended = False
date -= daydiff
i -= 1
continue
elif not day_attended:
# Didn't attend the day, streak broken
date -= daydiff
i -= 1
pass
else:
# Attended the day
streak += 1
# Move window to the previous day and try the row again
day_attended = False
prev_date = date
date -= daydiff
i -= 1
# Special case, when the last session started in the previous day
# Then the day is already attended
if i > 1 and date < periods[i-2][0] <= prev_date:
day_attended = True
if streak_end is None:
streak_end = (date - month_start).days + 1
continue
if streak_end:
streaks.append((streak_end - streak + 1, streak_end))
streak_end = None
streak = 0
if date.month != day_start.month:
break
# Handle loop exit state, i.e. the last streak
if day_attended:
streak += 1
streaks.append((streak_end - streak + 1, streak_end))
# We have all the data for the stats card
return await StatsCard.request(
(time_rank, coin_rank),
list(reversed(study_times)),
workout_total,
streaks,
skin=StatsCard.skin_args_for(ctx)
)
async def get_profile_card_for(ctx: Context, target):
lion = Lion.fetch(ctx.guild.id, target.id)
# Current economy balance (accounting for current session)
coins = lion.coins
season_time = lion.time
# Study League
guild_badges = tables.study_badges.fetch_rows_where(guildid=ctx.guild.id)
if lion.data.last_study_badgeid:
current_badge = tables.study_badges.fetch(lion.data.last_study_badgeid)
else:
current_badge = None
next_badge = min(
(badge for badge in guild_badges
if badge.required_time > (current_badge.required_time if current_badge else 0)),
key=lambda badge: badge.required_time,
default=None
)
if current_badge:
current_rank = (
role.name if (role := ctx.guild.get_role(current_badge.roleid)) else str(current_badge.roleid),
current_badge.required_time // 3600,
next_badge.required_time // 3600 if next_badge else None
)
else:
current_rank = None
if next_badge:
next_next_badge = min(
(badge for badge in guild_badges if badge.required_time > next_badge.required_time),
key=lambda badge: badge.required_time,
default=None
)
next_rank = (
role.name if (role := ctx.guild.get_role(next_badge.roleid)) else str(next_badge.roleid),
next_badge.required_time // 3600,
next_next_badge.required_time // 3600 if next_next_badge else None
)
else:
next_rank = None
achievements = await get_achievements_for(target)
# We have all the data for the profile card
avatar = get_avatar_key(ctx.client, target.id)
return await ProfileCard.request(
target.name,
'#{}'.format(target.discriminator),
coins,
season_time,
avatar=avatar,
gems=ctx.client.data.gem_transactions.queries.get_gems_for(target.id),
gifts=ctx.client.data.gem_transactions.queries.get_gifts_for(target.id),
achievements=[i for i, ach in enumerate(achievements) if ach.level_id > 0],
current_rank=current_rank,
next_rank=next_rank,
badges=lion.profile_tags,
skin=ProfileCard.skin_args_for(ctx)
)
@module.cmd(
"stats",
group="Statistics",
desc="View your server study statistics!"
)
@in_guild()
@ratelimit.ward(member=False)
async def cmd_stats(ctx):
"""
Usage``:
{prefix}stats
{prefix}stats <mention>
Description:
View your study statistics in this server, or those of the mentioned member.
"""
# Identify the target
if ctx.args:
if not ctx.msg.mentions:
return await ctx.error_reply("Please mention a user to view their statistics!")
target = ctx.msg.mentions[0]
else:
target = ctx.author
# System sync
Lion.sync()
# Fetch the cards
futures = (
asyncio.create_task(get_profile_card_for(ctx, target)),
asyncio.create_task(get_stats_card_for(ctx, target))
)
await futures[0]
await futures[1]
profile_image = futures[0].result()
stats_image = futures[1].result()
profile_file = image_as_file(profile_image, f"profile_{target.id}.png")
stats_file = image_as_file(stats_image, f"stats_{target.id}.png")
await ctx.reply(files=[profile_file, stats_file])
@module.cmd(
"profile",
group="Statistics",
desc="View your personal study profile!"
)
@in_guild()
@ratelimit.ward(member=False)
async def cmd_profile(ctx):
"""
Usage``:
{prefix}profile
{prefix}profile <mention>
Description:
View your server study profile, or that of the mentioned user.
"""
# Identify the target
if ctx.args:
if not ctx.msg.mentions:
return await ctx.error_reply("Please mention a user to view their profile!")
target = ctx.msg.mentions[0]
else:
target = ctx.author
# System sync
Lion.sync()
# Fetch the cards
profile_image = await get_profile_card_for(ctx, target)
profile_file = image_as_file(profile_image, f"profile_{target.id}.png")
await ctx.reply(file=profile_file)