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 disable_gbans(chat_id):
with GBAN_SETTING_LOCK:
chat = SESSION.query(GbanSettings).get(str(chat_id))
if not chat:
chat = GbanSettings(chat_id, False)
chat.setting = False
SESSION.add(chat)
SESSION.commit()
GBANSTAT_LIST.add(str(chat_id))
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 set_user(user_id, username):
with INSERTION_LOCK:
user = SESSION.query(LastFMUsers).get(str(user_id))
if not user:
user = LastFMUsers(str(user_id), str(username))
else:
user.username = str(username)
SESSION.add(user)
SESSION.commit()
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, setting)
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 set_user_bio(user_id, bio):
with INSERTION_LOCK:
userbio = SESSION.query(UserBio).get(user_id)
if userbio:
userbio.bio = bio
else:
userbio = UserBio(user_id, bio)
SESSION.add(userbio)
SESSION.commit()
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 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 get_warns(user_id, chat_id):
try:
user = SESSION.query(Warns).get((user_id, str(chat_id)))
if not user:
return None
reasons = user.reasons
num = user.num_warns
return num, reasons
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_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 set_clean_welcome(chat_id, clean_welcome):
with INSERTION_LOCK:
curr = SESSION.query(Welcome).get(str(chat_id))
if not curr:
curr = Welcome(str(chat_id))
curr.clean_welcome = int(clean_welcome)
SESSION.add(curr)
SESSION.commit()
def get_warn_setting(chat_id):
try:
setting = SESSION.query(WarnSettings).get(str(chat_id))
if setting:
return setting.warn_limit, setting.soft_warn
else:
return 3, False
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()
def set_warn_strength(chat_id, soft_warn):
with WARN_SETTINGS_LOCK:
curr_setting = SESSION.query(WarnSettings).get(str(chat_id))
if not curr_setting:
curr_setting = WarnSettings(chat_id, soft_warn=soft_warn)
curr_setting.soft_warn = soft_warn
SESSION.add(curr_setting)
SESSION.commit()
def set_warn_limit(chat_id, warn_limit):
with WARN_SETTINGS_LOCK:
curr_setting = SESSION.query(WarnSettings).get(str(chat_id))
if not curr_setting:
curr_setting = WarnSettings(chat_id, warn_limit=warn_limit)
curr_setting.warn_limit = warn_limit
SESSION.add(curr_setting)
SESSION.commit()
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_afk_users():
global AFK_USERS
try:
all_afk = SESSION.query(AFK).all()
AFK_USERS = {
user.user_id: user.reason
for user in all_afk if user.is_afk
}
finally:
SESSION.close()
def migrate_chat(old_chat_id, new_chat_id):
with WARN_INSERTION_LOCK:
chat_notes = (SESSION.query(Warns).filter(
Warns.chat_id == str(old_chat_id)).all())
for note in chat_notes:
note.chat_id = str(new_chat_id)
SESSION.commit()
with WARN_FILTER_INSERTION_LOCK:
chat_filters = (
SESSION.query(WarnFilters)
.filter(WarnFilters.chat_id == str(old_chat_id))
.all()
)
for filt in chat_filters:
filt.chat_id = str(new_chat_id)
SESSION.commit()
WARN_FILTERS[str(new_chat_id)] = WARN_FILTERS[str(old_chat_id)]
del WARN_FILTERS[str(old_chat_id)]
with WARN_SETTINGS_LOCK:
chat_settings = (
SESSION.query(WarnSettings)
.filter(WarnSettings.chat_id == str(old_chat_id))
.all()
)
for setting in chat_settings:
setting.chat_id = str(new_chat_id)
SESSION.commit()
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 set_feds_setting(user_id: int, setting: bool):
with FEDS_SETTINGS_LOCK:
global FEDERATION_NOTIFICATION
user_setting = SESSION.query(FedsUserSettings).get(user_id)
if not user_setting:
user_setting = FedsUserSettings(user_id)
user_setting.should_report = setting
FEDERATION_NOTIFICATION[str(user_id)] = setting
SESSION.add(user_setting)
SESSION.commit()
def toggle_afk(user_id, reason=""):
with INSERTION_LOCK:
curr = SESSION.query(AFK).get(user_id)
if not curr:
curr = AFK(user_id, reason, True)
elif curr.is_afk:
curr.is_afk = False
elif not curr.is_afk:
curr.is_afk = True
SESSION.add(curr)
SESSION.commit()
def enable_gbans(chat_id):
with GBAN_SETTING_LOCK:
chat = SESSION.query(GbanSettings).get(str(chat_id))
if not chat:
chat = GbanSettings(chat_id, True)
chat.setting = True
SESSION.add(chat)
SESSION.commit()
if str(chat_id) in GBANSTAT_LIST:
GBANSTAT_LIST.remove(str(chat_id))
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 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 set_chat_log_channel(chat_id, log_channel):
with LOGS_INSERTION_LOCK:
res = SESSION.query(GroupLogs).get(str(chat_id))
if res:
res.log_channel = log_channel
else:
res = GroupLogs(chat_id, log_channel)
SESSION.add(res)
CHANNELS[str(chat_id)] = log_channel
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()
global CHAT_BLACKLISTS
if CHAT_BLACKLISTS.get(str(chat_id), set()) == set():
CHAT_BLACKLISTS[str(chat_id)] = {trigger}
else:
CHAT_BLACKLISTS.get(str(chat_id), set()).add(trigger)
def __load_chat_settings_blacklists():
global CHAT_SETTINGS_BLACKLISTS
try:
chats_settings = SESSION.query(BlacklistSettings).all()
for x in chats_settings: # remove tuple by ( ,)
CHAT_SETTINGS_BLACKLISTS[x.chat_id] = {
"blacklist_type": x.blacklist_type,
"value": x.value,
}
finally:
SESSION.close()
def update_gban_reason(user_id, name, reason=None):
with GBANNED_USERS_LOCK:
user = SESSION.query(GloballyBannedUsers).get(user_id)
if not user:
return None
old_reason = user.reason
user.name = name
user.reason = reason
SESSION.merge(user)
SESSION.commit()
return old_reason