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 update_lock(chat_id, lock_type, locked):
with PERM_LOCK:
curr_perm = SESSION.query(Permissions).get(str(chat_id))
if not curr_perm:
curr_perm = init_permissions(chat_id)
if lock_type == "audio":
curr_perm.audio = locked
elif lock_type == "voice":
curr_perm.voice = locked
elif lock_type == "contact":
curr_perm.contact = locked
elif lock_type == "video":
curr_perm.video = locked
elif lock_type == "document":
curr_perm.document = locked
elif lock_type == "photo":
curr_perm.photo = locked
elif lock_type == "sticker":
curr_perm.sticker = locked
elif lock_type == "gif":
curr_perm.gif = locked
elif lock_type == 'url':
curr_perm.url = locked
elif lock_type == 'bots':
curr_perm.bots = locked
elif lock_type == 'forward':
curr_perm.forward = locked
elif lock_type == 'game':
curr_perm.game = locked
SESSION.add(curr_perm)
SESSION.commit()
def __load_log_channels():
global CHANNELS
try:
all_chats = SESSION.query(GroupLogs).all()
CHANNELS = {chat.chat_id: chat.log_channel for chat in all_chats}
finally:
SESSION.close()
def migrate_chat(old_chat_id, new_chat_id):
with CHAT_LOCK:
chat_notes = SESSION.query(CommandReactionChatSettings).filter(
CommandReactionChatSettings.chat_id == str(old_chat_id)).all()
for note in chat_notes:
note.chat_id = str(new_chat_id)
SESSION.commit()
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 is_locked(chat_id, lock_type):
curr_perm = SESSION.query(Permissions).get(str(chat_id))
SESSION.close()
if not curr_perm:
return False
elif lock_type == "sticker":
return curr_perm.sticker
elif lock_type == "photo":
return curr_perm.photo
elif lock_type == "audio":
return curr_perm.audio
elif lock_type == "voice":
return curr_perm.voice
elif lock_type == "contact":
return curr_perm.contact
elif lock_type == "video":
return curr_perm.video
elif lock_type == "document":
return curr_perm.document
elif lock_type == "gif":
return curr_perm.gif
elif lock_type == "url":
return curr_perm.url
elif lock_type == "bots":
return curr_perm.bots
elif lock_type == "forward":
return curr_perm.forward
elif lock_type == "game":
return curr_perm.game
def add_to_blacklist(chat_id, trigger):
with BLACKLIST_FILTER_INSERTION_LOCK:
blacklist_filt = BlackListFilters(str(chat_id), trigger)
SESSION.merge(blacklist_filt) # merge to avoid duplicate key issues
SESSION.commit()
CHAT_BLACKLISTS.setdefault(str(chat_id), set()).add(trigger)
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 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 get_welc_mutes_pref(chat_id):
welcomemutes = SESSION.query(WelcomeMute).get(str(chat_id))
SESSION.close()
if welcomemutes:
return welcomemutes.welcomemutes
return False
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_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 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 welcome_mutes(chat_id):
try:
welcomemutes = SESSION.query(WelcomeMute).get(str(chat_id))
if welcomemutes:
return welcomemutes.welcomemutes
return False
finally:
SESSION.close()
def __load_all_feds_settings(): global FEDERATION_NOTIFICATION try: getuser = SESSION.query(FedsUserSettings).all() for x in getuser: FEDERATION_NOTIFICATION[str(x.user_id)] = x.should_report finally: SESSION.close()
def clean_service(chat_id: Union[str, int]) -> bool:
try:
chat_setting = SESSION.query(CleanServiceSetting).get(str(chat_id))
if chat_setting:
return chat_setting.clean_service
return False
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 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 migrate_chat(old_chat_id, new_chat_id):
with GBAN_SETTING_LOCK:
chat = SESSION.query(GbanSettings).get(str(old_chat_id))
if chat:
chat.chat_id = new_chat_id
SESSION.add(chat)
SESSION.commit()
def get_rules(chat_id):
rules = SESSION.query(Rules).get(str(chat_id))
ret = ""
if rules:
ret = rules.rules
SESSION.close()
return ret
def command_reaction(chat_id: Union[str, int]) -> bool:
try:
chat_setting = SESSION.query(CommandReactionChatSettings).get(
str(chat_id))
if chat_setting:
return chat_setting.comm_reaction
return False
finally:
SESSION.close()
def __load_disabled_commands():
global DISABLED
try:
all_chats = SESSION.query(Disable).all()
for chat in all_chats:
DISABLED.setdefault(chat.chat_id, set()).add(chat.command)
finally:
SESSION.close()
def set_rules(chat_id, rules_text):
with INSERTION_LOCK:
rules = SESSION.query(Rules).get(str(chat_id))
if not rules:
rules = Rules(str(chat_id))
rules.rules = rules_text
SESSION.add(rules)
SESSION.commit()
def set_user_me_info(user_id, info):
with INSERTION_LOCK:
userinfo = SESSION.query(UserInfo).get(user_id)
if userinfo:
userinfo.info = info
else:
userinfo = UserInfo(user_id, info)
SESSION.add(userinfo)
SESSION.commit()
def set_clean_service(chat_id: Union[int, str], setting: bool):
with CS_LOCK:
chat_setting = SESSION.query(CleanServiceSetting).get(str(chat_id))
if not chat_setting:
chat_setting = CleanServiceSetting(chat_id)
chat_setting.clean_service = setting
SESSION.add(chat_setting)
SESSION.commit()
def allow_connect_to_chat(chat_id: Union[str, int]) -> bool:
try:
chat_setting = SESSION.query(ChatAccessConnectionSettings).get(
str(chat_id))
if chat_setting:
return chat_setting.allow_connect_to_chat
return False
finally:
SESSION.close()
def __load_flood_settings():
global CHAT_FLOOD
try:
all_chats = SESSION.query(FloodControl).all()
CHAT_FLOOD = {
chat.chat_id: (None, DEF_COUNT, chat.limit)
for chat in all_chats
}
finally:
SESSION.close()
def get_warn_mode(chat_id):
try:
setting = SESSION.query(WarnSettings).get(str(chat_id))
if setting:
return setting.warn_mode, setting.warn_mode
else:
return 3, False
finally:
SESSION.close()
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_warn_mode(chat_id, warn_mode):
with WARN_SETTINGS_LOCK:
curr_setting = SESSION.query(WarnSettings).get(str(chat_id))
if not curr_setting:
curr_setting = WarnSettings(chat_id, warn_mode=warn_mode)
curr_setting.warn_mode = warn_mode
SESSION.add(curr_setting)
SESSION.commit()
