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 get_buttons(chat_id, note_name):
try:
return (SESSION.query(Buttons).filter(
Buttons.chat_id == str(chat_id),
Buttons.note_name == note_name).order_by(Buttons.id).all())
finally:
SESSION.close()
def __load_feds_subscriber():
global FEDS_SUBSCRIBER
global MYFEDS_SUBSCRIBER
try:
feds = SESSION.query(FedSubs.fed_id).distinct().all()
for (fed_id, ) in feds: # remove tuple by ( ,)
FEDS_SUBSCRIBER[fed_id] = []
MYFEDS_SUBSCRIBER[fed_id] = []
all_fedsubs = SESSION.query(FedSubs).all()
for x in all_fedsubs:
FEDS_SUBSCRIBER[x.fed_id] += [x.fed_subs]
try:
MYFEDS_SUBSCRIBER[x.fed_subs] += [x.fed_id]
except KeyError:
getsubs = SESSION.query(FedSubs).get((x.fed_id, x.fed_subs))
if getsubs:
SESSION.delete(getsubs)
SESSION.commit()
FEDS_SUBSCRIBER = {x: set(y) for x, y in FEDS_SUBSCRIBER.items()}
MYFEDS_SUBSCRIBER = {x: set(y) for x, y in MYFEDS_SUBSCRIBER.items()}
finally:
SESSION.close()
def is_nightmode_indb(chat_id: str):
try:
s__ = SESSION.query(Nightmode).get(str(chat_id))
if s__:
return str(s__.chat_id)
finally:
SESSION.close()
def __migrate_filters():
try:
all_filters = SESSION.query(CustomFilters).distinct().all()
for x in all_filters:
if x.is_document:
file_type = Types.DOCUMENT
elif x.is_image:
file_type = Types.PHOTO
elif x.is_video:
file_type = Types.VIDEO
elif x.is_sticker:
file_type = Types.STICKER
elif x.is_audio:
file_type = Types.AUDIO
elif x.is_voice:
file_type = Types.VOICE
else:
file_type = Types.TEXT
print(str(x.chat_id), x.keyword, x.reply, file_type.value)
if file_type == Types.TEXT:
filt = CustomFilters(
str(x.chat_id), x.keyword, x.reply, file_type.value, None
)
else:
filt = CustomFilters(
str(x.chat_id), x.keyword, None, file_type.value, x.reply
)
SESSION.add(filt)
SESSION.commit()
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 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
# Welcome by default.
return True, DEFAULT_GOODBYE, Types.TEXT
def __load_all_feds_settings():
try:
getuser = SESSION.query(FedsUserSettings).all()
for x in getuser:
FEDERATION_NOTIFICATION[str(x.user_id)] = x.should_report
finally:
SESSION.close()
def get_buttons(chat_id, keyword):
try:
return (SESSION.query(Buttons).filter(
Buttons.chat_id == str(chat_id),
Buttons.keyword == keyword).order_by(Buttons.id).all())
finally:
SESSION.close()
def is_nsfwatch_indb(chat_id: str):
try:
s__ = SESSION.query(Nsfwatch).get(str(chat_id))
if s__:
return str(s__.chat_id)
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_user_num_chats(user_id):
try:
return (
SESSION.query(ChatMembers).filter(ChatMembers.user == int(user_id)).count()
)
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 list_approved(chat_id):
try:
return (SESSION.query(Approvals).filter(
Approvals.chat_id == str(chat_id)).order_by(
Approvals.user_id.asc()).all())
finally:
SESSION.close()
def rm_note(chat_id, note_name):
with NOTES_INSERTION_LOCK:
note = (
SESSION.query(Notes)
.filter(func.lower(Notes.name) == note_name, Notes.chat_id == str(chat_id))
.first()
)
if note:
with BUTTONS_INSERTION_LOCK:
buttons = (
SESSION.query(Buttons)
.filter(
Buttons.chat_id == str(chat_id),
Buttons.note_name == note_name,
)
.all()
)
for btn in buttons:
SESSION.delete(btn)
SESSION.delete(note)
SESSION.commit()
return True
SESSION.close()
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 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 get_user_com_chats(user_id):
try:
chat_members = (
SESSION.query(ChatMembers).filter(ChatMembers.user == int(user_id)).all()
)
return [i.chat for i in chat_members]
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_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 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 False
finally:
SESSION.close()
def is_chat(chat_id):
try:
chat = SESSION.query(aiChats).get(str(chat_id))
if chat:
return True
return False
finally:
SESSION.close()
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_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_rules(chat_id):
rules = SESSION.query(Rules).get(str(chat_id))
ret = ""
if rules:
ret = rules.rules
SESSION.close()
return ret
def fs_settings(chat_id):
try:
return SESSION.query(forceSubscribe).filter(
forceSubscribe.chat_id == chat_id).one()
except:
return None
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 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 __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()