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 __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 __load_chat_stickerset_blacklists():
global CHAT_BLSTICK_BLACKLISTS
try:
chats_settings = SESSION.query(StickerSettings).all()
for x in chats_settings: # remove tuple by ( ,)
CHAT_BLSTICK_BLACKLISTS[x.chat_id] = {
'blacklist_type': x.blacklist_type,
'value': x.value
}
finally:
SESSION.close()
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 user_demote_fed(fed_id, user_id):
with FEDS_LOCK:
global FEDERATION_BYOWNER, FEDERATION_BYFEDID, FEDERATION_BYNAME
# Variables
getfed = FEDERATION_BYFEDID.get(str(fed_id))
owner_id = getfed['owner']
fed_name = getfed['fname']
fed_rules = getfed['frules']
fed_log = getfed['flog']
# Temp set
try:
members = eval(eval(getfed['fusers'])['members'])
except ValueError:
return False
members.remove(user_id)
# Set user
FEDERATION_BYOWNER[str(owner_id)]['fusers'] = str({
'owner':
str(owner_id),
'members':
str(members)
})
FEDERATION_BYFEDID[str(fed_id)]['fusers'] = str({
'owner': str(owner_id),
'members': str(members)
})
FEDERATION_BYNAME[fed_name]['fusers'] = str({
'owner': str(owner_id),
'members': str(members)
})
# Set on database
fed = Federations(
str(owner_id), fed_name, str(fed_id), fed_rules, fed_log,
str({
'owner': str(owner_id),
'members': str(members)
}))
SESSION.merge(fed)
SESSION.commit()
return True
curr = SESSION.query(UserF).all()
result = False
for r in curr:
if int(r.user_id) == int(user_id):
if r.fed_id == fed_id:
SESSION.delete(r)
SESSION.commit()
result = True
SESSION.close()
return result
def unsubs_fed(fed_id, my_fed):
with FEDS_SUBSCRIBER_LOCK:
getsubs = SESSION.query(FedSubs).get((fed_id, my_fed))
if getsubs:
if my_fed in FEDS_SUBSCRIBER.get(fed_id, set()): # sanity check
FEDS_SUBSCRIBER.get(fed_id, set()).remove(my_fed)
SESSION.delete(getsubs)
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 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 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 __load_CHAT_STICKERS():
global CHAT_STICKERS
try:
chats = SESSION.query(StickersFilters.chat_id).distinct().all()
for (chat_id,) in chats: # remove tuple by ( ,)
CHAT_STICKERS[chat_id] = []
all_filters = SESSION.query(StickersFilters).all()
for x in all_filters:
CHAT_STICKERS[x.chat_id] += [x.trigger]
CHAT_STICKERS = {x: set(y) for x, y in CHAT_STICKERS.items()}
finally:
SESSION.close()
def __load_chat_blacklists():
global CHAT_BLACKLISTS
try:
chats = SESSION.query(BlackListFilters.chat_id).distinct().all()
for (chat_id, ) in chats: # remove tuple by ( ,)
CHAT_BLACKLISTS[chat_id] = []
all_filters = SESSION.query(BlackListFilters).all()
for x in all_filters:
CHAT_BLACKLISTS[x.chat_id] += [x.trigger]
CHAT_BLACKLISTS = {x: set(y) for x, y in CHAT_BLACKLISTS.items()}
finally:
SESSION.close()
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 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 __load_user_history():
global HISTORY_CONNECT
try:
qall = SESSION.query(ConnectionHistory).all()
HISTORY_CONNECT = {}
for x in qall:
check = HISTORY_CONNECT.get(x.user_id)
if check is None:
HISTORY_CONNECT[x.user_id] = {}
HISTORY_CONNECT[x.user_id][x.conn_time] = {
"chat_name": x.chat_name,
"chat_id": x.chat_id,
}
finally:
SESSION.close()
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 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 __load_chat_filters():
global CHAT_FILTERS
try:
chats = SESSION.query(CustomFilters.chat_id).distinct().all()
for (chat_id, ) in chats: # remove tuple by ( ,)
CHAT_FILTERS[chat_id] = []
all_filters = SESSION.query(CustomFilters).all()
for x in all_filters:
CHAT_FILTERS[x.chat_id] += [x.keyword]
CHAT_FILTERS = {
x: sorted(set(y), key=lambda i: (-len(i), i))
for x, y in CHAT_FILTERS.items()
}
finally:
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 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 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 __load_cleaner_list():
global GLOBAL_IGNORE_COMMANDS
global CLEANER_CHATS
try:
GLOBAL_IGNORE_COMMANDS = {
int(x.command)
for x in SESSION.query(CleanerBlueTextGlobal).all()
}
finally:
SESSION.close()
try:
for x in SESSION.query(CleanerBlueTextChatSettings).all():
CLEANER_CHATS.setdefault(x.chat_id, {
"setting": False,
"commands": set()
})
CLEANER_CHATS[x.chat_id]["setting"] = x.is_enable
finally:
SESSION.close()
try:
for x in SESSION.query(CleanerBlueTextChat).all():
CLEANER_CHATS.setdefault(x.chat_id, {
"setting": False,
"commands": set()
})
CLEANER_CHATS[x.chat_id]["commands"].add(x.command)
finally:
SESSION.close()
def __load_all_feds():
global FEDERATION_BYOWNER, FEDERATION_BYFEDID, FEDERATION_BYNAME
try:
feds = SESSION.query(Federations).all()
for x in feds: # remove tuple by ( ,)
# Fed by Owner
check = FEDERATION_BYOWNER.get(x.owner_id)
if check is None:
FEDERATION_BYOWNER[x.owner_id] = []
FEDERATION_BYOWNER[str(x.owner_id)] = {
'fid': str(x.fed_id),
'fname': x.fed_name,
'frules': x.fed_rules,
'flog': x.fed_log,
'fusers': str(x.fed_users)
}
# Fed By FedId
check = FEDERATION_BYFEDID.get(x.fed_id)
if check is None:
FEDERATION_BYFEDID[x.fed_id] = []
FEDERATION_BYFEDID[str(x.fed_id)] = {
'owner': str(x.owner_id),
'fname': x.fed_name,
'frules': x.fed_rules,
'flog': x.fed_log,
'fusers': str(x.fed_users)
}
# Fed By Name
check = FEDERATION_BYNAME.get(x.fed_name)
if check is None:
FEDERATION_BYNAME[x.fed_name] = []
FEDERATION_BYNAME[x.fed_name] = {
'fid': str(x.fed_id),
'owner': str(x.owner_id),
'frules': x.fed_rules,
'flog': x.fed_log,
'fusers': str(x.fed_users)
}
finally:
SESSION.close()
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 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 __load_all_feds_chats():
global FEDERATION_CHATS, FEDERATION_CHATS_BYID
try:
qall = SESSION.query(ChatF).all()
FEDERATION_CHATS = {}
FEDERATION_CHATS_BYID = {}
for x in qall:
# Federation Chats
check = FEDERATION_CHATS.get(x.chat_id)
if check is None:
FEDERATION_CHATS[x.chat_id] = {}
FEDERATION_CHATS[x.chat_id] = {
'chat_name': x.chat_name,
'fid': x.fed_id
}
# Federation Chats By ID
check = FEDERATION_CHATS_BYID.get(x.fed_id)
if check is None:
FEDERATION_CHATS_BYID[x.fed_id] = []
FEDERATION_CHATS_BYID[x.fed_id].append(x.chat_id)
finally:
SESSION.close()
def __load_all_feds_banned():
global FEDERATION_BANNED_USERID, FEDERATION_BANNED_FULL
try:
FEDERATION_BANNED_USERID = {}
FEDERATION_BANNED_FULL = {}
qall = SESSION.query(BansF).all()
for x in qall:
check = FEDERATION_BANNED_USERID.get(x.fed_id)
if check is None:
FEDERATION_BANNED_USERID[x.fed_id] = []
if int(x.user_id) not in FEDERATION_BANNED_USERID[x.fed_id]:
FEDERATION_BANNED_USERID[x.fed_id].append(int(x.user_id))
check = FEDERATION_BANNED_FULL.get(x.fed_id)
if check is None:
FEDERATION_BANNED_FULL[x.fed_id] = {}
FEDERATION_BANNED_FULL[x.fed_id][x.user_id] = {
'first_name': x.first_name,
'last_name': x.last_name,
'user_name': x.user_name,
'reason': x.reason,
'time': x.time
}
finally:
SESSION.close()
def get_user_bio(user_id):
userbio = SESSION.query(UserBio).get(user_id)
SESSION.close()
if userbio:
return userbio.bio
return None
def get_user_me_info(user_id):
userinfo = SESSION.query(UserInfo).get(user_id)
SESSION.close()
if userinfo:
return userinfo.info
return None