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))