def backup_indentity(first_name, last_name, bio):
cloner_db = SESSION.query(Cloner).get(str(Owner))
if cloner_db:
SESSION.delete(cloner_db)
cloner_db = Cloner(Owner, first_name, last_name, bio)
SESSION.add(cloner_db)
SESSION.commit()
def set_stanim_set(my_id, sticker):
sticker_db = SESSION.query(StickerAnimationSet).get(my_id)
if sticker_db:
SESSION.delete(sticker_db)
sticker_db = StickerAnimationSet(my_id, sticker)
SESSION.add(sticker_db)
SESSION.commit()
def switch_to_locale(chat_id, locale_name):
with LOCALES_INSERTION_LOCK:
prev = SESSION.query(Locales).get(str(chat_id))
if prev:
SESSION.delete(prev)
switch_locale = Locales(str(chat_id), locale_name)
SESSION.add(switch_locale)
SESSION.commit()
def set_afk(afk, reason):
global MY_AFK
afk_db = SESSION.query(AFK).get(str(Owner))
if afk_db:
SESSION.delete(afk_db)
afk_db = AFK(Owner, afk, reason)
SESSION.add(afk_db)
SESSION.commit()
MY_AFK[Owner] = {'afk': afk, 'reason': reason}
def remove_fav(user_id):
with FAV_INSERTION_LOCK:
to_check = get_fav(user_id)
if not to_check:
return False
rem = SESSION.query(Favourites).filter(Favourites.user_id == user_id)
rem.delete()
SESSION.commit()
return True
def add_fav(user_id, data):
with FAV_INSERTION_LOCK:
to_check = check_fav(user_id, data)
if not to_check:
adder = Favourites(int(user_id), str(data))
SESSION.add(adder)
SESSION.commit()
return True
return False
def set_req(user_id, username):
with INSERTION_LOCK:
user = SESSION.query(ReqUsers).get(str(user_id))
if not user:
user = ReqUsers(str(user_id), str(username))
else:
user.username = str(username)
SESSION.add(user)
SESSION.commit()
def del_whitelist(user_id):
with INSERTION_LOCK:
user = SESSION.query(WhitelistUsers).get(str(user_id))
if user:
SESSION.delete(user)
SESSION.commit()
else:
SESSION.close()
return False
def set_whitelist(user_id, username):
with INSERTION_LOCK:
user = SESSION.query(WhitelistUsers).get(str(user_id))
if not user:
user = WhitelistUsers(str(user_id), str(username))
else:
user.username = str(username)
SESSION.add(user)
SESSION.commit()
def update_chat(chat):
global MY_ALL_CHATS
if (chat.id in list(MY_ALL_CHATS) and MY_ALL_CHATS.get(chat.id)
and MY_ALL_CHATS[chat.id].get('name') == chat.title
and MY_ALL_CHATS[chat.id].get('username') == chat.username):
return
chat_db = SESSION.query(MyChats).get(str(chat.id))
if chat_db:
SESSION.delete(chat_db)
chat_db = MyChats(str(chat.id), chat.title, chat.username)
SESSION.add(chat_db)
SESSION.commit()
MY_ALL_CHATS[chat.id] = {'name': chat.title, 'username': chat.username}
def rm_selfnote(user_id, note_name):
global SELF_NOTES
with INSERTION_LOCK:
note = SESSION.query(SelfNotes).get((user_id, note_name))
if note:
SESSION.delete(note)
SESSION.commit()
SELF_NOTES[user_id].pop(note_name)
return True
else:
SESSION.close()
return False
def save_selfnote(user_id, note_name, note_data, msgtype, file=None):
global SELF_NOTES
with INSERTION_LOCK:
prev = SESSION.query(SelfNotes).get((user_id, note_name))
if prev:
SESSION.delete(prev)
note = SelfNotes(
user_id,
note_name,
note_data,
msgtype=int(msgtype),
file=file,
)
SESSION.add(note)
SESSION.commit()
if not SELF_NOTES.get(user_id):
SELF_NOTES[user_id] = {}
SELF_NOTES[user_id][note_name] = {
'value': note_data,
'type': msgtype,
'file': file,
}