def __load_chat_filters(): global CHAT_FILTERS try: chats = SESSION.query(CustomFilters.chat_id).distinct().all() for (chat_id,) in chats: # remove tuple by ( ,) CHAT_FILTERS[chat_id] = [] all_filters = SESSION.query(CustomFilters).all() for x in all_filters: CHAT_FILTERS[x.chat_id] += [x.keyword] CHAT_FILTERS = { x: sorted(set(y), key=lambda i: (-len(i), i)) for x, y in CHAT_FILTERS.items() } finally: SESSION.close()
def is_restr_locked(chat_id, lock_type): curr_restr = SESSION.query(Restrictions).get(str(chat_id)) SESSION.close() if not curr_restr: return False if lock_type == "messages": return curr_restr.messages elif lock_type == "media": return curr_restr.media elif lock_type == "other": return curr_restr.other elif lock_type == "previews": return curr_restr.preview elif lock_type == "all": return (curr_restr.messages and curr_restr.media and curr_restr.other and curr_restr.preview)
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 is 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 rename_fed(fed_id, owner_id, newname): with FEDS_LOCK: global FEDERATION_BYFEDID, FEDERATION_BYOWNER, FEDERATION_BYNAME fed = SESSION.query(Federations).get(fed_id) if not fed: return False fed.fed_name = newname SESSION.commit() # Update the dicts oldname = FEDERATION_BYFEDID[str(fed_id)]["fname"] tempdata = FEDERATION_BYNAME[oldname] FEDERATION_BYNAME.pop(oldname) FEDERATION_BYOWNER[str(owner_id)]["fname"] = newname FEDERATION_BYFEDID[str(fed_id)]["fname"] = newname FEDERATION_BYNAME[newname] = tempdata return True
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"] fed_log = getfed["flog"] # 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, fed_log, str({ "owner": str(owner_id), "members": str(members) }), ) SESSION.merge(fed) SESSION.commit() __load_all_feds_chats() return True
def new_add_filter(chat_id, keyword, reply_text, file_type, file_id, buttons): 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="there is should be a new reply", is_sticker=False, is_document=False, is_image=False, is_audio=False, is_voice=False, is_video=False, has_buttons=bool(buttons), reply_text=reply_text, file_type=file_type.value, file_id=file_id, ) 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 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 set_welcome_mutes(chat_id, welcomemutes): with WM_LOCK: prev = SESSION.query(WelcomeMute).get((str(chat_id))) if prev: SESSION.delete(prev) welcome_m = WelcomeMute(str(chat_id), welcomemutes) SESSION.add(welcome_m) SESSION.commit()
def switch_to_locale(chat_id, locale_name): with LOCALES_INSERTION_LOCK: prev = SESSION.query(Locales).get((str(chat_id))) if prev: SESSION.delete(prev) switch_locale = Locales(str(chat_id), locale_name) SESSION.add(switch_locale) SESSION.commit()
def un_fban_user(fed_id, user_id): with FEDS_LOCK: r = SESSION.query(BansF).all() for I in r: if I.fed_id == fed_id and int(I.user_id) == int(user_id): SESSION.delete(I) try: SESSION.commit() except: SESSION.rollback() return False finally: SESSION.commit() __load_all_feds_banned() return I
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 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: warned_user.reasons = warned_user.reasons + [ reason, ] # TODO:: double check this wizardry reasons = warned_user.reasons num = warned_user.num_warns SESSION.add(warned_user) SESSION.commit() return num, reasons
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) fed_log = getfed['flog'] # 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, fed_log, str(fed_members)) SESSION.merge(fed) SESSION.commit() return True
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 == "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 elif lock_type == "rtl": curr_perm.rtl = locked elif lock_type == "button": curr_perm.button = locked elif lock_type == "egame": curr_perm.egame = locked elif lock_type == "inline": curr_perm.inline = locked SESSION.add(curr_perm) SESSION.commit()
def __load_all_feds_chats(): global FEDERATION_CHATS, FEDERATION_CHATS_BYID try: qall = SESSION.query(ChatF).all() FEDERATION_CHATS = {} FEDERATION_CHATS_BYID = {} for x in qall: # Federation Chats check = FEDERATION_CHATS.get(x.chat_id) if check == None: FEDERATION_CHATS[x.chat_id] = {} FEDERATION_CHATS[x.chat_id] = {'chat_name': x.chat_name, 'fid': x.fed_id} # Federation Chats By ID check = FEDERATION_CHATS_BYID.get(x.fed_id) if check == None: FEDERATION_CHATS_BYID[x.fed_id] = [] FEDERATION_CHATS_BYID[x.fed_id].append(x.chat_id) finally: SESSION.close()
def __load_all_feds_banned(): global FEDERATION_BANNED_USERID, FEDERATION_BANNED_FULL try: FEDERATION_BANNED_USERID = {} FEDERATION_BANNED_FULL = {} qall = SESSION.query(BansF).all() for x in qall: check = FEDERATION_BANNED_USERID.get(x.fed_id) if check == None: FEDERATION_BANNED_USERID[x.fed_id] = [] if int(x.user_id) not in FEDERATION_BANNED_USERID[x.fed_id]: FEDERATION_BANNED_USERID[x.fed_id].append(int(x.user_id)) check = FEDERATION_BANNED_FULL.get(x.fed_id) if check == None: FEDERATION_BANNED_FULL[x.fed_id] = {} FEDERATION_BANNED_FULL[x.fed_id][x.user_id] = {'first_name': x.first_name, 'last_name': x.last_name, 'user_name': x.user_name, 'reason': x.reason, 'time': x.time} finally: SESSION.close()
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 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 is_locked(chat_id, lock_type): curr_perm = SESSION.query(Permissions).get(str(chat_id)) SESSION.close() if not curr_perm: return False elif lock_type == "sticker": return curr_perm.sticker elif lock_type == "photo": return curr_perm.photo elif lock_type == "audio": return curr_perm.audio elif lock_type == "voice": return curr_perm.voice elif lock_type == "contact": return curr_perm.contact elif lock_type == "video": return curr_perm.video elif lock_type == "document": return curr_perm.document elif lock_type == "gif": return curr_perm.gif elif lock_type == "url": return curr_perm.url elif lock_type == "bots": return curr_perm.bots elif lock_type == "forward": return curr_perm.forward elif lock_type == "game": return curr_perm.game elif lock_type == "location": return curr_perm.location elif lock_type == "rtl": return curr_perm.rtl elif lock_type == "button": return curr_perm.button elif lock_type == "egame": return curr_perm.egame elif lock_type == "inline": return curr_perm.inline elif lock_type == "channelchat": return curr_perm.channelchat
def disconnect(user_id): with CONNECTION_INSERTION_LOCK: disconnect = SESSION.query(Connection).get((int(user_id))) if disconnect: SESSION.delete(disconnect) SESSION.commit() return True SESSION.close() return False
def disapprove(chat_id, user_id): with APPROVE_INSERTION_LOCK: disapprove_user = SESSION.query(Approvals).get((str(chat_id), user_id)) if disapprove_user: SESSION.delete(disapprove_user) SESSION.commit() return True SESSION.close() return False
def connect(user_id, chat_id): with CONNECTION_INSERTION_LOCK: prev = SESSION.query(Connection).get((int(user_id))) if prev: SESSION.delete(prev) connect_to_chat = Connection(int(user_id), chat_id) SESSION.add(connect_to_chat) SESSION.commit() return True
def rm_note(chat_id, note_name): with NOTES_INSERTION_LOCK: note = SESSION.query(Notes).filter( func.lower(Notes.name) == note_name, Notes.chat_id == str(chat_id)).first() 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 reset_warns(user_id, chat_id): with WARN_INSERTION_LOCK: warned_user = SESSION.query(Warns).get((user_id, str(chat_id))) if warned_user: warned_user.num_warns = 0 warned_user.reasons = [] SESSION.add(warned_user) SESSION.commit() SESSION.close()
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_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 disapprove(chat_id, user_id): with APPROVE_INSERTION_LOCK: note = SESSION.query(Approvals).get((str(chat_id), user_id)) if note: SESSION.delete(note) SESSION.commit() return True else: SESSION.close() return False
def set_cleanbt(chat_id, is_enable): with CLEANER_CHAT_SETTINGS: curr = SESSION.query(CleanerBlueTextChatSettings).get(str(chat_id)) if curr: SESSION.delete(curr) newcurr = CleanerBlueTextChatSettings(str(chat_id), is_enable) SESSION.add(newcurr) SESSION.commit()
def set_fed_log(fed_id, chat_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_members = getfed["fusers"] fed_rules = getfed["frules"] fed_log = str(chat_id) # Set user FEDERATION_BYOWNER[str(owner_id)]["flog"] = fed_log FEDERATION_BYFEDID[str(fed_id)]["flog"] = fed_log FEDERATION_BYNAME[fed_name]["flog"] = fed_log # Set on database fed = Federations(str(owner_id), fed_name, str(fed_id), fed_rules, fed_log, str(fed_members)) SESSION.merge(fed) SESSION.commit() print(fed_log) return True
def set_privnotes_setting(chat_id, setting): with PRIVATE_NOTES_LOCK: res = SESSION.query(PrivateNotes).get(str(chat_id)) if not res: res = PrivateNotes(str(chat_id), setting) SESSION.add(res) SESSION.flush() else: res.setting = setting SESSION.commit()