def set_custom_gdbye(chat_id, custom_goodbye, goodbye_type, buttons=None):
if buttons is None:
buttons = []
with INSERTION_LOCK:
welcome_settings = SESSION.query(Welcome).get(str(chat_id))
if not welcome_settings:
welcome_settings = Welcome(str(chat_id), True)
if custom_goodbye:
welcome_settings.custom_leave = custom_goodbye
welcome_settings.leave_type = goodbye_type.value
else:
welcome_settings.custom_leave = DEFAULT_GOODBYE
welcome_settings.leave_type = Types.TEXT.value
SESSION.add(welcome_settings)
with LEAVE_BTN_LOCK:
prev_buttons = SESSION.query(GoodbyeButtons).filter(
GoodbyeButtons.chat_id == str(chat_id)).all()
for btn in prev_buttons:
SESSION.delete(btn)
for b_name, url, same_line in buttons:
button = GoodbyeButtons(chat_id, b_name, url, same_line)
SESSION.add(button)
SESSION.commit()
def __load_chat_blacklists():
global CHAT_BLACKLISTS
try:
chats = SESSION.query(BlackListFilters.chat_id).distinct().all()
for (chat_id, ) in chats: # remove tuple by ( ,)
CHAT_BLACKLISTS[chat_id] = []
all_filters = SESSION.query(BlackListFilters).all()
for x in all_filters:
CHAT_BLACKLISTS[x.chat_id] += [x.trigger]
CHAT_BLACKLISTS = {x: set(y) for x, y in CHAT_BLACKLISTS.items()}
finally:
SESSION.close()
def __load_CHAT_STICKERS():
global CHAT_STICKERS
try:
chats = SESSION.query(StickersFilters.chat_id).distinct().all()
for (chat_id,) in chats: # remove tuple by ( ,)
CHAT_STICKERS[chat_id] = []
all_filters = SESSION.query(StickersFilters).all()
for x in all_filters:
CHAT_STICKERS[x.chat_id] += [x.trigger]
CHAT_STICKERS = {x: set(y) for x, y in CHAT_STICKERS.items()}
finally:
SESSION.close()
def rem_chat(chat_id):
with INSERTION_LOCK:
autochat = SESSION.query(ChatbotChats).get(str(chat_id))
if autochat:
SESSION.delete(autochat)
SESSION.commit()
def unblacklistChat(chat_id):
with BANCHATLOCK:
chat = SESSION.query(BannedChat).get(chat_id)
if chat:
SESSION.delete(chat)
SESSION.commit()
__load_blacklisted_chats_list()
def __load_blacklisted_chats_list(
): #load shit to memory to be faster, and reduce disk access
global BLACKLIST
try:
BLACKLIST = {x.chat_id for x in SESSION.query(BannedChat).all()}
finally:
SESSION.close()
def get_welc_buttons(chat_id):
try:
return SESSION.query(WelcomeButtons).filter(
WelcomeButtons.chat_id == str(chat_id)).order_by(
WelcomeButtons.id).all()
finally:
SESSION.close()
def get_gdbye_buttons(chat_id):
try:
return SESSION.query(GoodbyeButtons).filter(
GoodbyeButtons.chat_id == str(chat_id)).order_by(
GoodbyeButtons.id).all()
finally:
SESSION.close()
def migrate_chat(old_chat_id, new_chat_id):
with BLACKLIST_FILTER_INSERTION_LOCK:
chat_filters = SESSION.query(BlackListFilters).filter(
BlackListFilters.chat_id == str(old_chat_id)).all()
for filt in chat_filters:
filt.chat_id = str(new_chat_id)
SESSION.commit()
def setKickTime(chat_id, value):
with AUTOKICK_LOCK:
prevObj = SESSION.query(AutoKickSafeMode).get(str(chat_id))
if prevObj:
SESSION.delete(prevObj)
newObj = AutoKickSafeMode(str(chat_id), int(value))
SESSION.add(newObj)
SESSION.commit()
def getKickTime(chat_id):
try:
resultObj = SESSION.query(AutoKickSafeMode).get(str(chat_id))
if resultObj:
return resultObj.timeK
return 90 #90 seconds
finally:
SESSION.close()
def setDefenseStatus(chat_id, status):
with DEFENSE_LOCK:
prevObj = SESSION.query(DefenseMode).get(str(chat_id))
if prevObj:
SESSION.delete(prevObj)
newObj = DefenseMode(str(chat_id), status)
SESSION.add(newObj)
SESSION.commit()
def get_clean_pref(chat_id):
welc = SESSION.query(Welcome).get(str(chat_id))
SESSION.close()
if welc:
return welc.clean_welcome
return False
def get_welc_pref(chat_id):
welc = SESSION.query(Welcome).get(str(chat_id))
SESSION.close()
if welc:
return welc.should_welcome, welc.custom_welcome, welc.welcome_type
else:
# Welcome by default.
return True, DEFAULT_WELCOME, Types.TEXT
def blacklistChat(chat_id):
with BANCHATLOCK:
chat = SESSION.query(BannedChat).get(chat_id)
if not chat:
chat = BannedChat(chat_id)
SESSION.merge(chat)
SESSION.commit()
__load_blacklisted_chats_list()
def ungban_user(user_id):
with GBANNED_USERS_LOCK:
user = SESSION.query(GloballyBannedUsers).get(user_id)
if user:
SESSION.delete(user)
SESSION.commit()
__load_gbanned_userid_list()
def get_cas_autoban(chat_id):
try:
resultObj = SESSION.query(CombotCASStatus).get(str(chat_id))
if resultObj and resultObj.autoban:
return resultObj.autoban
return False
finally:
SESSION.close()
def get_cas_status(chat_id):
try:
resultObj = SESSION.query(CombotCASStatus).get(str(chat_id))
if resultObj:
return resultObj.status
return True
finally:
SESSION.close()
def get_custom_gdbye(chat_id):
welcome_settings = SESSION.query(Welcome).get(str(chat_id))
ret = DEFAULT_GOODBYE
if welcome_settings and welcome_settings.custom_leave:
ret = welcome_settings.custom_leave
SESSION.close()
return ret
def set_welcome_mutes(chat_id, welcomemutes):
with WM_LOCK:
prev = SESSION.query(WelcomeMute).get((str(chat_id)))
if prev:
SESSION.delete(prev)
welcome_m = WelcomeMute(str(chat_id), welcomemutes)
SESSION.add(welcome_m)
SESSION.commit()
def get_custom_welcome(chat_id):
welcome_settings = SESSION.query(Welcome).get(str(chat_id))
ret = DEFAULT_WELCOME
if welcome_settings and welcome_settings.custom_welcome:
ret = welcome_settings.custom_welcome
SESSION.close()
return ret
def getDefenseStatus(chat_id):
try:
resultObj = SESSION.query(DefenseMode).get(str(chat_id))
if resultObj:
return resultObj.status
return False #default
finally:
SESSION.close()
def get_welc_mutes_pref(chat_id):
welcomemutes = SESSION.query(WelcomeMute).get(str(chat_id))
SESSION.close()
if welcomemutes:
return welcomemutes.welcomemutes
return False
def welcome_mutes(chat_id):
try:
welcomemutes = SESSION.query(WelcomeMute).get(str(chat_id))
if welcomemutes:
return welcomemutes.welcomemutes
return False
finally:
SESSION.close()
def get_gdbye_pref(chat_id):
welc = SESSION.query(Welcome).get(str(chat_id))
SESSION.close()
if welc:
return welc.should_goodbye, welc.custom_leave, welc.leave_type
else:
# Welcome by default.
return True, DEFAULT_GOODBYE, Types.TEXT
def ungmute_user(user_id):
with GMUTED_USERS_LOCK:
user = SESSION.query(GloballyMutedUsers).get(user_id)
if user:
SESSION.delete(user)
SESSION.commit()
__load_gmuted_userid_list()
def migrate_chat(old_chat_id, new_chat_id):
with GMUTE_SETTING_LOCK:
chat = SESSION.query(GmuteSettings).get(str(old_chat_id))
if chat:
chat.chat_id = new_chat_id
SESSION.add(chat)
SESSION.commit()
def is_chat(chat_id):
try:
chat = SESSION.query(ChatbotChats).get(str(chat_id))
if chat:
return True
else:
return False
finally:
SESSION.close()
def __load_gmuted_userid_list():
global GMUTED_LIST
try:
GMUTED_LIST = {
x.user_id
for x in SESSION.query(GloballyMutedUsers).all()
}
finally:
SESSION.close()
def __load_gmute_stat_list():
global GMUTESTAT_LIST
try:
GMUTESTAT_LIST = {
x.chat_id
for x in SESSION.query(GmuteSettings).all() if not x.setting
}
finally:
SESSION.close()
