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 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 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 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 blacklist_url(chat_id, domain):
with URL_BLACKLIST_FILTER_INSERTION_LOCK:
domain_filt = URLBlackListFilters(str(chat_id), domain)
SESSION.merge(domain_filt)
SESSION.commit()
CHAT_URL_BLACKLISTS.setdefault(str(chat_id), set()).add(domain)
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_user(user_id, username, chat_id=None, chat_name=None):
with INSERTION_LOCK:
user = SESSION.query(Users).get(user_id)
if not user:
user = Users(user_id, username)
SESSION.add(user)
SESSION.flush()
else:
user.username = username
if not chat_id or not chat_name:
SESSION.commit()
return
chat = SESSION.query(Chats).get(str(chat_id))
if not chat:
chat = Chats(str(chat_id), chat_name)
SESSION.add(chat)
SESSION.flush()
else:
chat.chat_name = chat_name
member = SESSION.query(ChatMembers).filter(
ChatMembers.chat == chat.chat_id,
ChatMembers.user == user.user_id).first()
if not member:
chat_member = ChatMembers(chat.chat_id, user.user_id)
SESSION.add(chat_member)
SESSION.commit()
def set_welcome_security(chat_id, security):
with WS_LOCK:
prev = SESSION.query(WelcomeSecurity).get((str(chat_id)))
if prev:
SESSION.delete(prev)
welcome_s = WelcomeSecurity(str(chat_id), security)
SESSION.add(welcome_s)
SESSION.commit()
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 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 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 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 ungpromote_user(user_id):
with SUDO_USERS_LOCK:
user = SESSION.query(SudoUsers).get(user_id)
if user:
SESSION.delete(user)
SESSION.commit()
__load_sudo_userid_list()
def update_url(row_id, new_entry_links):
with INSERTION_LOCK:
row = SESSION.query(RSS).get(row_id)
# set the new old_entry_link with the latest update from the RSS Feed
row.old_entry_link = new_entry_links[0]
# commit the changes to the DB
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 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_chat_setting(chat_id: Union[int, str], setting: bool):
with CHAT_LOCK:
chat_setting = SESSION.query(ReportingChatSettings).get(str(chat_id))
if not chat_setting:
chat_setting = ReportingChatSettings(chat_id)
chat_setting.should_report = setting
SESSION.add(chat_setting)
SESSION.commit()
def set_user_setting(user_id: int, setting: bool):
with USER_LOCK:
user_setting = SESSION.query(ReportingUserSettings).get(user_id)
if not user_setting:
user_setting = ReportingUserSettings(user_id)
user_setting.should_report = setting
SESSION.add(user_setting)
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 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_command_reaction(chat_id: Union[int, str], setting: bool):
with CHAT_LOCK:
chat_setting = SESSION.query(CommandReactionChatSettings).get(str(chat_id))
if not chat_setting:
chat_setting = CommandReactionChatSettings(chat_id)
chat_setting.comm_reaction = setting
SESSION.add(chat_setting)
SESSION.commit()
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 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 set_allow_connect_to_chat(chat_id: Union[int, str], setting: bool):
with CHAT_ACCESS_LOCK:
chat_setting = SESSION.query(ChatAccessConnectionSettings).get(str(chat_id))
if not chat_setting:
chat_setting = ChatAccessConnectionSettings(chat_id)
chat_setting.allow_connect_to_chat = setting
SESSION.add(chat_setting)
SESSION.commit()
def set_gdbye_preference(chat_id, should_goodbye):
with INSERTION_LOCK:
curr = SESSION.query(Welcome).get(str(chat_id))
if not curr:
curr = Welcome(str(chat_id), should_goodbye=should_goodbye)
else:
curr.should_goodbye = should_goodbye
SESSION.add(curr)
SESSION.commit()
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 migrate_chat(old_chat_id, new_chat_id):
with LOGS_INSERTION_LOCK:
chat = SESSION.query(GroupLogs).get(str(old_chat_id))
if chat:
chat.chat_id = str(new_chat_id)
SESSION.add(chat)
if str(old_chat_id) in CHANNELS:
CHANNELS[str(new_chat_id)] = CHANNELS.get(str(old_chat_id))
SESSION.commit()
def disable_antispam(chat_id):
with ASPAM_SETTING_LOCK:
chat = SESSION.query(AntispamSettings).get(str(chat_id))
if not chat:
chat = AntispamSettings(chat_id, False)
chat.setting = False
SESSION.add(chat)
SESSION.commit()
GBANSTAT_LIST.add(str(chat_id))
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