def get_royals(role: str = None):
with SESSION() as local_session:
if not role:
return local_session.query(Royals).all()
else:
return local_session.query(Royals).filter(
Royals.role_name == role).all()
def remove_royal(user_id: int):
with SESSION() as local_session:
try:
ret = local_session.query(Royals).get({"user_id": user_id})
if ret:
local_session.delete(ret)
local_session.commit()
except Exception:
traceback.print_exc()
local_session.rollback()
def set_royal_role(user_id: int, role: str):
with SESSION() as local_session:
try:
# Check if the user exists first and create them if they don't.
ret = local_session.query(Royals).get({"user_id": user_id})
if not ret:
ret = Royals(user_id, role)
local_session.add(ret)
else:
ret.role_name = role
local_session.commit()
local_session.flush()
except Exception:
traceback.print_exc()
local_session.rollback()
def get_royal_role(user_id: int):
with SESSION() as local_session:
ret = local_session.query(Royals).get({"user_id": user_id})
if ret:
return ret.role_name
return None
def is_royal(user_id: int, role: str = None): with SESSION() as local_session: if role: return bool(local_session.query(Royals).get((user_id, role))) else: return bool(local_session.query(Royals).get(user_id))
