Conatum
21cafac98a
Added the `statreset` command for resetting study statistics. Add the `Guild Admin` command group.
66 lines
1.9 KiB
Python
66 lines
1.9 KiB
Python
from io import StringIO
|
|
|
|
import discord
|
|
from wards import guild_admin
|
|
from data import tables
|
|
from core import Lion
|
|
|
|
from .module import module
|
|
|
|
|
|
@module.cmd("statreset",
|
|
desc="Perform a complete reset of the server's study statistics.",
|
|
group="Guild Admin")
|
|
@guild_admin()
|
|
async def cmd_statreset(ctx):
|
|
"""
|
|
Usage``:
|
|
{prefix}statreset
|
|
Description:
|
|
Perform a complete reset of the server's member statistics.
|
|
This includes tracked time, coins, and workout counts.
|
|
|
|
This may be used to set "seasons" of study.
|
|
|
|
Before the reset, I will send a csv file with the previous member statistics.
|
|
|
|
**This is not reversible.**
|
|
"""
|
|
if not await ctx.ask("Are you sure you want to completely reset the member statistics? "
|
|
"**THIS IS NOT REVERSIBLE!**"):
|
|
return
|
|
# Build the data csv
|
|
rows = tables.lions.select_where(
|
|
select_columns=('userid', 'tracked_time', 'coins', 'workout_count', 'b.roleid AS badge_roleid'),
|
|
_extra=(
|
|
"LEFT JOIN study_badges b ON last_study_badgeid = b.badgeid "
|
|
"WHERE members.guildid={}"
|
|
).format(ctx.guild.id)
|
|
)
|
|
header = "userid, tracked_time, coins, workouts, rank_roleid\n"
|
|
csv_rows = [
|
|
', '.join(str(data) for data in row)
|
|
for row in rows
|
|
]
|
|
|
|
with StringIO() as stats_file:
|
|
stats_file.write(header)
|
|
stats_file.write('\n'.join(csv_rows))
|
|
stats_file.seek(0)
|
|
|
|
out_file = discord.File(stats_file, filename="member_statistics.csv")
|
|
await ctx.reply(file=out_file)
|
|
|
|
# Reset the statistics
|
|
tables.lions.update_where(
|
|
{'tracked_time': 0, 'coins': 0, 'workout_count': 0},
|
|
guildid=ctx.guild.id
|
|
)
|
|
|
|
Lion.sync()
|
|
|
|
await ctx.embed_reply(
|
|
"The server member statistics have been reset!\n"
|
|
"(It may take a while for the member studybadges to update.)"
|
|
)
|