Browse Source

Fixing that the Roles column in the User table can store multiple roles properly by changing its data type to TEXT and ensuring the userid is unique.

BaumSplitter41 4 months ago
parent
commit
ce0638bfa4
1 changed files with 11 additions and 4 deletions
  1. 11 4
      VPD_BOT/main.py

+ 11 - 4
VPD_BOT/main.py

@@ -71,9 +71,10 @@ conn.database = dbdb
 cursor.execute("""
 CREATE TABLE IF NOT EXISTS User (
     id INT AUTO_INCREMENT PRIMARY KEY,
-    userid BIGINT,
+    userid BIGINT UNIQUE,
     discordname VARCHAR(100),
-    roles INT
+    rolesnumber INT,
+    Roles TEXT
 )
 """)
 
@@ -536,11 +537,17 @@ async def update_users_periodically():
     while not bot.is_closed():
         try:
             for guild in bot.guilds:
+                batch_count = 0
                 async for member in guild.fetch_members(limit=None):
                     cursor.execute(
-                        "INSERT INTO User (userid, discordname, roles) VALUES (%s, %s, %s) ON DUPLICATE KEY UPDATE discordname=%s, roles=%s",
-                        (member.id, str(member), len(member.roles), str(member), len(member.roles))
+                        "INSERT INTO User (userid, discordname, rolesnumber, roles) VALUES (%s, %s, %s, %s) ON DUPLICATE KEY UPDATE discordname=%s, rolesnumber=%s, roles=%s",
+                        (member.id, str(member), len(member.roles), str(member.roles), str(member), len(member.roles), str(member.roles))
                     )
+                    batch_count += 1
+                    if batch_count >= 100:
+                        conn.commit()
+                        batch_count = 0
+                if batch_count > 0:
                     conn.commit()
         except Exception as e:
             print(f"Error updating users: {e}")