18 lines
708 B
Python
18 lines
708 B
Python
from nio.events.room_events import RoomMessageText
|
|
from nio.rooms import MatrixRoom
|
|
|
|
|
|
async def command_stats(room: MatrixRoom, event: RoomMessageText, bot):
|
|
bot.logger.log("Showing stats...")
|
|
|
|
if not bot.database:
|
|
bot.logger.log("No database connection - cannot show stats")
|
|
bot.send_message(room, "Sorry, I'm not connected to a database, so I don't have any statistics on your usage.", True)
|
|
return
|
|
|
|
with bot.database.cursor() as cursor:
|
|
cursor.execute(
|
|
"SELECT SUM(tokens) FROM token_usage WHERE room_id = ?", (room.room_id,))
|
|
total_tokens = cursor.fetchone()[0] or 0
|
|
|
|
bot.send_message(room, f"Total tokens used: {total_tokens}", True)
|