Simon/Discord-ai-chatbot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

modified: bot.py

Browse Source
This commit is contained in:
SimolZimol
2024-10-29 19:49:59 +01:00
parent cbd5666f56
commit a5f3d9719f
1 changed files with 43 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

71
bot.py
View File

@@ -280,48 +280,51 @@ def save_global_permission(user_id, permission_level):
#----------------------------------------------------------------------------------------------------------- #-----------------------------------------------------------------------------------------------------------
async def update_all_users(): async def update_all_users():
try: connection = connect_to_database()
connection = connect_to_database() cursor = connection.cursor()
cursor = connection.cursor()
# Holen Sie alle guild_id-Einträge von der 'guilds'-Tabelle # Hole alle Guild-IDs aus der guilds-Datenbank
cursor.execute("SELECT guild_id FROM guilds") cursor.execute("SELECT guild_id FROM guilds")
guild_ids = [guild_id[0] for guild_id in cursor.fetchall()] # Extrahiert alle guild_id Werte als Liste guilds = cursor.fetchall()
count = 0 for (guild_id,) in guilds:
for guild_id in guild_ids: guild = client.get_guild(int(guild_id))
guild = client.get_guild(int(guild_id)) if not guild:
if guild is None: continue # Falls der Bot auf diesem Server nicht mehr aktiv ist, überspringen
continue
for member in guild.members: print(f"Updating users for guild: {guild_id}")
user_data = load_user_data(member.id, guild.id)
nickname = member.nick if member.nick else member.name # Aktualisiere Benutzerdaten für alle Mitglieder im Server
profile_picture = member.display_avatar.url for member in guild.members:
join_date = member.joined_at.date() if member.joined_at else None user_id = member.id
user_data = load_user_data(user_id, guild_id)
# Update nickname, profile_picture, and join_date in database print(f"Processing member: {member.display_name} (ID: {user_id})")
update_user_data(member.id, guild.id, "nickname", nickname)
update_user_data(member.id, guild.id, "profile_picture", profile_picture)
update_user_data(member.id, guild.id, "join_date", join_date)
# Check if the user is still on the server, update leave_date if not # Aktualisiere Nickname, Profilbild und Join-Datum für vorhandene Mitglieder
if not member.bot and member.status == discord.Status.offline: update_user_data(user_id, guild_id, "nickname", member.display_name)
leave_date = datetime.now().date() update_user_data(user_id, guild_id, "profile_picture", str(member.display_avatar.url))
update_user_data(member.id, guild.id, "leave_date", leave_date) update_user_data(user_id, guild_id, "join_date", member.joined_at.date())
else:
update_user_data(member.id, guild.id, "leave_date", None)
count += 1 # Entferne leave_date, falls der Benutzer noch auf dem Server ist und leave_date gesetzt war
if count >= 100: # After every 100 users, take a short pause if user_data.get("leave_date") is not None:
await asyncio.sleep(1) # Short pause print(f"Clearing leave_date for user: {member.display_name}")
count = 0 update_user_data(user_id, guild_id, "leave_date", None)
cursor.close() # Überprüfe alle Benutzer in `user_data`, die nicht mehr auf dem Server sind
connection.close() cursor.execute("SELECT user_id FROM user_data WHERE guild_id = %s", (guild_id,))
except Exception as e: all_users = cursor.fetchall()
logger.error(f"Error updating all users: {e}")
for (user_id,) in all_users:
member = guild.get_member(int(user_id))
if member is None: # Benutzer hat den Server verlassen
print(f"User {user_id} has left the guild {guild_id}, setting leave_date.")
leave_date = datetime.now().date()
update_user_data(user_id, guild_id, "leave_date", leave_date)
cursor.close()
connection.close()
print("User data update completed.")
def save_giveaway_to_db(guild_id, platform, name, prize_uuid, game_key): def save_giveaway_to_db(guild_id, platform, name, prize_uuid, game_key):
connection = connect_to_database() connection = connect_to_database()