secstate/kowalski
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix display to use display names

Browse Source
This commit is contained in:
Colin Powell
2025-02-07 20:48:06 -05:00
parent 697530f2dd
commit 9bc94066ab
1 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
kowalski.py
View File

@ -74,10 +74,13 @@ def get_username(user_id):
response = client.users_info(user=user_id) response = client.users_info(user=user_id)
if response.get("ok"): if response.get("ok"):
username = response.get("user", {}).get("name") username = response.get("user", {}).get("name")
return username display_name = response.get("user", {}).get("profile", {}).get("display_name")
if not display_name:
display_name = response.get("user", {}).get("profile", {}).get("real_name")
return username, display_name
except SlackApiError as e: except SlackApiError as e:
print(f"Error fetching user info: {e.response.get('error')}") print(f"Error fetching user info: {e.response.get('error')}")
return None return None, None
def update_message_count(user_id): def update_message_count(user_id):
"""Increment the message count for the user in SQLite.""" """Increment the message count for the user in SQLite."""
@ -97,8 +100,8 @@ def update_message_count(user_id):
return new_count return new_count
def record_message(sender_id, receiver_id, message): def record_message(sender_id, receiver_id, message):
sender = get_username(sender_id) sender, display_name = get_username(sender_id)
receiver = get_username(receiver_id) receiver, display_name = get_username(receiver_id)
cursor.execute("INSERT INTO messages (sender_user_id, sender_user, receiver_user_id, receiver_user, message) VALUES (?, ?, ?, ?, ?)", (sender_id, sender, receiver_id, receiver, 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() conn.commit()
@ -128,10 +131,10 @@ def handle_message_events(event, say):
if indicator_found: if indicator_found:
for user_id in mentioned_users: for user_id in mentioned_users:
username = get_username(user_id) username, display_name = get_username(user_id)
sender_is_receiver = sender_id == user_id and not username in BACKDOOR_USERS sender_is_receiver = sender_id == user_id and not username in BACKDOOR_USERS
if sender_is_receiver: if sender_is_receiver:
say(f"Nice try {username} ... ") say(f"Nice try {display_name} ... ")
break break
username = get_username(user_id) username = get_username(user_id)
@ -140,7 +143,7 @@ def handle_message_events(event, say):
if message and not sender_is_receiver: if message and not sender_is_receiver:
record_message(sender_id, user_id, message) record_message(sender_id, user_id, message)
random_congrats = random.choice(CONGRATS) random_congrats = random.choice(CONGRATS)
say(f"{username} has ${user_count}, {random_congrats}") say(f"{display_name} has ${user_count}, {random_congrats}")
# Start the bot # Start the bot
if __name__ == "__main__": if __name__ == "__main__":