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 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 set_frules(fed_id, rules):
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_members = getfed['fusers']
fed_rules = str(rules)
fed_log = getfed['flog']
# Set user
FEDERATION_BYOWNER[str(owner_id)]['frules'] = fed_rules
FEDERATION_BYFEDID[str(fed_id)]['frules'] = fed_rules
FEDERATION_BYNAME[fed_name]['frules'] = fed_rules
# Set on database
fed = Federations(str(owner_id), fed_name, str(fed_id), fed_rules,
fed_log, str(fed_members))
SESSION.merge(fed)
SESSION.commit()
return True
def get_allwarns(chat_id):
get = SESSION.query(Warns).all()
allwarns = []
for x in get:
if x.chat_id == str(chat_id) and x.num_warns > 0:
allwarns.append({
"user_id": x.user_id,
'warns': x.num_warns,
'reasons': x.reasons
})
return allwarns
def new_add_filter(chat_id, keyword, reply_text, file_type, file_id, buttons):
global CHAT_FILTERS
if buttons is None:
buttons = []
with CUST_FILT_LOCK:
prev = SESSION.query(CustomFilters).get((str(chat_id), keyword))
if prev:
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(prev)
filt = CustomFilters(str(chat_id),
keyword,
reply="there is should be a new reply",
is_sticker=False,
is_document=False,
is_image=False,
is_audio=False,
is_voice=False,
is_video=False,
has_buttons=bool(buttons),
reply_text=reply_text,
file_type=file_type.value,
file_id=file_id)
if keyword not in CHAT_FILTERS.get(str(chat_id), []):
CHAT_FILTERS[str(chat_id)] = sorted(
CHAT_FILTERS.get(str(chat_id), []) + [keyword],
key=lambda x: (-len(x), x))
SESSION.add(filt)
SESSION.commit()
for b_name, url, same_line in buttons:
add_note_button_to_db(chat_id, keyword, b_name, url, same_line)
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 private_rules(chat_id, is_private):
with PR_INSERTION_LOCK:
curr = SESSION.query(PrivateRules).get(str(chat_id))
if curr:
SESSION.delete(curr)
curr = PrivateRules(str(chat_id), is_private)
SESSION.add(curr)
SESSION.commit()
def private_note(chat_id, is_private, is_delete):
with PMNOTE_INSERTION_LOCK:
curr = SESSION.query(PrivateNote).get(str(chat_id))
if curr:
SESSION.delete(curr)
curr = PrivateNote(str(chat_id), is_private, is_delete)
SESSION.add(curr)
SESSION.commit()
def clear_user_bio(user_id):
with INSERTION_LOCK:
curr = SESSION.query(UserBio).get(user_id)
if curr:
SESSION.delete(curr)
SESSION.commit()
return True
SESSION.close()
return False
def rm_from_timeout(chat_id, user_id):
with TO_LOCK:
user_filt = SESSION.query(WelcomeTimeout).get((str(chat_id), user_id))
if user_filt:
SESSION.delete(user_filt)
SESSION.commit()
return True
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 __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 == 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 == 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 == 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 __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 == 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 == None:
FEDERATION_CHATS_BYID[x.fed_id] = []
FEDERATION_CHATS_BYID[x.fed_id].append(x.chat_id)
finally:
SESSION.close()
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']
# 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,
str({
'owner': str(owner_id),
'members': str(members)
}))
SESSION.merge(fed)
SESSION.commit()
return True
def set_blacklist_strength(chat_id, blacklist_type, value):
# for blacklist_type
# 0 = nothing
# 1 = delete
# 2 = warn
# 3 = mute
# 4 = kick
# 5 = ban
# 6 = tban
# 7 = tmute
with BLACKLIST_SETTINGS_INSERTION_LOCK:
global CHAT_SETTINGS_BLACKLISTS
curr_setting = SESSION.query(BlacklistSettings).get(str(chat_id))
if not curr_setting:
curr_setting = BlacklistSettings(chat_id, blacklist_type=int(blacklist_type), value=value)
curr_setting.blacklist_type = int(blacklist_type)
curr_setting.value = str(value)
CHAT_SETTINGS_BLACKLISTS[str(chat_id)] = {'blacklist_type': int(blacklist_type), 'value': value}
SESSION.add(curr_setting)
SESSION.commit()
def set_welcome_security(chat_id, security, extra_verify, mute_time, timeout,
timeout_mode, custom_text):
with WS_LOCK:
curr_setting = SESSION.query(WelcomeSecurity).get((str(chat_id)))
if not curr_setting:
curr_setting = WelcomeSecurity(chat_id,
security=security,
extra_verify=extra_verify,
mute_time=mute_time,
timeout=timeout,
timeout_mode=timeout_mode,
custom_text=custom_text)
curr_setting.security = bool(security)
curr_setting.extra_verify = bool(extra_verify)
curr_setting.mute_time = str(mute_time)
curr_setting.timeout = str(timeout)
curr_setting.timeout_mode = int(timeout_mode)
curr_setting.custom_text = str(custom_text)
SESSION.add(curr_setting)
SESSION.commit()
def update_lock(chat_id, lock_type, locked):
with PERM_LOCK:
curr_perm = SESSION.query(Permissions).get(str(chat_id))
if not curr_perm:
curr_perm = init_permissions(chat_id)
if lock_type == "audio":
curr_perm.audio = locked
elif lock_type == "voice":
curr_perm.voice = locked
elif lock_type == "contact":
curr_perm.contact = locked
elif lock_type == "video":
curr_perm.video = locked
elif lock_type == "document":
curr_perm.document = locked
elif lock_type == "photo":
curr_perm.photo = locked
elif lock_type == "sticker":
curr_perm.sticker = locked
elif lock_type == "gif":
curr_perm.gif = locked
elif lock_type == 'url':
curr_perm.url = locked
elif lock_type == 'bots':
curr_perm.bots = locked
elif lock_type == 'forward':
curr_perm.forward = locked
elif lock_type == 'game':
curr_perm.game = locked
elif lock_type == 'location':
curr_perm.location = locked
elif lock_type == 'rtl':
curr_perm.rtl = locked
elif lock_type == 'button':
curr_perm.button = locked
SESSION.add(curr_perm)
SESSION.commit()
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_join_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
members = eval(eval(getfed['fusers'])['members'])
members.append(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()
__load_all_feds_chats()
return True
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 == 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 == 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
}
finally:
SESSION.close()
def get_fban_user(fed_id, user_id):
list_fbanned = FEDERATION_BANNED_USERID.get(fed_id)
if list_fbanned == None:
FEDERATION_BANNED_USERID[fed_id] = []
if user_id in FEDERATION_BANNED_USERID[fed_id]:
r = SESSION.query(BansF).all()
reason = None
for I in r:
if I.fed_id == fed_id:
if int(I.user_id) == int(user_id):
reason = I.reason
return True, reason
else:
return False, None
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
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_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 rm_from_userlist(chat_id, user_id):
with UR_LOCK:
user_filt = SESSION.query(UserRestrict).get((str(chat_id), user_id))
if user_filt:
if user_id in CHAT_USERRESTRICT.get(str(chat_id)): # sanity check
CHAT_USERRESTRICT.get(str(chat_id)).pop(user_id)
SESSION.delete(user_filt)
SESSION.commit()
return True
SESSION.close()
return False
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_url_from_blacklist(chat_id, domain):
with URL_BLACKLIST_FILTER_INSERTION_LOCK:
domain_filt = SESSION.query(URLBlackListFilters).get(
(str(chat_id), domain))
if domain_filt:
if domain in CHAT_URL_BLACKLISTS.get(str(chat_id), set()):
CHAT_URL_BLACKLISTS.get(str(chat_id), set()).remove(domain)
SESSION.delete(domain_filt)
SESSION.commit()
return True
SESSION.close()
return False
