def group_message_counter(group_id, read_only=False, reset=False): with LOCK: try: if read_only: session = get_session() group = get_unique_from_query( session.query( model.Group).filter(model.Group.id == group_id)) return group.message_num if reset: session = get_session() group = get_unique_from_query( session.query( model.Group).filter(model.Group.id == group_id)) group.message_num = 0 return group.message_num else: session = get_session() group = get_unique_from_query( session.query( model.Group).filter(model.Group.id == group_id)) if group.message_num is None: group.message_num = 0 group.message_num += 1 return group.message_num finally: session.commit() session.close()
def get_admin_from_linked(chat_id): try: session = get_session() linked = get_unique_from_query( session.query(LinkedGroups).filter( LinkedGroups.linked_id == chat_id)) if linked is None: session.close() return None return get_unique_from_query( session.query(AdminGroups).filter( AdminGroups.id == linked.admin_id)) finally: session.close()
def commit_user(user_id, alias=None, level=None, profession=None, house=None, team=None, validation=None, profession_level=None, fort_level=None): with LOCK: session = get_session() user = get_unique_from_query( session.query(User).filter(User.id == user_id)) if alias: user.alias = alias if level: user.level = level if house: user.house = house if validation: user.validation = validation if profession: user.profession = profession if team: user.team = team if profession_level: user.profession_level = profession_level if fort_level: user.fort_level = fort_level session.commit() session.close()
def get_real_user(user_id): try: session = get_session() return get_unique_from_query( session.query(User).filter(User.id == user_id)) finally: session.close()
def get_user_by_name(username): try: session = get_session() return get_unique_from_query( session.query(User).filter(User.alias == username)) finally: session.close()
def get_particular_admin(chat_id): try: session = get_session() return get_unique_from_query( session.query(SettingsAdmin).filter(SettingsAdmin.id == chat_id)) finally: session.close()
def get_admin(chat_id): try: session = get_session() return get_unique_from_query( session.query(AdminGroups).filter(AdminGroups.id == chat_id)) finally: session.close()
def set_ghost(user_id, ghost_bool): with LOCK: session = get_session() user = get_unique_from_query( session.query(User).filter(User.id == user_id)) user.ghost = ghost_bool session.commit() session.close()
def get_real_group(group_id): try: session = get_session() group = get_unique_from_query( session.query(model.Group).filter(model.Group.id == group_id)) return group finally: session.close()
def get_welcome_settings(chat_id): try: session = get_session() group = get_unique_from_query( session.query(Welcome).filter(Welcome.chat_id == chat_id)) return group finally: session.close()
def set_fc(fc, user_id): with LOCK: session = get_session() user = get_unique_from_query( session.query(User).filter(User.id == user_id)) user.friend_id = fc session.commit() session.close()
def get_plant(plant_id): try: session = get_session() plant = get_unique_from_query( session.query(model.Plants).filter(model.Plants.id == plant_id)) return plant finally: session.close()
def get_join_settings(chat_id): try: session = get_session() group = get_unique_from_query( session.query(SettingsJoin).filter(SettingsJoin.id == chat_id)) return group finally: session.close()
def get_group_settings(group_id): try: session = get_session() group = get_unique_from_query( session.query(SettingsGroup).filter(SettingsGroup.id == group_id)) return group finally: session.close()
def get_warn_limit(group_id): try: session = get_session() group = get_unique_from_query( session.query(SettingsGroup).filter(SettingsGroup.id == group_id)) return group.warn.value finally: session.close()
def update_fclist(user_id): with LOCK: session = get_session() user = get_unique_from_query( session.query(User).filter(User.id == user_id)) user.fclists = not user.fclists session.commit() session.close()
def get_nanny_settings(chat_id): try: session = get_session() nanny = get_unique_from_query( session.query(SettingsNurse).filter(SettingsNurse.id == chat_id)) return nanny finally: session.close()
def update_ranking(user_id): with LOCK: session = get_session() user = get_unique_from_query( session.query(User).filter(User.id == user_id)) user.ranking = not user.ranking session.commit() session.close()
def set_staff(user_id, staff_bool): with LOCK: session = get_session() user = get_unique_from_query( session.query(User).filter(User.id == user_id)) user.staff = staff_bool session.commit() session.close()
def get_poi(poi_id): try: session = get_session() poi = get_unique_from_query( session.query(model.Portals).filter(model.Portals.id == poi_id)) return poi finally: session.close()
def set_nanny_settings(chat_id, settings_str): with LOCK: session = get_session() nanny = get_unique_from_query( session.query(SettingsNurse).filter(SettingsNurse.id == chat_id)) nanny.set_nurseset_from_str(settings_str) session.commit() session.close() return
def set_nanny_reply(chat_id, text): with LOCK: session = get_session() nanny = get_unique_from_query( session.query(SettingsNurse).filter(SettingsNurse.id == chat_id)) nanny.reply = text session.commit() session.close() return
def unban_user(user_id): with LOCK: session = get_session() user = get_unique_from_query( session.query(User).filter(User.id == user_id)) user.flag = "" user.banned = False session.commit() session.close()
def set_admin_settings(chat_id, settings_str): with LOCK: session = get_session() admin = get_unique_from_query( session.query(AdminGroups).filter(AdminGroups.id == chat_id)) admin.set_admset_from_str(settings_str) session.commit() session.close() return
def set_ladmin_settings(chat_id, settings_str): with LOCK: session = get_session() admin = get_unique_from_query( session.query(SettingsAdmin).filter(SettingsAdmin.id == chat_id)) admin.set_setadm_from_str(settings_str) session.commit() session.close() return
def set_welcome_settings(chat_id, settings_str): with LOCK: session = get_session() group = get_unique_from_query( session.query(Welcome).filter(Welcome.chat_id == chat_id)) group.set_welcomeset_from_str() session.commit() session.close() return
def set_join_settings(chat_id, settings_str): with LOCK: session = get_session() group = get_unique_from_query( session.query(SettingsJoin).filter(SettingsJoin.id == chat_id)) group.set_joinset_from_str(settings_str) session.commit() session.close() return
def set_welcome_cooldown(chat_id, delete_cooldown): with LOCK: session = get_session() group = get_unique_from_query( session.query(SettingsJoin).filter(SettingsJoin.id == chat_id)) group.delete_cooldown = delete_cooldown session.commit() session.close() return
def set_max_members(chat_id, max_members): with LOCK: session = get_session() group = get_unique_from_query( session.query(SettingsJoin).filter(SettingsJoin.id == chat_id)) group.max_members = max_members session.commit() session.close() return
def ban_user(user_id, flag): with LOCK: session = get_session() user = get_unique_from_query( session.query(User).filter(User.id == user_id)) user.flag = flag user.banned = True session.commit() session.close()