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()