def get_user_num_chats(user_id): try: return ( SESSION.query(ChatMembers).filter(ChatMembers.user == int(user_id)).count() ) finally: SESSION.close()
def rem_chat(chat_id): with INSERTION_LOCK: chat = SESSION.query(Chats).get(str(chat_id)) if chat: SESSION.delete(chat) SESSION.commit() else: SESSION.close()
def get_userid_by_name(username): try: return ( SESSION.query(Users) .filter(func.lower(Users.username) == username.lower()) .all() ) finally: SESSION.close()
def is_nsfw(chat_id): try: chat = SESSION.query(NSFWChats).get(str(chat_id)) if chat: return True else: return False finally: SESSION.close()
def del_user(user_id): with INSERTION_LOCK: curr = SESSION.query(Users).get(user_id) if curr: SESSION.delete(curr) SESSION.commit() return True ChatMembers.query.filter(ChatMembers.user == user_id).delete() SESSION.commit() SESSION.close() return False
def num_users(): try: return SESSION.query(Users).count() finally: SESSION.close()
def num_chats(): try: return SESSION.query(Chats).count() finally: SESSION.close()
def get_all_chats(): try: return SESSION.query(Chats).all() finally: SESSION.close()
def get_chat_members(chat_id): try: return SESSION.query(ChatMembers).filter(ChatMembers.chat == str(chat_id)).all() finally: SESSION.close()
def get_name_by_userid(user_id): try: return SESSION.query(Users).get(Users.user_id == int(user_id)).first() finally: SESSION.close()
def get_all_nsfw_chats(): try: return SESSION.query(NSFWChats.chat_id).all() finally: SESSION.close()