From 37d32b3faf4701ec9d7d20e1f9a6429d5cc7bf23 Mon Sep 17 00:00:00 2001 From: SimolZimol <70102430+SimolZimol@users.noreply.github.com> Date: Thu, 5 Sep 2024 14:29:42 +0200 Subject: [PATCH] modified: bot.py --- bot.py | 48 +++++++++++++++++++++--------------------------- 1 file changed, 21 insertions(+), 27 deletions(-) diff --git a/bot.py b/bot.py index 21a2b56..d03a602 100644 --- a/bot.py +++ b/bot.py @@ -145,13 +145,30 @@ def update_user_data(user_id, field, value): db_connection.commit() +def connect_to_database(): + return mysql.connector.connect( + host=DB_HOST, + port=DB_PORT, + user=DB_USER, + password=DB_PASSWORD, + database=DB_DATABASE + ) + +def close_database_connection(connection): + connection.close() + def load_user_data_from_mysql(user_id): + connection = connect_to_database() + cursor = connection.cursor() select_query = "SELECT * FROM user_data WHERE user_id = %s" - db_cursor.execute(select_query, (user_id,)) - result = db_cursor.fetchone() + cursor.execute(select_query, (user_id,)) + result = cursor.fetchone() + + cursor.close() + close_database_connection(connection) if result: - user_data = { + return { "user_id": result[0], "permission": result[1], "points": result[2], @@ -162,30 +179,7 @@ def load_user_data_from_mysql(user_id): "chat_history": json.loads(result[7]) if result[7] else [] } else: - # Wenn keine Benutzerdaten vorhanden sind, erstelle neue Daten - user_data = { - "user_id": user_id, - "permission": 0, # Standardberechtigung - "points": 0, # Standardpunkte - "ban": 0, # Standardbannstatus - "askmultus": 0, # Standardwert für askmultus - "filter_value": 0, # Standardwert für Filter - "rank": 0, # Standardrang - "chat_history": [] # Leerer Chatverlauf - } - - # Fügen Sie die neuen Benutzerdaten zur Datenbank hinzu - insert_user_data( - user_data["user_id"], - user_data["permission"], - user_data["points"], - user_data["ban"], - user_data["askmultus"], - user_data["filter_value"], - user_data["chat_history"] - ) - - return user_data + return None def save_user_data_to_mysql(user_data): update_query = """