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 fban_user(fed_id, user_id, first_name, last_name, user_name, reason, time):
with FEDS_LOCK:
r = SESSION.query(BansF).all()
for I in r:
if I.fed_id == fed_id:
if int(I.user_id) == int(user_id):
SESSION.delete(I)
r = BansF(str(fed_id), str(user_id), first_name, last_name, user_name,
reason, time)
SESSION.add(r)
try:
SESSION.commit()
except:
SESSION.rollback()
return False
finally:
SESSION.commit()
__load_all_feds_banned()
return r
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 __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 __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 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 multi_fban_user(multi_fed_id, multi_user_id, multi_first_name,
multi_last_name, multi_user_name, multi_reason):
if True: # with FEDS_LOCK:
counter = 0
time = 0
for x in range(len(multi_fed_id)):
fed_id = multi_fed_id[x]
user_id = multi_user_id[x]
first_name = multi_first_name[x]
last_name = multi_last_name[x]
user_name = multi_user_name[x]
reason = multi_reason[x]
r = SESSION.query(BansF).all()
for I in r:
if I.fed_id == fed_id:
if int(I.user_id) == int(user_id):
SESSION.delete(I)
r = BansF(str(fed_id), str(user_id), first_name, last_name,
user_name, reason, time)
SESSION.add(r)
counter += 1
if str(str(counter)[-2:]) == "00":
print(user_id)
print(first_name)
print(reason)
print(counter)
try:
SESSION.commit()
except:
SESSION.rollback()
return False
finally:
SESSION.commit()
__load_all_feds_banned()
print("Done")
return counter
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 num_warn_filters():
try:
return SESSION.query(WarnFilters).count()
finally:
SESSION.close()
def get_warn_filter(chat_id, keyword):
try:
return SESSION.query(WarnFilters).get((str(chat_id), keyword))
finally:
SESSION.close()
def num_stickers_filters():
try:
return SESSION.query(StickersFilters).count()
finally:
SESSION.close()
def rem_nsfw(chat_id):
with INSERTION_LOCK:
nsfwchat = SESSION.query(NSFWChats).get(str(chat_id))
if nsfwchat:
SESSION.delete(nsfwchat)
SESSION.commit()
def get_gbanned_user(user_id):
try:
return SESSION.query(GloballyBannedUsers).get(user_id)
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_repo(chat_id, name):
try:
return SESSION.query(GitHub).get((str(chat_id), name))
finally:
SESSION.close()
def get_user_me_info(user_id):
userinfo = SESSION.query(UserInfo).get(user_id)
SESSION.close()
if userinfo:
return userinfo.info
return None
def num_blacklist_filters():
try:
return SESSION.query(BlackListFilters).count()
finally:
SESSION.close()
def __load_gbanned_userid_list():
global GBANNED_LIST
try:
GBANNED_LIST = {x.user_id for x in SESSION.query(GloballyBannedUsers).all()}
finally:
SESSION.close()
def num_blacklist_chat_filters(chat_id):
try:
return (SESSION.query(BlackListFilters.chat_id).filter(
BlackListFilters.chat_id == str(chat_id)).count())
finally:
SESSION.close()
def get_gban_list():
try:
return [x.to_dict() for x in SESSION.query(GloballyBannedUsers).all()]
finally:
SESSION.close()
def get_all_chat_notes(chat_id):
try:
return (SESSION.query(Notes).filter(
Notes.chat_id == str(chat_id)).order_by(Notes.name.asc()).all())
finally:
SESSION.close()
def get_all_nsfw_chats():
try:
return SESSION.query(NSFWChats.chat_id).all()
finally:
SESSION.close()
def num_notes():
try:
return SESSION.query(Notes).count()
finally:
SESSION.close()
def num_stickers_chat_filters(chat_id):
try:
return SESSION.query(StickersFilters.chat_id).filter(
StickersFilters.chat_id == str(chat_id)).count()
finally:
SESSION.close()
def get_allclearcmd(chat_id):
try:
return SESSION.query(ClearCmd).filter(
ClearCmd.chat_id == str(chat_id)).all()
finally:
SESSION.close()
def num_warns():
try:
return SESSION.query(func.sum(Warns.num_warns)).scalar() or 0
finally:
SESSION.close()
def num_chats():
try:
return SESSION.query(func.count(distinct(Disable.chat_id))).scalar()
finally:
SESSION.close()
def num_warn_filter_chats():
try:
return SESSION.query(func.count(distinct(WarnFilters.chat_id))).scalar()
finally:
SESSION.close()
def num_disabled():
try:
return SESSION.query(Disable).count()
finally:
SESSION.close()