modified: bot.py
This commit is contained in:
48
bot.py
48
bot.py
@@ -145,13 +145,30 @@ def update_user_data(user_id, field, value):
|
|||||||
|
|
||||||
db_connection.commit()
|
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):
|
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"
|
select_query = "SELECT * FROM user_data WHERE user_id = %s"
|
||||||
db_cursor.execute(select_query, (user_id,))
|
cursor.execute(select_query, (user_id,))
|
||||||
result = db_cursor.fetchone()
|
result = cursor.fetchone()
|
||||||
|
|
||||||
|
cursor.close()
|
||||||
|
close_database_connection(connection)
|
||||||
|
|
||||||
if result:
|
if result:
|
||||||
user_data = {
|
return {
|
||||||
"user_id": result[0],
|
"user_id": result[0],
|
||||||
"permission": result[1],
|
"permission": result[1],
|
||||||
"points": result[2],
|
"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 []
|
"chat_history": json.loads(result[7]) if result[7] else []
|
||||||
}
|
}
|
||||||
else:
|
else:
|
||||||
# Wenn keine Benutzerdaten vorhanden sind, erstelle neue Daten
|
return None
|
||||||
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
|
|
||||||
|
|
||||||
def save_user_data_to_mysql(user_data):
|
def save_user_data_to_mysql(user_data):
|
||||||
update_query = """
|
update_query = """
|
||||||
|
|||||||
Reference in New Issue
Block a user