| 123456789101112131415161718 |
- 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)
|
