alterware/aw-bot
1
0
Fork 0
mirror of https://github.com/alterware/aw-bot.git synced 2025-10-26 22:16:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: migrate DB table

Browse Source
This commit is contained in:
6arelyFuture
2025-03-25 10:41:44 +01:00
parent 05809597b7
commit f3019d93b8
4 changed files with 10 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
database/__init__.py
View File

@@ -59,28 +59,28 @@ def remove_pattern(pattern_id: int):
conn.close()
def migrate_users_with_role(user_id: int, role_id: int):
def migrate_users_with_role(user_id: int, role_id: int, user_name: str):
"""Migrates existing users with the role to the new table."""
conn = sqlite3.connect(DB_PATH)
cursor = conn.cursor()
cursor.execute(
"INSERT OR IGNORE INTO user_roles (user_id, role_id, date_assigned) VALUES (?, ?, ?)",
(user_id, role_id, aware_utcnow().isoformat()),
"INSERT OR IGNORE INTO user_roles (user_id, role_id, date_assigned, user_name) VALUES (?, ?, ?, ?)",
(user_id, role_id, aware_utcnow().isoformat(), user_name),
)
conn.commit()
conn.close()
def add_user_to_role(user_id: int, role_id: int):
def add_user_to_role(user_id: int, role_id: int, user_name: str):
"""Adds a new user when they receive the role."""
conn = sqlite3.connect(DB_PATH)
cursor = conn.cursor()
cursor.execute(
"INSERT OR REPLACE INTO user_roles (user_id, role_id, date_assigned) VALUES (?, ?, ?)",
(user_id, role_id, aware_utcnow().isoformat()),
"INSERT OR REPLACE INTO user_roles (user_id, role_id, date_assigned, user_name) VALUES (?, ?, ?, ?)",
(user_id, role_id, aware_utcnow().isoformat(), user_name),
)
conn.commit()