def set_blacklist_strength(chat_id, blacklist_type, value):
# for blacklist_type
# 0 = nothing
# 1 = delete
# 2 = warn
# 3 = mute
# 4 = kick
# 5 = ban
# 6 = tban
# 7 = tmute
with BLACKLIST_SETTINGS_INSERTION_LOCK:
global CHAT_SETTINGS_BLACKLISTS
curr_setting = SESSION.query(BlacklistSettings).get(str(chat_id))
if not curr_setting:
curr_setting = BlacklistSettings(
chat_id, blacklist_type=int(blacklist_type), value=value)
curr_setting.blacklist_type = int(blacklist_type)
curr_setting.value = str(value)
CHAT_SETTINGS_BLACKLISTS[str(chat_id)] = {
"blacklist_type": int(blacklist_type),
"value": value,
}
SESSION.add(curr_setting)
SESSION.commit()
def migrate_chat(old_chat_id, new_chat_id):
with CHAT_LOCK:
chat_notes = (SESSION.query(ReportingChatSettings).filter(
ReportingChatSettings.chat_id == str(old_chat_id)).all())
for note in chat_notes:
note.chat_id = str(new_chat_id)
SESSION.commit()
def set_custom_welcome(chat_id,
custom_content,
custom_welcome,
welcome_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_welcome or custom_content:
welcome_settings.custom_content = custom_content
welcome_settings.custom_welcome = custom_welcome
welcome_settings.welcome_type = welcome_type.value
else:
welcome_settings.custom_welcome = DEFAULT_WELCOME
welcome_settings.welcome_type = Types.TEXT.value
SESSION.add(welcome_settings)
with WELC_BTN_LOCK:
prev_buttons = (SESSION.query(WelcomeButtons).filter(
WelcomeButtons.chat_id == str(chat_id)).all())
for btn in prev_buttons:
SESSION.delete(btn)
for b_name, url, same_line in buttons:
button = WelcomeButtons(chat_id, b_name, url, same_line)
SESSION.add(button)
SESSION.commit()
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 rem_chat(chat_id):
with INSERTION_LOCK:
autochat = SESSION.query(ChatbotChats).get(str(chat_id))
if autochat:
SESSION.delete(autochat)
SESSION.commit()
def set_nsfw(chat_id):
with INSERTION_LOCK:
nsfwchat = SESSION.query(NSFWChats).get(str(chat_id))
if not nsfwchat:
nsfwchat = NSFWChats(str(chat_id))
SESSION.add(nsfwchat)
SESSION.commit()
def migrate_chat(old_chat_id, new_chat_id):
with INSERTION_LOCK:
chat = SESSION.query(Welcome).get(str(old_chat_id))
if chat:
chat.chat_id = str(new_chat_id)
with WELC_BTN_LOCK:
chat_buttons = (
SESSION.query(WelcomeButtons)
.filter(WelcomeButtons.chat_id == str(old_chat_id))
.all()
)
for btn in chat_buttons:
btn.chat_id = str(new_chat_id)
with LEAVE_BTN_LOCK:
chat_buttons = (
SESSION.query(GoodbyeButtons)
.filter(GoodbyeButtons.chat_id == str(old_chat_id))
.all()
)
for btn in chat_buttons:
btn.chat_id = str(new_chat_id)
SESSION.commit()
def add_note_to_db(chat_id,
note_name,
note_data,
msgtype,
buttons=None,
file=None):
if not buttons:
buttons = []
with NOTES_INSERTION_LOCK:
prev = SESSION.query(Notes).get((str(chat_id), note_name))
if prev:
with BUTTONS_INSERTION_LOCK:
prev_buttons = (SESSION.query(Buttons).filter(
Buttons.chat_id == str(chat_id),
Buttons.note_name == note_name).all())
for btn in prev_buttons:
SESSION.delete(btn)
SESSION.delete(prev)
note = Notes(str(chat_id),
note_name,
note_data or "",
msgtype=msgtype.value,
file=file)
SESSION.add(note)
SESSION.commit()
for b_name, url, same_line in buttons:
add_note_button_to_db(chat_id, note_name, b_name, url, same_line)
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 add_filter(
chat_id,
keyword,
reply,
snip_type,
media_id,
media_access_hash,
media_file_reference,
):
adder = SESSION.query(Filters).get((str(chat_id), keyword))
if adder:
adder.reply = reply
adder.snip_type = snip_type
adder.media_id = media_id
adder.media_access_hash = media_access_hash
adder.media_file_reference = media_file_reference
else:
adder = Filters(
chat_id,
keyword,
reply,
snip_type,
media_id,
media_access_hash,
media_file_reference,
)
SESSION.add(adder)
SESSION.commit()
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 get_filter(chat_id, keyword):
try:
return SESSION.query(Filters).get((str(chat_id), keyword))
except BaseException:
return None
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 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 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 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 migrate_chat(old_chat_id, new_chat_id):
with STICKERS_FILTER_INSERTION_LOCK:
chat_filters = (SESSION.query(StickersFilters).filter(
StickersFilters.chat_id == str(old_chat_id)).all())
for filt in chat_filters:
filt.chat_id = str(new_chat_id)
SESSION.commit()
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 __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 is_talkmode_indb(chat_id: str):
try:
s__ = SESSION.query(Talkmode).get(str(chat_id))
if s__:
return str(s__.chat_id)
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 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_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_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 remove_url(tg_chat_id, tg_feed_link):
with INSERTION_LOCK:
# this loops to delete any possible duplicates for the same TG User ID, TG Chat ID and link
for row in check_url_availability(tg_chat_id, tg_feed_link):
# add the action to the DB query
SESSION.delete(row)
SESSION.commit()
def add_channel(chat_id, channel):
adder = SESSION.query(forceSubscribe).get(chat_id)
if adder:
adder.channel = channel
else:
adder = forceSubscribe(chat_id, channel)
SESSION.add(adder)
SESSION.commit()
def fs_settings(chat_id):
try:
return SESSION.query(forceSubscribe).filter(
forceSubscribe.chat_id == chat_id).one()
except:
return None
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
