It's version 0.6 I add users DB

This commit is contained in:
Niken
2025-11-16 14:15:44 +03:00
parent e1837400ef
commit be9ec785f4
5 changed files with 192 additions and 50 deletions
+33
View File
@@ -0,0 +1,33 @@
from .DB import get_db
def save_user(user_id: int, group: str = "30тс"):
db = get_db()
cur = db.cursor()
cur.execute("INSERT INTO users (user_id, user_group) VALUES (?, ?)", (user_id, group))
db.commit()
cur.close()
db.close()
def set_group(user_id: int, group: str = "30тс"):
db = get_db()
cur = db.cursor()
cur.execute("UPDATE users SET user_group = ? WHERE user_id = ?", (group, user_id))
db.commit()
cur.close()
db.close()
def get_group(user_id: int, default: str = "30тс") -> str:
db = get_db()
cur = db.cursor()
cur.execute("SELECT user_group FROM users WHERE user_id = ?", (user_id,))
row = cur.fetchone()
if row:
group = row[0]
else:
# если пользователя нет — регистрируем с дефолтной группой
cur.execute("INSERT INTO users (user_id, user_group) VALUES (?, ?)", (user_id, default))
db.commit()
group = default
cur.close()
db.close()
return group