def warn_user(user_id, chat_id, reason=None):
with WARN_INSERTION_LOCK:
warned_user = SESSION.query(Warns).get((user_id, str(chat_id)))
if not warned_user:
warned_user = Warns(user_id, str(chat_id))
warned_user.num_warns += 1
if reason:
warned_user.reasons = warned_user.reasons + [
reason
] # TODO:: double check this wizardry
reasons = warned_user.reasons
num = warned_user.num_warns
SESSION.add(warned_user)
SESSION.commit()
return num, reasons
def set_cleanbt(chat_id, is_enable):
with CLEANER_CHAT_SETTINGS:
curr = SESSION.query(CleanerBlueTextChatSettings).get(str(chat_id))
if not curr:
curr = CleanerBlueTextChatSettings(str(chat_id), is_enable)
else:
curr.is_enabled = is_enable
if str(chat_id) not in CLEANER_CHATS:
CLEANER_CHATS.setdefault(str(chat_id), {
"setting": False,
"commands": set()
})
CLEANER_CHATS[str(chat_id)]["setting"] = is_enable
SESSION.add(curr)
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 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 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
else:
SESSION.close()
return False
def is_locked(chat_id, lock_type):
curr_perm = SESSION.query(Permissions).get(str(chat_id))
SESSION.close()
if not curr_perm:
return False
elif lock_type == "sticker":
return curr_perm.sticker
elif lock_type == "photo":
return curr_perm.photo
elif lock_type == "audio":
return curr_perm.audio
elif lock_type == "voice":
return curr_perm.voice
elif lock_type == "contact":
return curr_perm.contact
elif lock_type == "video":
return curr_perm.video
elif lock_type == "document":
return curr_perm.document
elif lock_type == "gif":
return curr_perm.gif
elif lock_type == "url":
return curr_perm.url
elif lock_type == "bots":
return curr_perm.bots
elif lock_type == "forward":
return curr_perm.forward
elif lock_type == "game":
return curr_perm.game
elif lock_type == "location":
return curr_perm.location
elif lock_type == "rtl":
return curr_perm.rtl
elif lock_type == "button":
return curr_perm.button
elif lock_type == "egame":
return curr_perm.egame
elif lock_type == "inline":
return curr_perm.inline
def disapprove(chat_id, user_id):
with APPROVE_INSERTION_LOCK:
disapprove_user = SESSION.query(Approvals).get((str(chat_id), user_id))
if disapprove_user:
SESSION.delete(disapprove_user)
SESSION.commit()
return True
else:
SESSION.close()
return False
def update_restriction(chat_id, restr_type, locked):
with RESTR_LOCK:
curr_restr = SESSION.query(Restrictions).get(str(chat_id))
if not curr_restr:
curr_restr = init_restrictions(chat_id)
if restr_type == "messages":
curr_restr.messages = locked
elif restr_type == "media":
curr_restr.media = locked
elif restr_type == "other":
curr_restr.other = locked
elif restr_type == "previews":
curr_restr.preview = locked
elif restr_type == "all":
curr_restr.messages = locked
curr_restr.media = locked
curr_restr.other = locked
curr_restr.preview = locked
SESSION.add(curr_restr)
SESSION.commit()
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 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 reset_warns(user_id, chat_id):
with WARN_INSERTION_LOCK:
warned_user = SESSION.query(Warns).get((user_id, str(chat_id)))
if warned_user:
warned_user.num_warns = 0
warned_user.reasons = []
SESSION.add(warned_user)
SESSION.commit()
SESSION.close()
def is_restr_locked(chat_id, lock_type):
curr_restr = SESSION.query(Restrictions).get(str(chat_id))
SESSION.close()
if not curr_restr:
return False
if lock_type == "messages":
return curr_restr.messages
elif lock_type == "media":
return curr_restr.media
elif lock_type == "other":
return curr_restr.other
elif lock_type == "previews":
return curr_restr.preview
elif lock_type == "all":
return (
curr_restr.messages
and curr_restr.media
and curr_restr.other
and curr_restr.preview
)
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 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 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 remove_warn(user_id, chat_id):
with WARN_INSERTION_LOCK:
removed = False
warned_user = SESSION.query(Warns).get((user_id, str(chat_id)))
if warned_user and warned_user.num_warns > 0:
warned_user.num_warns -= 1
warned_user.reasons = warned_user.reasons[:-1]
SESSION.add(warned_user)
SESSION.commit()
removed = True
SESSION.close()
return removed
def disable_command(chat_id, disable):
with DISABLE_INSERTION_LOCK:
disabled = SESSION.query(Disable).get((str(chat_id), disable))
if not disabled:
DISABLED.setdefault(str(chat_id), set()).add(disable)
disabled = Disable(str(chat_id), disable)
SESSION.add(disabled)
SESSION.commit()
return True
SESSION.close()
return False
def global_ignore_command(command):
command = command.lower()
with CLEANER_GLOBAL_LOCK:
ignored = SESSION.query(CleanerBlueTextGlobal).get(str(command))
if not ignored:
GLOBAL_IGNORE_COMMANDS.add(command)
ignored = CleanerBlueTextGlobal(str(command))
SESSION.add(ignored)
SESSION.commit()
return True
SESSION.close()
return False
def global_unignore_command(command):
command = command.lower()
with CLEANER_GLOBAL_LOCK:
unignored = SESSION.query(CleanerBlueTextGlobal).get(str(command))
if unignored:
if command in GLOBAL_IGNORE_COMMANDS:
GLOBAL_IGNORE_COMMANDS.remove(command)
SESSION.delete(command)
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 rm_from_stickers(chat_id, trigger):
with STICKERS_FILTER_INSERTION_LOCK:
stickers_filt = SESSION.query(StickersFilters).get(
(str(chat_id), trigger))
if stickers_filt:
if trigger in CHAT_STICKERS.get(str(chat_id),
set()): # sanity check
CHAT_STICKERS.get(str(chat_id), set()).remove(trigger)
SESSION.delete(stickers_filt)
SESSION.commit()
return True
SESSION.close()
return False
def migrate_chat(old_chat_id, new_chat_id):
with PERM_LOCK:
perms = SESSION.query(Permissions).get(str(old_chat_id))
if perms:
perms.chat_id = str(new_chat_id)
SESSION.commit()
with RESTR_LOCK:
rest = SESSION.query(Restrictions).get(str(old_chat_id))
if rest:
rest.chat_id = str(new_chat_id)
SESSION.commit()
def migrate_chat(old_chat_id, new_chat_id):
with INSERTION_LOCK:
chat = SESSION.query(Chats).get(str(old_chat_id))
if chat:
chat.chat_id = str(new_chat_id)
SESSION.commit()
chat_members = (
SESSION.query(ChatMembers)
.filter(ChatMembers.chat == str(old_chat_id))
.all()
)
for member in chat_members:
member.chat = str(new_chat_id)
SESSION.commit()
def migrate_chat(old_chat_id, new_chat_id):
with CUST_FILT_LOCK:
chat_filters = (SESSION.query(CustomFilters).filter(
CustomFilters.chat_id == str(old_chat_id)).all())
for filt in chat_filters:
filt.chat_id = str(new_chat_id)
SESSION.commit()
old_filt = CHAT_FILTERS.get(str(old_chat_id))
if old_filt:
CHAT_FILTERS[str(new_chat_id)] = old_filt
del CHAT_FILTERS[str(old_chat_id)]
with BUTTON_LOCK:
chat_buttons = (SESSION.query(Buttons).filter(
Buttons.chat_id == str(old_chat_id)).all())
for btn in chat_buttons:
btn.chat_id = str(new_chat_id)
SESSION.commit()
def chat_unignore_command(chat_id, unignore):
unignore = unignore.lower()
with CLEANER_CHAT_LOCK:
unignored = SESSION.query(CleanerBlueTextChat).get(
(str(chat_id), unignore))
if unignored:
if str(chat_id) not in CLEANER_CHATS:
CLEANER_CHATS.setdefault(str(chat_id), {
"setting": False,
"commands": set()
})
if unignore in CLEANER_CHATS.get(str(chat_id)).get("commands"):
CLEANER_CHATS[str(chat_id)]["commands"].remove(unignore)
SESSION.delete(unignored)
SESSION.commit()
return True
SESSION.close()
return False
def chat_ignore_command(chat_id, ignore):
ignore = ignore.lower()
with CLEANER_CHAT_LOCK:
ignored = SESSION.query(CleanerBlueTextChat).get(
(str(chat_id), ignore))
if not ignored:
if str(chat_id) not in CLEANER_CHATS:
CLEANER_CHATS.setdefault(str(chat_id), {
"setting": False,
"commands": set()
})
CLEANER_CHATS[str(chat_id)]["commands"].add(ignore)
ignored = CleanerBlueTextChat(str(chat_id), ignore)
SESSION.add(ignored)
SESSION.commit()
return True
SESSION.close()
return False
def num_logchannels():
try:
return SESSION.query(func.count(distinct(GroupLogs.chat_id))).scalar()
finally:
SESSION.close()
def check_afk_status(user_id):
try:
return SESSION.query(AFK).get(user_id)
finally:
SESSION.close()
def get_locks(chat_id):
try:
return SESSION.query(Permissions).get(str(chat_id))
finally:
SESSION.close()
def get_restr(chat_id):
try:
return SESSION.query(Restrictions).get(str(chat_id))
finally:
SESSION.close()
