def remove_url(tg_chat_id, tg_feed_link): with INSERTION_LOCK: # this loops to delete any possible duplicates for the same TG User ID, TG Chat ID and link for row in check_url_availability(tg_chat_id, tg_feed_link): # add the action to the DB query SESSION.delete(row) SESSION.commit()
def get_welc_mutes_pref(chat_id): welcomemutes = SESSION.query(WelcomeMute).get(str(chat_id)) SESSION.close() if welcomemutes: return welcomemutes.welcomemutes return False
def get_gdbye_pref(chat_id): welc = SESSION.query(Welcome).get(str(chat_id)) SESSION.close() if welc: return welc.should_goodbye, welc.custom_leave, welc.leave_type else: # Welcome by default. return True, DEFAULT_GOODBYE, Types.TEXT
def migrate_chat(old_chat_id, new_chat_id): with BLACKLIST_FILTER_INSERTION_LOCK: chat_filters = ( SESSION.query(BlackListFilters).filter( BlackListFilters.chat_id == str(old_chat_id)).all()) for filt in chat_filters: filt.chat_id = str(new_chat_id) SESSION.commit()
def welcome_mutes(chat_id): try: welcomemutes = SESSION.query(WelcomeMute).get(str(chat_id)) if welcomemutes: return welcomemutes.welcomemutes return False finally: SESSION.close()
def __load_all_feds_settings(): global FEDERATION_NOTIFICATION try: getuser = SESSION.query(FedsUserSettings).all() for x in getuser: FEDERATION_NOTIFICATION[str(x.user_id)] = x.should_report finally: SESSION.close()
def allow_connect_to_chat(chat_id: Union[str, int]) -> bool: try: chat_setting = SESSION.query(ChatAccessConnectionSettings).get(str(chat_id)) if chat_setting: return chat_setting.allow_connect_to_chat return False finally: SESSION.close()
def get_clean_pref(chat_id): welc = SESSION.query(Welcome).get(str(chat_id)) SESSION.close() if welc: return welc.clean_welcome return False
def num_warn_chat_filters(chat_id): try: return ( SESSION.query(WarnFilters.chat_id) .filter(WarnFilters.chat_id == str(chat_id)) .count() ) finally: SESSION.close()
def get_human_checks(user_id, chat_id): try: human_check = SESSION.query(WelcomeMuteUsers).get((user_id, str(chat_id))) if not human_check: return None human_check = human_check.human_check return human_check finally: SESSION.close()
def __load_gbanned_userid_list(): global GBANNED_LIST try: GBANNED_LIST = { x.user_id for x in SESSION.query(GloballyBannedUsers).all() } finally: SESSION.close()
def set_chat_setting(chat_id: Union[int, str], setting: bool): with CHAT_LOCK: chat_setting = SESSION.query(ReportingChatSettings).get(str(chat_id)) if not chat_setting: chat_setting = ReportingChatSettings(chat_id) chat_setting.should_report = setting SESSION.add(chat_setting) SESSION.commit()
def new_fed(owner_id, fed_name, fed_id): with FEDS_LOCK: global FEDERATION_BYOWNER, FEDERATION_BYFEDID, FEDERATION_BYNAME fed = Federations(str(owner_id), fed_name, str(fed_id), 'Rules is not set in this federation.', None, str({ 'owner': str(owner_id), 'members': '[]' })) SESSION.add(fed) SESSION.commit() FEDERATION_BYOWNER[str(owner_id)] = ({ 'fid': str(fed_id), 'fname': fed_name, 'frules': 'Rules is not set in this federation.', 'flog': None, 'fusers': str({ 'owner': str(owner_id), 'members': '[]' }) }) FEDERATION_BYFEDID[str(fed_id)] = ({ 'owner': str(owner_id), 'fname': fed_name, 'frules': 'Rules is not set in this federation.', 'flog': None, 'fusers': str({ 'owner': str(owner_id), 'members': '[]' }) }) FEDERATION_BYNAME[fed_name] = ({ 'fid': str(fed_id), 'owner': str(owner_id), 'frules': 'Rules is not set in this federation.', 'flog': None, 'fusers': str({ 'owner': str(owner_id), 'members': '[]' }) }) return fed
def __load_blacklist_userid_list(): global BLACKLIST_USERS try: BLACKLIST_USERS = { int(x.user_id) for x in SESSION.query(BlacklistUsers).all() } finally: SESSION.close()
def update_url(row_id, new_entry_links): with INSERTION_LOCK: row = SESSION.query(RSS).get(row_id) # set the new old_entry_link with the latest update from the RSS Feed row.old_entry_link = new_entry_links[0] # commit the changes to the DB SESSION.commit()
def __load_disabled_commands(): global DISABLED try: all_chats = SESSION.query(Disable).all() for chat in all_chats: DISABLED.setdefault(chat.chat_id, set()).add(chat.command) finally: SESSION.close()
def get_note(chat_id, note_name): try: return ( SESSION.query(Notes) .filter(func.lower(Notes.name) == note_name, Notes.chat_id == str(chat_id)) .first() ) finally: SESSION.close()
def is_enabled(chat_id): try: resultcurr = SESSION.query(CleanerBlueTextChatSettings).get( str(chat_id)) if resultcurr: return resultcurr.is_enable return False #default finally: SESSION.close()
def get_welc_pref(chat_id): welc = SESSION.query(Welcome).get(str(chat_id)) SESSION.close() if welc: return welc.should_welcome, welc.custom_welcome, welc.custom_content, welc.welcome_type else: # Welcome by default. return True, DEFAULT_WELCOME, None, Types.TEXT
def is_chat(chat_id): try: chat = SESSION.query(ChatbotChats).get(str(chat_id)) if chat: return True else: return False finally: SESSION.close()
def set_private_notes(chat_id, setting: bool): with PRIVATE_NOTES_INSERTION_LOCK: private_notes = SESSION.query(PrivateNotes).get(str(chat_id)) if not private_notes: private_notes = PrivateNotes(str(chat_id), setting=setting) private_notes.setting = setting SESSION.add(private_notes) SESSION.commit()
def set_allow_connect_to_chat(chat_id: Union[int, str], setting: bool): with CHAT_ACCESS_LOCK: chat_setting = SESSION.query(ChatAccessConnectionSettings).get(str(chat_id)) if not chat_setting: chat_setting = ChatAccessConnectionSettings(chat_id, setting) chat_setting.allow_connect_to_chat = setting SESSION.add(chat_setting) SESSION.commit()
def set_user_setting(user_id: int, setting: bool): with USER_LOCK: user_setting = SESSION.query(ReportingUserSettings).get(user_id) if not user_setting: user_setting = ReportingUserSettings(user_id) user_setting.should_report = setting SESSION.add(user_setting) SESSION.commit()
def set_clearcmd(chat_id, cmd, time): with CLEAR_CMD_LOCK: clear_cmd = SESSION.query(ClearCmd).get((str(chat_id), cmd)) if not clear_cmd: clear_cmd = ClearCmd(str(chat_id), cmd, time) clear_cmd.time = time SESSION.add(clear_cmd) SESSION.commit()
def __load_gban_stat_list(): global GBANSTAT_LIST try: GBANSTAT_LIST = { x.chat_id for x in SESSION.query(GbanSettings).all() if not x.setting } finally: SESSION.close()
def set_rules(chat_id, rules_text): with INSERTION_LOCK: rules = SESSION.query(Rules).get(str(chat_id)) if not rules: rules = Rules(str(chat_id)) rules.rules = rules_text SESSION.add(rules) SESSION.commit()
def set_clean_service(chat_id: Union[int, str], setting: bool): with CS_LOCK: chat_setting = SESSION.query(CleanServiceSetting).get(str(chat_id)) if not chat_setting: chat_setting = CleanServiceSetting(chat_id) chat_setting.clean_service = setting SESSION.add(chat_setting) SESSION.commit()
def __load_afk_users(): global AFK_USERS try: all_afk = SESSION.query(AFK).all() AFK_USERS = { user.user_id: {"reason": user.reason, "time": user.time} for user in all_afk if user.is_afk } finally: SESSION.close()
def set_user_me_info(user_id, info): with INSERTION_LOCK: userinfo = SESSION.query(UserInfo).get(user_id) if userinfo: userinfo.info = info else: userinfo = UserInfo(user_id, info) SESSION.add(userinfo) SESSION.commit()
def check_url_availability(tg_chat_id, tg_feed_link): try: return ( SESSION.query(RSS) .filter(RSS.feed_link == tg_feed_link, RSS.chat_id == tg_chat_id) .all() ) finally: SESSION.close()