def is_gmuted(sender): try: return SESSION.query(GMute).all() except: return None finally: SESSION.close()
def add_to_blacklist(chat_id, trigger): with BLACKLIST_FILTER_INSERTION_LOCK: blacklist_filt = BlackListFilters(str(chat_id), trigger) SESSION.merge(blacklist_filt) # merge to avoid duplicate key issues SESSION.commit() CHAT_BLACKLISTS.setdefault(str(chat_id), set()).add(trigger)
def gt_fz(chat_id, keyword): try: return SESSION.query(Filterz).get((str(chat_id), keyword)) except: return None finally: SESSION.close()
def get_all_filters(chat_id): try: return SESSION.query(Filters).filter(Filters.chat_id == str(chat_id)).all() except: return None finally: SESSION.close()
def a_to_blx(chat_id, trigger): with BLACKLIST_FILTER_INSERTION: blacklist_filt = Blfx(str(chat_id), trigger) SESSION.merge(blacklist_filt) # merge to avoid duplicate key issues SESSION.commit() CHAT_BLACKLIST.setdefault(str(chat_id), set()).add(trigger)
def get_all_snips(): try: return SESSION.query(Snips).all() except: return None finally: SESSION.close()
def is_approved(chat_id): try: return SESSION.query(PMPermit).filter(PMPermit.chat_id == str(chat_id)).one() except: return None finally: SESSION.close()
def get_snips(keyword): try: return SESSION.query(Snips).get(keyword) except: return None finally: SESSION.close()
def add_welcome_setting(chat_id, custom_welcome_message, should_clean_welcome, previous_welcome, media_file_id): # adder = SESSION.query(Welcome).get(chat_id) adder = Welcome(chat_id, custom_welcome_message, should_clean_welcome, previous_welcome, media_file_id) SESSION.add(adder) SESSION.commit()
def is_gbanned(sender, pika_id): try: _pikaG = SESSION.query(GBan).get((str(sender), str(pika_id))) if _pikaG: return str(_pikaG.reason) finally: SESSION.close()
def gt_afz(chat_id): try: return SESSION.query(Filterz).filter( Filterz.chat_id == str(chat_id)).all() except: return None finally: SESSION.close()
def add_note2(chat_id, keyword, reply): adder = SESSION.query(Notes2).get((str(chat_id), keyword)) if adder: adder.reply = reply else: adder = Notes2(str(chat_id), keyword, reply) SESSION.add(adder) SESSION.commit()
def get_current_welcome_settings(chat_id): try: return SESSION.query(Welcome).filter( Welcome.chat_id == str(chat_id)).one() except: return None finally: SESSION.close()
def get_notes(chat_id, client_id): try: return SESSION.query(Notes).filter(Notes.chat_id == str(chat_id), Notes.client_id == client_id).all() except: return None finally: SESSION.close()
def rm_note(chat_id, keyword, client_id): to_check = get_note(chat_id, keyword, client_id) if not to_check: return False else: rem = SESSION.query(Notes).get((str(chat_id), keyword, client_id)) SESSION.delete(rem) SESSION.commit() return True
def init_locks(chat_id, reset=False): curr_restr = SESSION.query(Locks).get(str(chat_id)) if reset: SESSION.delete(curr_restr) SESSION.flush() restr = Locks(str(chat_id)) SESSION.add(restr) SESSION.commit() return restr
def add_snip(keyword, reply, snip_type, media_id, media_access_hash, media_file_reference): adder = SESSION.query(Snips).get(keyword) if adder: adder.reply = reply adder.snip_type = snip_type adder.media_id = media_id adder.media_access_hash = media_access_hash adder.media_file_reference = media_file_reference else: adder = Snips(keyword, reply, snip_type, media_id, media_access_hash, media_file_reference) SESSION.add(adder) SESSION.commit()
def add_filter(chat_id, keyword, reply, snip_type, media_id, media_access_hash, media_file_reference): adder = SESSION.query(Filters).get((str(chat_id), keyword)) if adder: adder.reply = reply adder.snip_type = snip_type adder.media_id = media_id adder.media_access_hash = media_access_hash adder.media_file_reference = media_file_reference else: adder = Filters(chat_id, keyword, reply, snip_type, media_id, media_access_hash, media_file_reference) SESSION.add(adder) SESSION.commit()
def is_locked(chat_id, lock_type): curr_perm = SESSION.query(Locks).get(str(chat_id)) SESSION.close() if not curr_perm: return False elif lock_type == "bots": return curr_perm.bots elif lock_type == "commands": return curr_perm.commands elif lock_type == "email": return curr_perm.email elif lock_type == "forward": return curr_perm.forward elif lock_type == "url": return curr_perm.url
def __lc_blx(): global CHAT_BLACKLIST try: chats = SESSION.query(Blfx.chat_id).distinct().all() for (chat_id,) in chats: # remove tuple by ( ,) CHAT_BLACKLIST[chat_id] = [] all_filters = SESSION.query(Blfx).all() for x in all_filters: CHAT_BLACKLIST[x.chat_id] += [x.trigger] CHAT_BLACKLIST = {x: set(y) for x, y in CHAT_BLACKLIST.items()} finally: SESSION.close()
def __load_chat_blacklists(): global CHAT_BLACKLISTS try: chats = SESSION.query(BlackListFilters.chat_id).distinct().all() for (chat_id,) in chats: # remove tuple by ( ,) CHAT_BLACKLISTS[chat_id] = [] all_filters = SESSION.query(BlackListFilters).all() for x in all_filters: CHAT_BLACKLISTS[x.chat_id] += [x.trigger] CHAT_BLACKLISTS = {x: set(y) for x, y in CHAT_BLACKLISTS.items()} finally: SESSION.close()
def update_lock(chat_id, lock_type, locked): curr_perm = SESSION.query(Locks).get(str(chat_id)) if not curr_perm: curr_perm = init_locks(chat_id) if lock_type == "bots": curr_perm.bots = locked elif lock_type == "commands": curr_perm.commands = locked elif lock_type == "email": curr_perm.email = locked elif lock_type == "forward": curr_perm.forward = locked elif lock_type == "url": curr_perm.url = locked SESSION.add(curr_perm) SESSION.commit()
def rf_blx(chat_id, trigger): with BLACKLIST_FILTER_INSERTION: blacklist_filt = SESSION.query(Blfx).get((str(chat_id), trigger)) if blacklist_filt: if trigger in CHAT_BLACKLIST.get(str(chat_id), set()): # sanity check CHAT_BLACKLIST.get(str(chat_id), set()).remove(trigger) SESSION.delete(blacklist_filt) SESSION.commit() return True SESSION.close() return False
def ungmute(sender, pika_id): rem = SESSION.query(GMute).get((str(sender), pika_id)) if rem: SESSION.delete(rem) SESSION.commit()
def gmute(sender, pika_id): adder = GMute(str(sender), pika_id) SESSION.add(adder) SESSION.commit()
def ungban(sender, pika_id): rem = SESSION.query(GBan).get((str(sender), str(pika_id))) if rem: SESSION.delete(rem) SESSION.commit()
def gban(sender, pika_id, reason): adder = GBan(str(sender), str(pika_id), str(reason)) SESSION.add(adder) SESSION.commit()
def n_bl_fcx(): try: return SESSION.query(func.count(distinct(Blfx.chat_id))).scalar() finally: SESSION.close()
def n_bl_cfx(chat_id): try: return SESSION.query(Blfx.chat_id).filter(Blfx.chat_id == str(chat_id)).count() finally: SESSION.close()
def no_blfx(): try: return SESSION.query(Blfx).count() finally: SESSION.close()