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()
|
||||
|
||||
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 = """
|
||||
|
||||
Reference in New Issue
Block a user