Niken
57 lines
1.4 KiB
Python
57 lines
1.4 KiB
Python
from config import Config
|
|
|
|
_DEFAULT_GROUP = "30тс"
|
|
|
|
|
|
def save_user(user_id: int, group: str = _DEFAULT_GROUP):
|
|
if Config.DISABLE_STORAGE:
|
|
return
|
|
from .DB import get_db
|
|
|
|
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 = _DEFAULT_GROUP):
|
|
if Config.DISABLE_STORAGE:
|
|
return
|
|
from .DB import get_db
|
|
|
|
db = get_db()
|
|
cur = db.cursor()
|
|
cur.execute("SELECT 1 FROM users WHERE user_id = ?", (user_id,))
|
|
exists = cur.fetchone()
|
|
|
|
if exists:
|
|
cur.execute("UPDATE users SET user_group = ? WHERE user_id = ?", (group, user_id))
|
|
else:
|
|
cur.execute("INSERT INTO users (user_id, user_group) VALUES (?, ?)", (user_id, group))
|
|
|
|
db.commit()
|
|
cur.close()
|
|
db.close()
|
|
|
|
|
|
def get_group(user_id: int, default: str = _DEFAULT_GROUP) -> str:
|
|
if Config.DISABLE_STORAGE:
|
|
return default
|
|
from .DB import get_db
|
|
|
|
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
|