def set_afk(user_id, reason=""):
with INSERTION_LOCK:
curr = SESSION.query(AFK).get(user_id)
if not curr:
curr = AFK(user_id, reason, True)
else:
curr.is_afk = True
curr.reason = reason
AFK_USERS[user_id] = reason
SESSION.add(curr)
SESSION.commit()
def set_flood(chat_id, amount):
with INSERTION_FLOOD_LOCK:
flood = SESSION.query(FloodControl).get(str(chat_id))
if not flood:
flood = FloodControl(str(chat_id))
flood.user_id = None
flood.limit = amount
CHAT_FLOOD[str(chat_id)] = (None, DEF_COUNT, amount)
SESSION.add(flood)
SESSION.commit()
def set_human_checks(user_id, chat_id):
with INSERTION_LOCK:
human_check = SESSION.query(WelcomeMuteUsers).get((user_id, str(chat_id)))
if not human_check:
human_check = WelcomeMuteUsers(user_id, str(chat_id), True)
else:
human_check.human_check = True
SESSION.add(human_check)
SESSION.commit()
return human_check
def migrate_chat(old_chat_id, new_chat_id):
with NOTES_INSERTION_LOCK:
chat_notes = (SESSION.query(Notes).filter(
Notes.chat_id == str(old_chat_id)).all())
for note in chat_notes:
note.chat_id = str(new_chat_id)
with BUTTONS_INSERTION_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 __load_chat_warn_filters():
global WARN_FILTERS
try:
chats = SESSION.query(WarnFilters.chat_id).distinct().all()
for (chat_id,) in chats: # remove tuple by ( ,)
WARN_FILTERS[chat_id] = []
all_filters = SESSION.query(WarnFilters).all()
for x in all_filters:
WARN_FILTERS[x.chat_id] += [x.keyword]
WARN_FILTERS = {x: sorted(set(y), key=lambda i: (-len(i), i)) for x, y in WARN_FILTERS.items()}
finally:
SESSION.close()
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 subs_fed(fed_id, my_fed):
check = get_spec_subs(fed_id, my_fed)
if check:
return False
with FEDS_SUBSCRIBER_LOCK:
subsfed = FedSubs(fed_id, my_fed)
SESSION.merge(subsfed) # merge to avoid duplicate key issues
SESSION.commit()
global FEDS_SUBSCRIBER
if FEDS_SUBSCRIBER.get(fed_id, set()) == set():
FEDS_SUBSCRIBER[fed_id] = {my_fed}
else:
FEDS_SUBSCRIBER.get(fed_id, set()).add(my_fed)
return True
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 __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 == 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 chat_join_fed(fed_id, chat_name, chat_id):
with FEDS_LOCK:
global FEDERATION_CHATS, FEDERATION_CHATS_BYID
r = ChatF(chat_id, chat_name, fed_id)
SESSION.add(r)
FEDERATION_CHATS[str(chat_id)] = {
"chat_name": chat_name,
"fid": fed_id
}
checkid = FEDERATION_CHATS_BYID.get(fed_id)
if checkid == None:
FEDERATION_CHATS_BYID[fed_id] = []
FEDERATION_CHATS_BYID[fed_id].append(str(chat_id))
SESSION.commit()
return r
def del_user(user_id):
with INSERTION_LOCK:
curr = SESSION.query(Users).get(user_id)
if curr:
SESSION.delete(curr)
SESSION.commit()
return True
ChatMembers.query.filter(ChatMembers.user == user_id).delete()
SESSION.commit()
SESSION.close()
return False
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.add(chat)
SESSION.flush()
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.add(member)
SESSION.commit()
def new_fed(owner_id, fed_name, fed_id):
with FEDS_LOCK:
global FEDERATION_BYOWNER, FEDERATION_BYFEDID, FEDERATION_BYNAME
fed = Federations(
str(owner_id),
fed_name,
str(fed_id),
"Rules is not set in this federation.",
None,
str({
"owner": str(owner_id),
"members": "[]"
}),
)
SESSION.add(fed)
SESSION.commit()
FEDERATION_BYOWNER[str(owner_id)] = {
"fid": str(fed_id),
"fname": fed_name,
"frules": "Rules is not set in this federation.",
"flog": None,
"fusers": str({
"owner": str(owner_id),
"members": "[]"
}),
}
FEDERATION_BYFEDID[str(fed_id)] = {
"owner": str(owner_id),
"fname": fed_name,
"frules": "Rules is not set in this federation.",
"flog": None,
"fusers": str({
"owner": str(owner_id),
"members": "[]"
}),
}
FEDERATION_BYNAME[fed_name] = {
"fid": str(fed_id),
"owner": str(owner_id),
"frules": "Rules is not set in this federation.",
"flog": None,
"fusers": str({
"owner": str(owner_id),
"members": "[]"
}),
}
return fed
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 get_blacklist_setting(chat_id):
try:
setting = CHAT_BLSTICK_BLACKLISTS.get(str(chat_id))
if setting:
return setting['blacklist_type'], setting['value']
else:
return 1, "0"
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 new_fed(owner_id, fed_name, fed_id):
with FEDS_LOCK:
global FEDERATION_BYOWNER, FEDERATION_BYFEDID, FEDERATION_BYNAME
fed = Federations(str(owner_id), fed_name, str(fed_id), 'Rules is not set in this federation.', None,
str({'owner': str(owner_id), 'members': '[]'}))
SESSION.add(fed)
SESSION.commit()
FEDERATION_BYOWNER[str(owner_id)] = (
{'fid': str(fed_id), 'fname': fed_name, 'frules': 'Rules is not set in this federation.', 'flog': None,
'fusers': str({'owner': str(owner_id), 'members': '[]'})})
FEDERATION_BYFEDID[str(fed_id)] = (
{'owner': str(owner_id), 'fname': fed_name, 'frules': 'Rules is not set in this federation.', 'flog': None,
'fusers': str({'owner': str(owner_id), 'members': '[]'})})
FEDERATION_BYNAME[fed_name] = (
{'fid': str(fed_id), 'owner': str(owner_id), 'frules': 'Rules is not set in this federation.', 'flog': None,
'fusers': str({'owner': str(owner_id), 'members': '[]'})})
return fed
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 __migrate_filters():
try:
all_filters = SESSION.query(CustomFilters).distinct().all()
for x in all_filters:
if x.is_document:
file_type = Types.DOCUMENT
elif x.is_image:
file_type = Types.PHOTO
elif x.is_video:
file_type = Types.VIDEO
elif x.is_sticker:
file_type = Types.STICKER
elif x.is_audio:
file_type = Types.AUDIO
elif x.is_voice:
file_type = Types.VOICE
else:
file_type = Types.TEXT
print(str(x.chat_id), x.keyword, x.reply, file_type.value)
if file_type == Types.TEXT:
filt = CustomFilters(str(x.chat_id), x.keyword, x.reply,
file_type.value, None)
else:
filt = CustomFilters(str(x.chat_id), x.keyword, None,
file_type.value, x.reply)
SESSION.add(filt)
SESSION.commit()
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_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 chat_leave_fed(chat_id):
with FEDS_LOCK:
global FEDERATION_CHATS, FEDERATION_CHATS_BYID
# Set variables
fed_info = FEDERATION_CHATS.get(str(chat_id))
if fed_info == None:
return False
fed_id = fed_info["fid"]
# Delete from cache
FEDERATION_CHATS.pop(str(chat_id))
FEDERATION_CHATS_BYID[str(fed_id)].remove(str(chat_id))
# Delete from db
curr = SESSION.query(ChatF).all()
for U in curr:
if int(U.chat_id) == int(chat_id):
SESSION.delete(U)
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 STICKSET_FILTER_INSERTION_LOCK:
global CHAT_BLSTICK_BLACKLISTS
curr_setting = SESSION.query(StickerSettings).get(str(chat_id))
if not curr_setting:
curr_setting = StickerSettings(chat_id,
blacklist_type=int(blacklist_type),
value=value)
curr_setting.blacklist_type = int(blacklist_type)
curr_setting.value = str(value)
CHAT_BLSTICK_BLACKLISTS[str(chat_id)] = {
'blacklist_type': int(blacklist_type),
'value': value
}
SESSION.add(curr_setting)
SESSION.commit()
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 rem_chat(chat_id):
with INSERTION_LOCK:
chat = SESSION.query(Chats).get(str(chat_id))
if chat:
SESSION.delete(chat)
SESSION.commit()
else:
SESSION.close()
def set_welcome_mutes(chat_id, welcomemutes):
with WM_LOCK:
prev = SESSION.query(WelcomeMute).get((str(chat_id)))
if prev:
SESSION.delete(prev)
welcome_m = WelcomeMute(str(chat_id), welcomemutes)
SESSION.add(welcome_m)
SESSION.commit()
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 set_flood_strength(chat_id, flood_type, value):
# for flood_type
# 1 = ban
# 2 = kick
# 3 = mute
# 4 = tban
# 5 = tmute
with INSERTION_FLOOD_SETTINGS_LOCK:
curr_setting = SESSION.query(FloodSettings).get(str(chat_id))
if not curr_setting:
curr_setting = FloodSettings(chat_id,
flood_type=int(flood_type),
value=value)
curr_setting.flood_type = int(flood_type)
curr_setting.value = str(value)
SESSION.add(curr_setting)
SESSION.commit()
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, 'time': x.time}
finally:
SESSION.close()
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
