modified: bot.py

This commit is contained in:
SimolZimol
2024-09-05 14:29:42 +02:00
parent 6d024b814e
commit 37d32b3faf

48
bot.py
View File

@@ -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 = """