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 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 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 switch_to_locale(chat_id, locale_name):
with LOCALES_INSERTION_LOCK:
prev = SESSION.query(Locales).get((str(chat_id)))
if prev:
SESSION.delete(prev)
switch_locale = Locales(str(chat_id), locale_name)
SESSION.add(switch_locale)
SESSION.commit()
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 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 init_restrictions(chat_id, reset=False):
curr_restr = SESSION.query(Restrictions).get(str(chat_id))
if reset:
SESSION.delete(curr_restr)
SESSION.flush()
restr = Restrictions(str(chat_id))
SESSION.add(restr)
SESSION.commit()
return restr
def set_safemode(chat_id, safemode_status=True):
with SAFEMODE_INSERTION_LOCK:
curr = SESSION.query(Safemode).get((str(chat_id)))
if curr:
SESSION.delete(curr)
switch_status = Safemode(str(chat_id), safemode_status)
SESSION.add(switch_status)
SESSION.commit()
def connect(user_id, chat_id):
with CONNECTION_INSERTION_LOCK:
prev = SESSION.query(Connection).get((int(user_id)))
if prev:
SESSION.delete(prev)
connect_to_chat = Connection(int(user_id), chat_id)
SESSION.add(connect_to_chat)
SESSION.commit()
return True
def add_history(user_id, chat_id1, chat_id2, chat_id3, updated):
with HISTORY_LOCK:
prev = SESSION.query(ConnectionHistory).get((int(user_id)))
if prev:
SESSION.delete(prev)
history = ConnectionHistory(user_id, chat_id1, chat_id2, chat_id3,
updated)
SESSION.add(history)
SESSION.commit()
def init_permissions(chat_id, reset=False):
curr_perm = SESSION.query(Permissions).get(str(chat_id))
if reset:
SESSION.delete(curr_perm)
SESSION.flush()
perm = Permissions(str(chat_id))
SESSION.add(perm)
SESSION.commit()
return perm
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 clear_user_bio(user_id):
with INSERTION_LOCK:
curr = SESSION.query(UserBio).get(user_id)
if curr:
SESSION.delete(curr)
SESSION.commit()
return True
SESSION.close()
return False
def chat_leave_fed(chat_id):
with FEDS_LOCK:
H = False
curr = SESSION.query(ChatF).all()
for U in curr:
if int(U.chat_id) == int(chat_id):
H = True
SESSION.delete(U)
SESSION.commit()
return H
def set_frules(fed_id, rules):
with FEDS_LOCK:
r = SESSION.query(RulesF).get(fed_id)
if r:
print("fund prev")
SESSION.delete(r)
r = RulesF(str(fed_id), rules)
SESSION.add(r)
SESSION.commit()
return r
def stop_chat_logging(chat_id):
with LOGS_INSERTION_LOCK:
res = SESSION.query(GroupLogs).get(str(chat_id))
if res:
if str(chat_id) in CHANNELS:
del CHANNELS[str(chat_id)]
log_channel = res.log_channel
SESSION.delete(res)
SESSION.commit()
return log_channel
def remove_warn_filter(chat_id, keyword):
with WARN_FILTER_INSERTION_LOCK:
warn_filt = SESSION.query(WarnFilters).get((str(chat_id), keyword))
if warn_filt:
if keyword in WARN_FILTERS.get(str(chat_id), []): # sanity check
WARN_FILTERS.get(str(chat_id), []).remove(keyword)
SESSION.delete(warn_filt)
SESSION.commit()
return True
SESSION.close()
return False
def del_user(user_id):
with INSERTION_LOCK:
curr = SESSION.query(Users).get(user_id)
if curr:
SESSION.delete(curr)
SESSION.commit()
return True
ChatMembers.query.filter(ChatMembers.user == user_id).delete()
SESSION.commit()
SESSION.close()
return False
def un_fban_user(fed_id, user_id):
with FEDS_LOCK:
r = SESSION.query(BansF).all()
for I in r:
print("1")
if I.fed_id == fed_id:
print("2")
print(I.user_id, user_id)
if int(I.user_id) == int(user_id):
print("fund prev")
SESSION.delete(I)
SESSION.commit()
return I
def rm_afk(user_id):
with INSERTION_LOCK:
curr = SESSION.query(AFK).get(user_id)
if curr:
if user_id in AFK_USERS: # sanity check
del AFK_USERS[user_id]
SESSION.delete(curr)
SESSION.commit()
return True
SESSION.close()
return False
def rm_url_from_blacklist(chat_id, domain):
with URL_BLACKLIST_FILTER_INSERTION_LOCK:
domain_filt = SESSION.query(URLBlackListFilters).get(
(str(chat_id), domain))
if domain_filt:
if domain in CHAT_URL_BLACKLISTS.get(str(chat_id), set()):
CHAT_URL_BLACKLISTS.get(str(chat_id), set()).remove(domain)
SESSION.delete(domain_filt)
SESSION.commit()
return True
SESSION.close()
return False
def del_fed(fed_id, chat_id):
with FEDS_LOCK:
curr = SESSION.query(Federations).get(fed_id)
print(curr)
if curr:
SESSION.delete(curr)
SESSION.commit()
curr = SESSION.query(ChatF).get(str(chat_id))
if curr:
SESSION.delete(curr)
SESSION.commit()
curr = SESSION.query(UserF).all()
for I in curr:
if I.fed_id == fed_id:
SESSION.delete(I)
SESSION.commit()
curr = SESSION.query(RulesF).get(fed_id)
if curr:
SESSION.delete(curr)
SESSION.commit()
return
def enable_command(chat_id, enable):
with DISABLE_INSERTION_LOCK:
disabled = SESSION.query(Disable).get((str(chat_id), enable))
if disabled:
if enable in DISABLED.get(str(chat_id)): # sanity check
DISABLED.setdefault(str(chat_id), set()).remove(enable)
SESSION.delete(disabled)
SESSION.commit()
return True
SESSION.close()
return False
def rm_from_blacklist(chat_id, trigger):
with BLACKLIST_FILTER_INSERTION_LOCK:
blacklist_filt = SESSION.query(BlackListFilters).get(
(str(chat_id), trigger))
if blacklist_filt:
if trigger in CHAT_BLACKLISTS.get(str(chat_id),
set()): # sanity check
CHAT_BLACKLISTS.get(str(chat_id), set()).remove(trigger)
SESSION.delete(blacklist_filt)
SESSION.commit()
return True
SESSION.close()
return False
def user_demote_fed(fed_id, user_id):
with FEDS_LOCK:
curr = SESSION.query(UserF).all()
result = False
for r in curr:
print(r.user_id, user_id, r.fed_id, fed_id)
if int(r.user_id) == int(user_id):
print("yes1 - ", r.user_id, user_id)
if r.fed_id == fed_id:
print("yes2 - ", r.fed_id, fed_id)
SESSION.delete(r)
SESSION.commit()
result = True
SESSION.close()
return result
def rm_note(chat_id, note_name):
with NOTES_INSERTION_LOCK:
note = SESSION.query(Notes).get((str(chat_id), note_name))
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
else:
SESSION.close()
return False
def remove_filter(chat_id, keyword):
global CHAT_FILTERS
with CUST_FILT_LOCK:
filt = SESSION.query(CustomFilters).get((str(chat_id), keyword))
if filt:
if keyword in CHAT_FILTERS.get(str(chat_id), []): # Sanity check
CHAT_FILTERS.get(str(chat_id), []).remove(keyword)
with BUTTON_LOCK:
prev_buttons = SESSION.query(Buttons).filter(
Buttons.chat_id == str(chat_id),
Buttons.keyword == keyword).all()
for btn in prev_buttons:
SESSION.delete(btn)
SESSION.delete(filt)
SESSION.commit()
return True
SESSION.close()
return False
def add_filter(chat_id,
keyword,
reply,
is_sticker=False,
is_document=False,
is_image=False,
is_audio=False,
is_voice=False,
is_video=False,
buttons=None):
global CHAT_FILTERS
if buttons is None:
buttons = []
with CUST_FILT_LOCK:
prev = SESSION.query(CustomFilters).get((str(chat_id), keyword))
if prev:
with BUTTON_LOCK:
prev_buttons = SESSION.query(Buttons).filter(
Buttons.chat_id == str(chat_id),
Buttons.keyword == keyword).all()
for btn in prev_buttons:
SESSION.delete(btn)
SESSION.delete(prev)
filt = CustomFilters(str(chat_id), keyword, reply, is_sticker,
is_document, is_image, is_audio, is_voice,
is_video, bool(buttons))
if keyword not in CHAT_FILTERS.get(str(chat_id), []):
CHAT_FILTERS[str(chat_id)] = sorted(
CHAT_FILTERS.get(str(chat_id), []) + [keyword],
key=lambda x: (-len(x), x))
SESSION.add(filt)
SESSION.commit()
for b_name, url, same_line in buttons:
add_note_button_to_db(chat_id, keyword, b_name, url, same_line)
