format etc
This commit is contained in:
149
garfpy/aod.py
149
garfpy/aod.py
@ -11,6 +11,7 @@ from collections import defaultdict
|
||||
meows_file = "meow_counts.json"
|
||||
stats_file = "user_stats.json"
|
||||
|
||||
|
||||
def json_load(file_path, default):
|
||||
if os.path.isfile(file_path):
|
||||
with open(file_path, "r") as f:
|
||||
@ -18,77 +19,105 @@ def json_load(file_path, default):
|
||||
else:
|
||||
return default
|
||||
|
||||
|
||||
meow_counts = defaultdict(int, json_load(meows_file, {}))
|
||||
user_stats = json_load(stats_file, {})
|
||||
|
||||
|
||||
async def aod_message(garfbot, message):
|
||||
if "meow" in message.content.lower():
|
||||
logger.info(f"Meow detected! {message.author.name} said: {message.content}")
|
||||
if "meow" in message.content.lower():
|
||||
logger.info(f"Meow detected! {message.author.name} said: {message.content}")
|
||||
|
||||
meow_counts[str(message.author.id)] += 1
|
||||
meow_counts[str(message.author.id)] += 1
|
||||
|
||||
with open(meows_file, "w") as f:
|
||||
json.dump(dict(meow_counts), f)
|
||||
with open(meows_file, "w") as f:
|
||||
json.dump(dict(meow_counts), f)
|
||||
|
||||
if message.content.lower() == "meowcount":
|
||||
response = f"My records show that <@{message.author.id}> has meowed {meow_counts[str(message.author.id)]} time(s). Have a nice day."
|
||||
await message.channel.send(response)
|
||||
if message.content.lower() == "meowcount":
|
||||
response = f"My records show that <@{message.author.id}> has meowed {meow_counts[str(message.author.id)]} time(s). Have a nice day."
|
||||
await message.channel.send(response)
|
||||
|
||||
if message.content.lower() == "top meowers":
|
||||
top_meowers = sorted(meow_counts.items(), key=itemgetter(1), reverse=True)[:10]
|
||||
embed = discord.Embed(title="Top Meowers :cat:", color=0x000000)
|
||||
for i, (user_id, meow_count) in enumerate(top_meowers):
|
||||
user = await garfbot.fetch_user(int(user_id))
|
||||
embed.add_field(name=f"{i+1}. {user.name}", value=f"{meow_count} meows", inline=False)
|
||||
await message.channel.send(embed=embed)
|
||||
if message.content.lower() == "top meowers":
|
||||
top_meowers = sorted(meow_counts.items(), key=itemgetter(1), reverse=True)[
|
||||
:10
|
||||
]
|
||||
embed = discord.Embed(title="Top Meowers :cat:", color=0x000000)
|
||||
for i, (user_id, meow_count) in enumerate(top_meowers):
|
||||
user = await garfbot.fetch_user(int(user_id))
|
||||
embed.add_field(
|
||||
name=f"{i + 1}. {user.name}",
|
||||
value=f"{meow_count} meows",
|
||||
inline=False,
|
||||
)
|
||||
await message.channel.send(embed=embed)
|
||||
|
||||
if message.content.lower() == "checking in":
|
||||
user_id = str(message.author.id)
|
||||
if user_id in user_stats and user_stats[user_id]["check_in_time"] is not None:
|
||||
await message.channel.send(f"{message.author.mention} You are already checked in. Please check out first.")
|
||||
return
|
||||
if message.content.lower() == "checking in":
|
||||
user_id = str(message.author.id)
|
||||
if user_id in user_stats and user_stats[user_id]["check_in_time"] is not None:
|
||||
await message.channel.send(
|
||||
f"{message.author.mention} You are already checked in. Please check out first."
|
||||
)
|
||||
return
|
||||
|
||||
check_in_time = datetime.now().timestamp()
|
||||
if user_id not in user_stats:
|
||||
user_stats[user_id] = {"check_ins": 0, "total_time": 0, "check_in_time": None}
|
||||
user_stats[user_id]["check_in_time"] = check_in_time
|
||||
await message.channel.send(f"{message.author.mention} You have been checked in. Please mute your microphone.")
|
||||
check_in_time = datetime.now().timestamp()
|
||||
if user_id not in user_stats:
|
||||
user_stats[user_id] = {
|
||||
"check_ins": 0,
|
||||
"total_time": 0,
|
||||
"check_in_time": None,
|
||||
}
|
||||
user_stats[user_id]["check_in_time"] = check_in_time
|
||||
await message.channel.send(
|
||||
f"{message.author.mention} You have been checked in. Please mute your microphone."
|
||||
)
|
||||
|
||||
elif message.content.lower() == "checking out":
|
||||
user_id = str(message.author.id)
|
||||
if user_id not in user_stats or user_stats[user_id]["check_in_time"] is None:
|
||||
await message.channel.send(f"{message.author.mention} You have not checked in yet. Please check in first.")
|
||||
return
|
||||
elif message.content.lower() == "checking out":
|
||||
user_id = str(message.author.id)
|
||||
if user_id not in user_stats or user_stats[user_id]["check_in_time"] is None:
|
||||
await message.channel.send(
|
||||
f"{message.author.mention} You have not checked in yet. Please check in first."
|
||||
)
|
||||
return
|
||||
|
||||
check_out_time = datetime.now().timestamp()
|
||||
check_in_time = user_stats[user_id]["check_in_time"]
|
||||
time_delta = check_out_time - check_in_time
|
||||
user_stats[user_id]["check_ins"] += 1
|
||||
user_stats[user_id]["total_time"] += time_delta
|
||||
user_stats[user_id]["check_in_time"] = None
|
||||
check_out_time = datetime.now().timestamp()
|
||||
check_in_time = user_stats[user_id]["check_in_time"]
|
||||
time_delta = check_out_time - check_in_time
|
||||
user_stats[user_id]["check_ins"] += 1
|
||||
user_stats[user_id]["total_time"] += time_delta
|
||||
user_stats[user_id]["check_in_time"] = None
|
||||
|
||||
with open("user_stats.json", "w") as f:
|
||||
json.dump(user_stats, f)
|
||||
await message.channel.send(f"{message.author.mention} You have been checked out. Your session was {time_delta:.2f} seconds.")
|
||||
with open("user_stats.json", "w") as f:
|
||||
json.dump(user_stats, f)
|
||||
await message.channel.send(
|
||||
f"{message.author.mention} You have been checked out. Your session was {time_delta:.2f} seconds."
|
||||
)
|
||||
|
||||
elif message.content.lower() == "stats":
|
||||
stats_embed = discord.Embed(title="User stats :trophy:", color=0x000000)
|
||||
sorted_user_stats = sorted(user_stats.items(), key=lambda x: x[1]["total_time"], reverse=True)
|
||||
table_rows = [["Name", "Check-ins", "Total Time"]]
|
||||
for user_id, stats in sorted_user_stats:
|
||||
if stats["check_in_time"] is None:
|
||||
total_time_seconds = stats["total_time"]
|
||||
hours, total_time_seconds = divmod(total_time_seconds, 3600)
|
||||
minutes, total_time_seconds = divmod(total_time_seconds, 60)
|
||||
seconds, fractions = divmod(total_time_seconds, 1)
|
||||
fractions_str = f"{fractions:.3f}"[2:]
|
||||
username = garfbot.get_user(int(user_id)).name
|
||||
table_rows.append([username, str(stats["check_ins"]), f"{int(hours)}h {int(minutes)}m {int(seconds)}s {fractions_str}ms"])
|
||||
else:
|
||||
username = garfbot.get_user(int(user_id)).name
|
||||
table_rows.append([username, "Currently checked in", "-"])
|
||||
table_columns = list(zip(*table_rows[1:]))
|
||||
table_fields = table_rows[0]
|
||||
for field, values in zip(table_fields, table_columns):
|
||||
stats_embed.add_field(name=field, value="\n".join(values), inline=True)
|
||||
await message.channel.send(embed=stats_embed)
|
||||
elif message.content.lower() == "stats":
|
||||
stats_embed = discord.Embed(title="User stats :trophy:", color=0x000000)
|
||||
sorted_user_stats = sorted(
|
||||
user_stats.items(), key=lambda x: x[1]["total_time"], reverse=True
|
||||
)
|
||||
table_rows = [["Name", "Check-ins", "Total Time"]]
|
||||
for user_id, stats in sorted_user_stats:
|
||||
if stats["check_in_time"] is None:
|
||||
total_time_seconds = stats["total_time"]
|
||||
hours, total_time_seconds = divmod(total_time_seconds, 3600)
|
||||
minutes, total_time_seconds = divmod(total_time_seconds, 60)
|
||||
seconds, fractions = divmod(total_time_seconds, 1)
|
||||
fractions_str = f"{fractions:.3f}"[2:]
|
||||
username = garfbot.get_user(int(user_id)).name
|
||||
table_rows.append(
|
||||
[
|
||||
username,
|
||||
str(stats["check_ins"]),
|
||||
f"{int(hours)}h {int(minutes)}m {int(seconds)}s {fractions_str}ms",
|
||||
]
|
||||
)
|
||||
else:
|
||||
username = garfbot.get_user(int(user_id)).name
|
||||
table_rows.append([username, "Currently checked in", "-"])
|
||||
table_columns = list(zip(*table_rows[1:]))
|
||||
table_fields = table_rows[0]
|
||||
for field, values in zip(table_fields, table_columns):
|
||||
stats_embed.add_field(name=field, value="\n".join(values), inline=True)
|
||||
await message.channel.send(embed=stats_embed)
|
||||
|
||||
Reference in New Issue
Block a user