From 1a9f18e98bba0f2b947b516fe1e15d6fe7fd5b08 Mon Sep 17 00:00:00 2001 From: Colin Powell Date: Fri, 7 Feb 2025 21:03:00 -0500 Subject: [PATCH] Allow us to see how rich we are --- kowalski.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/kowalski.py b/kowalski.py index 93d3457..0c01210 100644 --- a/kowalski.py +++ b/kowalski.py @@ -105,6 +105,20 @@ def record_message(sender_id, receiver_id, message): cursor.execute("INSERT INTO messages (sender_user_id, sender_user, receiver_user_id, receiver_user, message) VALUES (?, ?, ?, ?, ?)", (sender_id, sender, receiver_id, receiver, message)) conn.commit() +def get_all_counts(): + """Fetch all user message counts from SQLite and format them as a string.""" + cursor.execute("SELECT user_id, count FROM message_counts") + rows = cursor.fetchall() + + if not rows: + return "No message counts recorded yet." + + response = "*The Richest:*\n" + for user_id, count in rows: + user, display = get_username(user_id) + response += f"- <@{display}> has ${count}\n" + + return response @app.event("message") def handle_message_events(event, say): @@ -118,6 +132,11 @@ def handle_message_events(event, say): print("No message .. probs a delete") return + if text.lower() == "how rich are we?": + response = get_all_counts() + say(response) + return + indicator_found = len(text.split(INDICATOR)) > 1 try: mentioned_users = re.findall(r"<@(\w+)>", text)