def get_chat_filters(chat_id):
try:
return SESSION.query(CustomFilters).filter(
CustomFilters.chat_id == str(chat_id)).order_by(
func.length(CustomFilters.keyword).desc()).order_by(
CustomFilters.keyword.asc()).all()
finally:
SESSION.close()
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 user_should_report(user_id: int) -> bool:
try:
user_setting = SESSION.query(ReportingUserSettings).get(user_id)
if user_setting:
return user_setting.should_report
return True
finally:
SESSION.close()
def get_note(chat_id, note_name):
try:
return SESSION.query(Notes).filter(
func.lower(Notes.name) == note_name,
Notes.chat_id == str(chat_id)
).first()
finally:
SESSION.close()
def chat_should_report(chat_id: Union[str, int]) -> bool:
try:
chat_setting = SESSION.query(ReportingChatSettings).get(str(chat_id))
if chat_setting:
return chat_setting.should_report
return True
finally:
SESSION.close()
def get_rules(chat_id):
rules = SESSION.query(Rules).get(str(chat_id))
ret = ""
if rules:
ret = rules.rules
SESSION.close()
return ret
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_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 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_welc_mutes_pref(chat_id):
welcomemutes = SESSION.query(WelcomeMute).get(str(chat_id))
SESSION.close()
if welcomemutes:
return welcomemutes.welcomemutes
return False
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_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 __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 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 rem_chat(chat_id):
with INSERTION_LOCK:
chat = SESSION.query(Chats).get(str(chat_id))
if chat:
SESSION.delete(chat)
SESSION.commit()
else:
SESSION.close()
def num_blacklist_chat_filters(chat_id):
try:
return (
SESSION.query(BlackListFilters.chat_id)
.filter(BlackListFilters.chat_id == str(chat_id))
.count()
)
finally:
SESSION.close()
def get_human_checks(user_id, chat_id):
try:
human_check = SESSION.query(WelcomeMuteUsers).get((user_id, str(chat_id)))
if not human_check:
return None
human_check = human_check.human_check
return human_check
finally:
SESSION.close()
def migrate_chat(old_chat_id, new_chat_id):
with INSERTION_FLOOD_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 num_stickers_filters():
try:
return SESSION.query(StickersFilters).count()
finally:
SESSION.close()
def get_userid_by_name(username):
try:
return (
SESSION.query(Users)
.filter(func.lower(Users.username) == username.lower())
.all()
)
finally:
SESSION.close()
def user_demote_fed(fed_id, user_id):
with FEDS_LOCK:
global FEDERATION_BYOWNER, FEDERATION_BYFEDID, FEDERATION_BYNAME
# Variables
getfed = FEDERATION_BYFEDID.get(str(fed_id))
owner_id = getfed["owner"]
fed_name = getfed["fname"]
fed_rules = getfed["frules"]
fed_log = getfed["flog"]
# Temp set
try:
members = eval(eval(getfed["fusers"])["members"])
except ValueError:
return False
members.remove(user_id)
# Set user
FEDERATION_BYOWNER[str(owner_id)]["fusers"] = str({
"owner":
str(owner_id),
"members":
str(members)
})
FEDERATION_BYFEDID[str(fed_id)]["fusers"] = str({
"owner": str(owner_id),
"members": str(members)
})
FEDERATION_BYNAME[fed_name]["fusers"] = str({
"owner": str(owner_id),
"members": str(members)
})
# Set on database
fed = Federations(
str(owner_id),
fed_name,
str(fed_id),
fed_rules,
fed_log,
str({
"owner": str(owner_id),
"members": str(members)
}),
)
SESSION.merge(fed)
SESSION.commit()
return True
curr = SESSION.query(UserF).all()
result = False
for r in curr:
if int(r.user_id) == int(user_id):
if r.fed_id == fed_id:
SESSION.delete(r)
SESSION.commit()
result = True
SESSION.close()
return result
def num_warn_chat_filters(chat_id):
try:
return (
SESSION.query(WarnFilters.chat_id)
.filter(WarnFilters.chat_id == str(chat_id))
.count()
)
finally:
SESSION.close()
def __load_gbanned_userid_list():
global GBANNED_LIST
try:
GBANNED_LIST = {
x.user_id
for x in SESSION.query(GloballyBannedUsers).all()
}
finally:
SESSION.close()
def __load_gban_stat_list():
global GBANSTAT_LIST
try:
GBANSTAT_LIST = {
x.chat_id
for x in SESSION.query(GbanSettings).all() if not x.setting
}
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 disconnect(user_id):
with CONNECTION_INSERTION_LOCK:
disconnect = SESSION.query(Connection).get((int(user_id)))
if disconnect:
SESSION.delete(disconnect)
SESSION.commit()
return True
else:
SESSION.close()
return False
def get_blacklist_setting(chat_id):
try:
setting = CHAT_SETTINGS_BLACKLISTS.get(str(chat_id))
if setting:
return setting['blacklist_type'], setting['value']
else:
return 1, "0"
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_flood_setting(chat_id):
try:
setting = SESSION.query(FloodSettings).get(str(chat_id))
if setting:
return setting.flood_type, setting.value
else:
return 1, "0"
finally:
SESSION.close()