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 __load_chat_warn_filters(): global WARN_FILTERS try: chats = SESSION.query(WarnFilters.chat_id).distinct().all() for (chat_id,) in chats: # remove tuple by ( ,) WARN_FILTERS[chat_id] = [] all_filters = SESSION.query(WarnFilters).all() for x in all_filters: WARN_FILTERS[x.chat_id] += [x.keyword] WARN_FILTERS = { x: sorted(set(y), key=lambda i: (-len(i), i)) for x, y in WARN_FILTERS.items() } finally: SESSION.close()
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 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
def __load_all_feds(): global FEDERATION_BYOWNER, FEDERATION_BYFEDID, FEDERATION_BYNAME try: feds = SESSION.query(Federations).all() for x in feds: # remove tuple by ( ,) # Fed by Owner check = FEDERATION_BYOWNER.get(x.owner_id) if check is None: FEDERATION_BYOWNER[x.owner_id] = [] FEDERATION_BYOWNER[str(x.owner_id)] = { 'fid': str(x.fed_id), 'fname': x.fed_name, 'frules': x.fed_rules, 'flog': x.fed_log, 'fusers': str(x.fed_users) } # Fed By FedId check = FEDERATION_BYFEDID.get(x.fed_id) if check is None: FEDERATION_BYFEDID[x.fed_id] = [] FEDERATION_BYFEDID[str(x.fed_id)] = { 'owner': str(x.owner_id), 'fname': x.fed_name, 'frules': x.fed_rules, 'flog': x.fed_log, 'fusers': str(x.fed_users) } # Fed By Name check = FEDERATION_BYNAME.get(x.fed_name) if check is None: FEDERATION_BYNAME[x.fed_name] = [] FEDERATION_BYNAME[x.fed_name] = { 'fid': str(x.fed_id), 'owner': str(x.owner_id), 'frules': x.fed_rules, 'flog': x.fed_log, 'fusers': str(x.fed_users) } finally: SESSION.close()
def __load_cleaner_list(): global GLOBAL_IGNORE_COMMANDS global CLEANER_CHATS try: GLOBAL_IGNORE_COMMANDS = { int(x.command) for x in SESSION.query(CleanerBlueTextGlobal).all() } finally: SESSION.close() try: for x in SESSION.query(CleanerBlueTextChatSettings).all(): CLEANER_CHATS.setdefault(x.chat_id, { "setting": False, "commands": set() }) CLEANER_CHATS[x.chat_id]["setting"] = x.is_enable finally: SESSION.close() try: for x in SESSION.query(CleanerBlueTextChat).all(): CLEANER_CHATS.setdefault(x.chat_id, { "setting": False, "commands": set() }) CLEANER_CHATS[x.chat_id]["commands"].add(x.command) finally: SESSION.close()
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 is 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 is None: FEDERATION_CHATS_BYID[x.fed_id] = [] FEDERATION_CHATS_BYID[x.fed_id].append(x.chat_id) finally: SESSION.close()
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 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 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 __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 is 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 is 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 num_warns(): try: return SESSION.query(func.sum(Warns.num_warns)).scalar() or 0 finally: SESSION.close()
def num_warn_filters(): try: return SESSION.query(WarnFilters).count() finally: SESSION.close()
def num_users(): try: return SESSION.query(Users).count() finally: SESSION.close()
def get_chat_warn_filters(chat_id): try: return SESSION.query(WarnFilters).filter( WarnFilters.chat_id == str(chat_id)).all() finally: SESSION.close()
def get_locks(chat_id): try: return SESSION.query(Permissions).get(str(chat_id)) finally: SESSION.close()
def num_logchannels(): try: return SESSION.query(func.count(distinct(GroupLogs.chat_id))).scalar() finally: SESSION.close()
def num_filters(): try: return SESSION.query(CustomFilters).count() finally: SESSION.close()
def get_restr(chat_id): try: return SESSION.query(Restrictions).get(str(chat_id)) finally: SESSION.close()
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 num_warn_filter_chats(): try: return SESSION.query(func.count(distinct(WarnFilters.chat_id))).scalar() finally: SESSION.close()
def curr_connection(chat_id): try: return SESSION.query(Connection).get((str(chat_id))) finally: SESSION.close()
def get_all_filters(): try: return SESSION.query(CustomFilters).all() finally: SESSION.close()
def get_connected_chat(user_id): try: return SESSION.query(Connection).get((int(user_id))) finally: SESSION.close()
def get_filter(chat_id, keyword): try: return SESSION.query(CustomFilters).get((str(chat_id), keyword)) finally: SESSION.close()
def get_all_chats(): try: return SESSION.query(ChatbotChats.chat_id).all() finally: SESSION.close()
def check_afk_status(user_id): try: return SESSION.query(AFK).get(user_id) finally: SESSION.close()
def get_all_chat_notes(chat_id): try: return SESSION.query(Notes).filter( Notes.chat_id == str(chat_id)).order_by(Notes.name.asc()).all() finally: SESSION.close()
def num_chats(): try: return SESSION.query(func.count(distinct(Rules.chat_id))).scalar() finally: SESSION.close()
def num_notes(): try: return SESSION.query(Notes).count() finally: SESSION.close()