def add_note_to_db(chat_id, note_name, note_message_id):
with NOTES_INSERTION_LOCK:
prev = SESSION.query(Notes).get((str(chat_id), note_name))
if prev:
SESSION.delete(prev)
note = Notes(str(chat_id), note_name, note_message_id)
SESSION.add(note)
SESSION.commit()
def migrate_chat(old_chat_id, new_chat_id):
with INSERTION_LOCK:
flood = SESSION.query(FloodControl).get(str(old_chat_id))
if flood:
CHAT_FLOOD[str(new_chat_id)] = CHAT_FLOOD.get(str(old_chat_id), DEF_OBJ)
flood.chat_id = str(new_chat_id)
SESSION.commit()
SESSION.close()
def set_credential(chat_id, credential_string):
with INSERTION_LOCK:
saved_cred = SESSION.query(gDriveCreds).get(chat_id)
if not saved_cred:
saved_cred = gDriveCreds(chat_id)
saved_cred.credential_string = pickle.dumps(credential_string)
SESSION.add(saved_cred)
SESSION.commit()
def rm_note(chat_id, note_name):
with NOTES_INSERTION_LOCK:
note = SESSION.query(Notes).get((str(chat_id), note_name))
if note:
SESSION.delete(note)
SESSION.commit()
return True
else:
SESSION.close()
return False
def add_welcome_setting(chat_id, should_clean_welcome, previous_welcome, f_mesg_id):
adder = SESSION.query(Welcome).get(chat_id)
if adder:
adder.should_clean_welcome = should_clean_welcome
adder.previous_welcome = previous_welcome
adder.f_mesg_id = f_mesg_id
else:
adder = Welcome(chat_id, should_clean_welcome, previous_welcome, f_mesg_id)
SESSION.add(adder)
SESSION.commit()
def set_flood(chat_id, amount):
with INSERTION_LOCK:
flood = SESSION.query(FloodControl).get(str(chat_id))
if not flood:
flood = FloodControl(str(chat_id))
flood.user_id = None
flood.limit = amount
CHAT_FLOOD[str(chat_id)] = (None, DEF_COUNT, amount)
SESSION.add(flood)
SESSION.commit()
def clear_credential(chat_id):
with INSERTION_LOCK:
saved_cred = SESSION.query(gDriveCreds).get(chat_id)
if saved_cred:
SESSION.delete(saved_cred)
SESSION.commit()
def update_previous_welcome(chat_id, previous_welcome):
row = SESSION.query(Welcome).get(chat_id)
row.previous_welcome = previous_welcome
# commit the changes to the DB
SESSION.commit()
def rm_welcome_setting(chat_id):
rem = SESSION.query(Welcome).get(chat_id)
if rem:
SESSION.delete(rem)
SESSION.commit()