def migrate_chat(old_chat_id, new_chat_id): with INSERTION_LOCK: chat = SESSION.query(Welcome).get(str(old_chat_id)) if chat: chat.chat_id = str(new_chat_id) with WELC_BTN_LOCK: chat_buttons = SESSION.query(WelcomeButtons).filter(WelcomeButtons.chat_id == str(old_chat_id)).all() for btn in chat_buttons: btn.chat_id = str(new_chat_id) with LEAVE_BTN_LOCK: chat_buttons = SESSION.query(GoodbyeButtons).filter(GoodbyeButtons.chat_id == str(old_chat_id)).all() for btn in chat_buttons: btn.chat_id = str(new_chat_id) SESSION.commit()
def rm_note(chat_id, note_name): with NOTES_INSERTION_LOCK: note = SESSION.query(Notes).get((str(chat_id), note_name)) if note: with BUTTONS_INSERTION_LOCK: buttons = SESSION.query(Buttons).filter(Buttons.chat_id == str(chat_id), Buttons.note_name == note_name).all() for btn in buttons: SESSION.delete(btn) SESSION.delete(note) SESSION.commit() return True else: SESSION.close() return False
def set_frules(fed_id, rules): with FEDS_LOCK: r = SESSION.query(RulesF).get(fed_id) if r: print("fund prev") SESSION.delete(r) r = RulesF(str(fed_id), rules) SESSION.add(r) try: SESSION.commit() return r except: SESSION.rollback() return False finally: SESSION.commit()
def fban_user(fed_id, user_id, reason): with FEDS_LOCK: r = SESSION.query(BansF).all() for I in r: print("1") if I.fed_id == fed_id: print("2") print(I.user_id, user_id) if int(I.user_id) == int(user_id): print("fund prev") SESSION.delete(I) r = BansF(str(fed_id), user_id, reason) SESSION.add(r) SESSION.commit() return r
def set_flood_strength(chat_id, flood_type, value): # for flood_type # 1 = ban # 2 = kick # 3 = mute # 4 = tban # 5 = tmute with INSERTION_FLOOD_SETTINGS_LOCK: curr_setting = SESSION.query(FloodSettings).get(str(chat_id)) if not curr_setting: curr_setting = FloodSettings(chat_id, flood_type=int(flood_type), value=value) curr_setting.flood_type = int(flood_type) curr_setting.value = str(value) SESSION.add(curr_setting) SESSION.commit()
def set_cleanbt(chat_id, is_enable): with INSERTION_LOCK: curr = SESSION.query(CleanerBlueText).get(str(chat_id)) if curr: SESSION.delete(curr) curr = CleanerBlueText(str(chat_id), is_enable) if is_enable: if str(chat_id) not in CLEANER_BT_CHATS: CLEANER_BT_CHATS.append(str(chat_id)) else: if str(chat_id) in CLEANER_BT_CHATS: CLEANER_BT_CHATS.remove(str(chat_id)) SESSION.add(curr) SESSION.commit()
def disabledel_set(chat_id, is_enable): with DISABLEDEL_INSERTION_LOCK: curr = SESSION.query(DisableDelete).get(str(chat_id)) if curr: SESSION.delete(curr) curr = DisableDelete(str(chat_id), is_enable) if is_enable: if str(chat_id) not in DISABLEDEL: DISABLEDEL.append(str(chat_id)) else: if str(chat_id) in DISABLEDEL: DISABLEDEL.remove(str(chat_id)) SESSION.add(curr) SESSION.commit()
def chat_leave_fed(chat_id): with FEDS_LOCK: global FEDERATION_CHATS, FEDERATION_CHATS_BYID # Set variables fed_info = FEDERATION_CHATS.get(str(chat_id)) if fed_info == None: return False fed_id = fed_info['fid'] # Delete from cache FEDERATION_CHATS.pop(str(chat_id)) FEDERATION_CHATS_BYID[str(fed_id)].remove(str(chat_id)) # Delete from db curr = SESSION.query(ChatF).all() for U in curr: if int(U.chat_id) == int(chat_id): SESSION.delete(U) SESSION.commit() return True
def set_frules(fed_id, rules): with FEDS_LOCK: global FEDERATION_BYOWNER, FEDERATION_BYFEDID, FEDERATION_BYNAME # Variables getfed = FEDERATION_BYFEDID.get(str(fed_id)) owner_id = getfed['owner'] fed_name = getfed['fname'] fed_members = getfed['fusers'] fed_rules = str(rules) # Set user FEDERATION_BYOWNER[str(owner_id)]['frules'] = fed_rules FEDERATION_BYFEDID[str(fed_id)]['frules'] = fed_rules FEDERATION_BYNAME[fed_name]['frules'] = fed_rules # Set on database fed = Federations(str(owner_id), fed_name, str(fed_id), fed_rules, str(fed_members)) SESSION.merge(fed) SESSION.commit() return True
def add_note_to_db(chat_id, note_name, note_data, msgtype, buttons=None, file=None): if not buttons: buttons = [] with NOTES_INSERTION_LOCK: prev = SESSION.query(Notes).get((str(chat_id), note_name)) if prev: with BUTTONS_INSERTION_LOCK: prev_buttons = SESSION.query(Buttons).filter(Buttons.chat_id == str(chat_id), Buttons.note_name == note_name).all() for btn in prev_buttons: SESSION.delete(btn) SESSION.delete(prev) note = Notes(str(chat_id), note_name, note_data or "", msgtype=msgtype.value, file=file) SESSION.add(note) SESSION.commit() for b_name, url, same_line in buttons: add_note_button_to_db(chat_id, note_name, b_name, url, same_line)
def set_cleanbt(chat_id, is_enable): with CLEANER_CHAT_SETTINGS: curr = SESSION.query(CleanerBlueTextChatSettings).get(str(chat_id)) if not curr: curr = CleanerBlueTextChatSettings(str(chat_id), is_enable) else: curr.is_enabled = is_enable if str(chat_id) not in CLEANER_CHATS: CLEANER_CHATS.setdefault(str(chat_id), { "setting": False, "commands": set() }) CLEANER_CHATS[str(chat_id)]["setting"] = is_enable SESSION.add(curr) SESSION.commit()
def un_fban_user(fed_id, user_id): with FEDS_LOCK: r = SESSION.query(BansF).all() for I in r: print("1") if I.fed_id == fed_id: print("2") print(I.user_id, user_id) if int(I.user_id) == int(user_id): print("fund prev") SESSION.delete(I) try: SESSION.commit() return I except: SESSION.rollback() return False finally: SESSION.commit()
def migrate_chat(old_chat_id, new_chat_id): with INSERTION_LOCK: chat = SESSION.query(Chats).get(str(old_chat_id)) if chat: chat.chat_id = str(new_chat_id) SESSION.add(chat) SESSION.flush() chat_members = ( SESSION.query(ChatMembers) .filter(ChatMembers.chat == str(old_chat_id)) .all() ) for member in chat_members: member.chat = str(new_chat_id) SESSION.add(member) SESSION.commit()
def fban_user(fed_id, user_id, first_name, last_name, user_name, reason): with FEDS_LOCK: r = SESSION.query(BansF).all() for I in r: if I.fed_id == fed_id: if int(I.user_id) == int(user_id): SESSION.delete(I) r = BansF(str(fed_id), str(user_id), first_name, last_name, user_name, reason) SESSION.add(r) try: SESSION.commit() except: SESSION.rollback() return False finally: SESSION.commit() __load_all_feds_banned() return r
def user_join_fed(fed_id, user_id): with FEDS_LOCK: global FEDERATION_BYOWNER, FEDERATION_BYFEDID, FEDERATION_BYNAME # Variables getfed = FEDERATION_BYFEDID.get(str(fed_id)) owner_id = getfed['owner'] fed_name = getfed['fname'] fed_rules = getfed['frules'] # Temp set members = eval(eval(getfed['fusers'])['members']) members.append(user_id) # Set user FEDERATION_BYOWNER[str(owner_id)]['fusers'] = str({'owner': str(owner_id), 'members': str(members)}) FEDERATION_BYFEDID[str(fed_id)]['fusers'] = str({'owner': str(owner_id), 'members': str(members)}) FEDERATION_BYNAME[fed_name]['fusers'] = str({'owner': str(owner_id), 'members': str(members)}) # Set on database fed = Federations(str(owner_id), fed_name, str(fed_id), fed_rules, str({'owner': str(owner_id), 'members': str(members)})) SESSION.merge(fed) SESSION.commit() __load_all_feds_chats() return True
def remove_filter(chat_id, keyword): global CHAT_FILTERS with CUST_FILT_LOCK: filt = SESSION.query(CustomFilters).get((str(chat_id), keyword)) if filt: if keyword in CHAT_FILTERS.get(str(chat_id), []): # Sanity check CHAT_FILTERS.get(str(chat_id), []).remove(keyword) with BUTTON_LOCK: prev_buttons = SESSION.query(Buttons).filter( Buttons.chat_id == str(chat_id), Buttons.keyword == keyword).all() for btn in prev_buttons: SESSION.delete(btn) SESSION.delete(filt) SESSION.commit() return True SESSION.close() return False
def update_restriction(chat_id, restr_type, locked): with RESTR_LOCK: curr_restr = SESSION.query(Restrictions).get(str(chat_id)) if not curr_restr: curr_restr = init_restrictions(chat_id) if restr_type == "messages": curr_restr.messages = locked elif restr_type == "media": curr_restr.media = locked elif restr_type == "other": curr_restr.other = locked elif restr_type == "previews": curr_restr.preview = locked elif restr_type == "all": curr_restr.messages = locked curr_restr.media = locked curr_restr.other = locked curr_restr.preview = locked SESSION.add(curr_restr) SESSION.commit()
def remove_warn(user_id, chat_id): with WARN_INSERTION_LOCK: removed = False warned_user = SESSION.query(Warns).get((user_id, str(chat_id))) temp_reason = [] if warned_user and warned_user.num_warns > 0: warned_user.num_warns -= 1 if warned_user and warned_user.reasons is not None: for reason in warned_user.reasons: temp_reason.append(reason) del temp_reason[-1] warned_user.reasons = temp_reason SESSION.add(warned_user) SESSION.commit() removed = True SESSION.close() return removed
def add_filter(chat_id, keyword, reply, is_sticker=False, is_document=False, is_image=False, is_audio=False, is_voice=False, is_video=False, buttons=None): global CHAT_FILTERS if buttons is None: buttons = [] with CUST_FILT_LOCK: prev = SESSION.query(CustomFilters).get((str(chat_id), keyword)) if prev: with BUTTON_LOCK: prev_buttons = SESSION.query(Buttons).filter( Buttons.chat_id == str(chat_id), Buttons.keyword == keyword).all() for btn in prev_buttons: SESSION.delete(btn) SESSION.delete(prev) filt = CustomFilters(str(chat_id), keyword, reply, is_sticker, is_document, is_image, is_audio, is_voice, is_video, bool(buttons)) if keyword not in CHAT_FILTERS.get(str(chat_id), []): CHAT_FILTERS[str(chat_id)] = sorted( CHAT_FILTERS.get(str(chat_id), []) + [keyword], key=lambda x: (-len(x), x)) SESSION.add(filt) SESSION.commit() for b_name, url, same_line in buttons: add_note_button_to_db(chat_id, keyword, b_name, url, same_line)
def chat_ignore_command(chat_id, ignore): ignore = ignore.lower() with CLEANER_CHAT_LOCK: ignored = SESSION.query(CleanerBlueTextChat).get( (str(chat_id), ignore)) if not ignored: if str(chat_id) not in CLEANER_CHATS: CLEANER_CHATS.setdefault(str(chat_id), { "setting": False, "commands": set() }) CLEANER_CHATS[str(chat_id)]["commands"].add(ignore) ignored = CleanerBlueTextChat(str(chat_id), ignore) SESSION.add(ignored) SESSION.commit() return True SESSION.close() return False
def user_demote_fed(fed_id, user_id): with FEDS_LOCK: global FEDERATION_BYOWNER, FEDERATION_BYFEDID, FEDERATION_BYNAME # Variables getfed = FEDERATION_BYFEDID.get(str(fed_id)) owner_id = getfed['owner'] fed_name = getfed['fname'] fed_rules = getfed['frules'] # Temp set try: members = eval(eval(getfed['fusers'])['members']) except ValueError: return False members.remove(user_id) # Set user FEDERATION_BYOWNER[str(owner_id)]['fusers'] = str({ 'owner': str(owner_id), 'members': str(members) }) FEDERATION_BYFEDID[str(fed_id)]['fusers'] = str({ 'owner': str(owner_id), 'members': str(members) }) FEDERATION_BYNAME[fed_name]['fusers'] = str({ 'owner': str(owner_id), 'members': str(members) }) # Set on database fed = Federations( str(owner_id), fed_name, str(fed_id), fed_rules, str({ 'owner': str(owner_id), 'members': str(members) })) SESSION.merge(fed) SESSION.commit() return True
def chat_unignore_command(chat_id, unignore): unignore = unignore.lower() with CLEANER_CHAT_LOCK: unignored = SESSION.query(CleanerBlueTextChat).get( (str(chat_id), unignore)) if unignored: if str(chat_id) not in CLEANER_CHATS: CLEANER_CHATS.setdefault(str(chat_id), { "setting": False, "commands": set() }) if unignore in CLEANER_CHATS.get(str(chat_id)).get("commands"): CLEANER_CHATS[str(chat_id)]["commands"].remove(unignore) SESSION.delete(unignored) SESSION.commit() return True SESSION.close() return False
def update_lock(chat_id, lock_type, locked): with PERM_LOCK: curr_perm = SESSION.query(Permissions).get(str(chat_id)) if not curr_perm: curr_perm = init_permissions(chat_id) if lock_type == "audio": curr_perm.audio = locked elif lock_type == "voice": curr_perm.voice = locked elif lock_type == "contact": curr_perm.contact = locked elif lock_type == "video": curr_perm.video = locked elif lock_type == "videonote": curr_perm.videonote = locked elif lock_type == "document": curr_perm.document = locked elif lock_type == "photo": curr_perm.photo = locked elif lock_type == "sticker": curr_perm.sticker = locked elif lock_type == "gif": curr_perm.gif = locked elif lock_type == 'url': curr_perm.url = locked elif lock_type == 'bots': curr_perm.bots = locked elif lock_type == 'forward': curr_perm.forward = locked elif lock_type == 'game': curr_perm.game = locked elif lock_type == 'location': curr_perm.location = locked SESSION.add(curr_perm) SESSION.commit()
def warn_user(user_id, chat_id, reason=None): with WARN_INSERTION_LOCK: warned_user = SESSION.query(Warns).get((user_id, str(chat_id))) if not warned_user: warned_user = Warns(user_id, str(chat_id)) warned_user.num_warns += 1 if reason is "": reason = "No reason given." if reason: if warned_user.reasons is None: warned_user.reasons = [reason] else: warned_user.reasons = warned_user.reasons + [reason] # TODO:: double check this Daan: Not really wizardry, it adds a new entry to a list/array which can be done this way, basically append equivalent reasons = warned_user.reasons num = warned_user.num_warns SESSION.add(warned_user) SESSION.commit() return num, reasons
def migrate_chat(old_chat_id, new_chat_id): with WARN_INSERTION_LOCK: chat_notes = SESSION.query(Warns).filter(Warns.chat_id == str(old_chat_id)).all() for note in chat_notes: note.chat_id = str(new_chat_id) SESSION.commit() with WARN_FILTER_INSERTION_LOCK: chat_filters = SESSION.query(WarnFilters).filter(WarnFilters.chat_id == str(old_chat_id)).all() for filt in chat_filters: filt.chat_id = str(new_chat_id) SESSION.commit() WARN_FILTERS[str(new_chat_id)] = WARN_FILTERS[str(old_chat_id)] del WARN_FILTERS[str(old_chat_id)] with WARN_SETTINGS_LOCK: chat_settings = SESSION.query(WarnSettings).filter(WarnSettings.chat_id == str(old_chat_id)).all() for setting in chat_settings: setting.chat_id = str(new_chat_id) SESSION.commit()
def del_fed(fed_id): with FEDS_LOCK: global FEDERATION_BYOWNER, FEDERATION_BYFEDID, FEDERATION_BYNAME, FEDERATION_CHATS, FEDERATION_CHATS_BYID, FEDERATION_BANNED_USERID, FEDERATION_BANNED_FULL getcache = FEDERATION_BYFEDID.get(fed_id) if getcache == None: return False # Variables getfed = FEDERATION_BYFEDID.get(fed_id) owner_id = getfed['owner'] fed_name = getfed['fname'] # Delete from cache FEDERATION_BYOWNER.pop(owner_id) FEDERATION_BYFEDID.pop(fed_id) FEDERATION_BYNAME.pop(fed_name) if FEDERATION_CHATS_BYID.get(fed_id): for x in FEDERATION_CHATS_BYID[fed_id]: delchats = SESSION.query(ChatF).get(str(x)) if delchats: SESSION.delete(delchats) SESSION.commit() FEDERATION_CHATS.pop(x) FEDERATION_CHATS_BYID.pop(fed_id) # Delete fedban users getall = FEDERATION_BANNED_USERID.get(fed_id) if getall: for x in getall: banlist = SESSION.query(BansF).get((fed_id, str(x))) if banlist: SESSION.delete(banlist) SESSION.commit() FEDERATION_BANNED_USERID.pop(fed_id) FEDERATION_BANNED_FULL.pop(fed_id) # Delete from database curr = SESSION.query(Federations).get(fed_id) if curr: SESSION.delete(curr) SESSION.commit() return True
def ensure_bot_in_db(): with INSERTION_LOCK: bot = Users(BotID, BotUsername) SESSION.merge(bot) SESSION.commit()
def ensure_bot_in_db(): with INSERTION_LOCK: bot = Users(dispatcher.bot.id, dispatcher.bot.username) SESSION.merge(bot) SESSION.commit()
def add_note_button_to_db(chat_id, note_name, b_name, url, same_line): with BUTTONS_INSERTION_LOCK: button = Buttons(chat_id, note_name, b_name, url, same_line) SESSION.add(button) SESSION.commit()
def add_url(tg_chat_id, tg_feed_link, tg_old_entry_link): with INSERTION_LOCK: action = RSS(tg_chat_id, tg_feed_link, tg_old_entry_link) SESSION.add(action) SESSION.commit()