def unmute(chat_id, sender): rem = (SESSION.query(Mute).filter(Mute.chat_id == str(chat_id), Mute.sender == str(sender)).all()) if len(rem): for item in rem: SESSION.delete(item) SESSION.commit()
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 unkread(chat): rem = SESSION.query(KRead).get((str(chat))) if rem: SESSION.delete(rem) SESSION.commit() return True return False
def kread(chat): if SESSION.query(KRead).get((str(chat))): return False adder = KRead(str(chat)) SESSION.add(adder) SESSION.commit() return True
def rm_note(chat_id, keyword): to_check = get_note(chat_id, keyword) if not to_check: return False else: SESSION.delete(to_check) SESSION.commit() return True
def remove_snip(keyword): to_check = get_snip(keyword) if not to_check: return False else: SESSION.delete(to_check) SESSION.commit() return True
def rm_from_blacklist(chat_id, trigger): with BLACKLIST_FILTER_INSERTION_LOCK: blacklist_filt = SESSION.query(BlackListFilters).get( (str(chat_id), trigger)) if blacklist_filt: # sanity check if trigger in CHAT_BLACKLISTS.get(str(chat_id), set()): CHAT_BLACKLISTS.get(str(chat_id), set()).remove(trigger) SESSION.delete(blacklist_filt) SESSION.commit() return True SESSION.close() return False
def add_note(chat_id, keyword, reply, f_mesg_id): to_check = get_note(chat_id, keyword) if not to_check: adder = Notes(str(chat_id), keyword, reply, f_mesg_id) SESSION.add(adder) SESSION.commit() return True else: rem = SESSION.query(Notes).get((str(chat_id), keyword)) SESSION.delete(rem) SESSION.commit() adder = Notes(str(chat_id), keyword, reply, f_mesg_id) SESSION.add(adder) SESSION.commit() return False
def add_snip(keyword, reply, f_mesg_id): to_check = get_snip(keyword) if not to_check: adder = Snips(keyword, reply, f_mesg_id) SESSION.add(adder) SESSION.commit() return True else: rem = SESSION.query(Snips).filter(Snips.snip == keyword) rem.delete() SESSION.commit() adder = Snips(keyword, reply, f_mesg_id) SESSION.add(adder) SESSION.commit() return False
def dissprove(chat_id): rem = SESSION.query(PMPermit).get(str(chat_id)) if rem: SESSION.delete(rem) SESSION.commit()
def approve(chat_id): adder = PMPermit(str(chat_id)) SESSION.add(adder) SESSION.commit()
def mute(chat_id, sender): adder = Mute(str(chat_id), str(sender)) SESSION.add(adder) SESSION.commit()
def ungmute(sender): rem = SESSION.query(GMute).get((str(sender))) if rem: SESSION.delete(rem) SESSION.commit()
def gmute(sender): adder = GMute(str(sender)) SESSION.add(adder) SESSION.commit()
def ungban(sender): rem = SESSION.query(GBan).get((str(sender))) if rem: SESSION.delete(rem) SESSION.commit()
def gban(sender): adder = GBan(str(sender)) SESSION.add(adder) SESSION.commit()