def kread(chat): if SESSION.query(KRead).get((str(chat))): return False adder = KRead(str(chat)) SESSION.add(adder) SESSION.commit() return True
def set(user_id, credentials): saved_creds = SESSION.query(GDriveCreds).get(user_id) if not saved_creds: saved_creds = GDriveCreds(user_id) saved_creds.credentials_string = dumps(credentials) SESSION.add(saved_creds) SESSION.commit()
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 add_s( user_id, chat_id, session_id, session_expires ): adder = SESSION.query(LydiaAI).get((user_id, chat_id)) if adder: adder.session_id = session_id adder.session_expires = session_expires else: adder = LydiaAI( user_id, chat_id, session_id, session_expires ) SESSION.add(adder) SESSION.commit()
def approve(chat_id): adder = PMPermit(str(chat_id)) SESSION.add(adder) SESSION.commit()
def gban(sender): adder = GBan(str(sender)) SESSION.add(adder) SESSION.commit()
def mute(chat_id, sender): adder = Mute(str(chat_id), str(sender)) SESSION.add(adder) SESSION.commit()
def gmute(sender): adder = GMute(str(sender)) SESSION.add(adder) SESSION.commit()