def migrate_chat(old_chat_id, new_chat_id):
with INSERTION_LOCK:
old_chat = SESSION.query(Chats).get(str(old_chat_id))
if old_chat:
old_chat.chat_id = str(new_chat_id)
SESSION.merge(old_chat)
SESSION.commit()
SESSION.close()
def add_to_blacklist(chat_id, trigger):
with INSERTION_LOCK:
try:
blacklist_filt = BlackListFilters(str(chat_id), trigger)
SESSION.merge(blacklist_filt)
SESSION.commit()
CHAT_BLACKLISTS.setdefault(str(chat_id), set()).add(trigger)
finally:
SESSION.close()
def gban_user(user_id, name, reason=None):
with GBAN_LOCK:
user = SESSION.query(GloballyBannedUsers).get(user_id)
try:
if not user:
user = GloballyBannedUsers(user_id, name, reason)
else:
user.reason = reason
user.name = name
SESSION.merge(user)
SESSION.commit()
finally:
SESSION.close()
__load_gbanned_userid_list()
return
def update_gban_reason(user_id, name, reason=None):
with GBAN_LOCK:
try:
user = SESSION.query(GloballyBannedUsers).get(user_id)
if not user:
user = GloballyBannedUsers(user_id, name, reason)
old_reason = ""
else:
old_reason = user.reason
user.name = name
user.reason = reason
SESSION.merge(user)
SESSION.commit()
finally:
SESSION.close()
return old_reason
