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_user_num_chats(user_id):
try:
return (SESSION.query(ChatMembers).filter(
ChatMembers.user == int(user_id)).count())
finally:
SESSION.close()
def num_users():
try:
return SESSION.query(Users).count()
finally:
SESSION.close()
def get_name_by_userid(user_id):
try:
return SESSION.query(Users).get(Users.user_id == int(user_id)).first()
finally:
SESSION.close()
def get_all_chats():
try:
return SESSION.query(Chats).all()
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"]
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 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 num_chats():
try:
return SESSION.query(func.count(distinct(
CustomFilters.chat_id))).scalar()
finally:
SESSION.close()
def add_url(tg_chat_id, tg_feed_link, tg_old_entry_link):
with INSERTION_LOCK:
action = RSS(tg_chat_id, tg_feed_link, tg_old_entry_link)
SESSION.add(action)
SESSION.commit()
def add_note_button_to_db(chat_id, keyword, b_name, url, same_line):
with BUTTON_LOCK:
button = Buttons(chat_id, keyword, b_name, url, same_line)
SESSION.add(button)
SESSION.commit()
def num_filters():
try:
return SESSION.query(CustomFilters).count()
finally:
SESSION.close()
def get_filter(chat_id, keyword):
try:
return SESSION.query(CustomFilters).get((str(chat_id), keyword))
finally:
SESSION.close()
def get_all_filters():
try:
return SESSION.query(CustomFilters).all()
finally:
SESSION.close()
def check_afk_status(user_id):
try:
return SESSION.query(AFK).get(user_id)
finally:
SESSION.close()
def is_approved(chat_id, user_id):
try:
return SESSION.query(Approvals).get((str(chat_id), user_id))
finally:
SESSION.close()
def get_urls(tg_chat_id):
try:
return SESSION.query(RSS).filter(RSS.chat_id == tg_chat_id).all()
finally:
SESSION.close()
def del_fed(fed_id):
with FEDS_LOCK:
global FEDERATION_BYOWNER, FEDERATION_BYFEDID, FEDERATION_BYNAME, FEDERATION_CHATS, FEDERATION_CHATS_BYID, FEDERATION_BANNED_USERID, FEDERATION_BANNED_FULL
getcache = FEDERATION_BYFEDID.get(fed_id)
if getcache is None:
return False
# Variables
getfed = FEDERATION_BYFEDID.get(fed_id)
owner_id = getfed["owner"]
fed_name = getfed["fname"]
# Delete from cache
FEDERATION_BYOWNER.pop(owner_id)
FEDERATION_BYFEDID.pop(fed_id)
FEDERATION_BYNAME.pop(fed_name)
if FEDERATION_CHATS_BYID.get(fed_id):
for x in FEDERATION_CHATS_BYID[fed_id]:
delchats = SESSION.query(ChatF).get(str(x))
if delchats:
SESSION.delete(delchats)
SESSION.commit()
FEDERATION_CHATS.pop(x)
FEDERATION_CHATS_BYID.pop(fed_id)
# Delete fedban users
getall = FEDERATION_BANNED_USERID.get(fed_id)
if getall:
for x in getall:
banlist = SESSION.query(BansF).get((fed_id, str(x)))
if banlist:
SESSION.delete(banlist)
SESSION.commit()
if FEDERATION_BANNED_USERID.get(fed_id):
FEDERATION_BANNED_USERID.pop(fed_id)
if FEDERATION_BANNED_FULL.get(fed_id):
FEDERATION_BANNED_FULL.pop(fed_id)
# Delete fedsubs
getall = MYFEDS_SUBSCRIBER.get(fed_id)
if getall:
for x in getall:
getsubs = SESSION.query(FedSubs).get((fed_id, str(x)))
if getsubs:
SESSION.delete(getsubs)
SESSION.commit()
if FEDS_SUBSCRIBER.get(fed_id):
FEDS_SUBSCRIBER.pop(fed_id)
if MYFEDS_SUBSCRIBER.get(fed_id):
MYFEDS_SUBSCRIBER.pop(fed_id)
# Delete from database
curr = SESSION.query(Federations).get(fed_id)
if curr:
SESSION.delete(curr)
SESSION.commit()
return True
def get_all():
try:
return SESSION.query(RSS).all()
finally:
SESSION.close()
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 curr_connection(chat_id):
try:
return SESSION.query(Connection).get((str(chat_id)))
finally:
SESSION.close()
def get_userid_by_name(username):
try:
return (SESSION.query(Users).filter(
func.lower(Users.username) == username.lower()).all())
finally:
SESSION.close()
def add_history_conn(user_id, chat_id, chat_name):
global HISTORY_CONNECT
with CONNECTION_HISTORY_LOCK:
conn_time = int(time.time())
if HISTORY_CONNECT.get(int(user_id)):
counting = (
SESSION.query(ConnectionHistory.user_id)
.filter(ConnectionHistory.user_id == str(user_id))
.count()
)
getchat_id = {}
for x in HISTORY_CONNECT[int(user_id)]:
getchat_id[HISTORY_CONNECT[int(user_id)][x]["chat_id"]] = x
if chat_id in getchat_id:
todeltime = getchat_id[str(chat_id)]
delold = SESSION.query(ConnectionHistory).get(
(int(user_id), str(chat_id))
)
if delold:
SESSION.delete(delold)
HISTORY_CONNECT[int(user_id)].pop(todeltime)
elif counting >= 5:
todel = list(HISTORY_CONNECT[int(user_id)])
todel.reverse()
todel = todel[4:]
for x in todel:
chat_old = HISTORY_CONNECT[int(user_id)][x]["chat_id"]
delold = SESSION.query(ConnectionHistory).get(
(int(user_id), str(chat_old))
)
if delold:
SESSION.delete(delold)
HISTORY_CONNECT[int(user_id)].pop(x)
else:
HISTORY_CONNECT[int(user_id)] = {}
delold = SESSION.query(ConnectionHistory).get((int(user_id), str(chat_id)))
if delold:
SESSION.delete(delold)
history = ConnectionHistory(int(user_id), str(chat_id), chat_name, conn_time)
SESSION.add(history)
SESSION.commit()
HISTORY_CONNECT[int(user_id)][conn_time] = {
"chat_name": chat_name,
"chat_id": str(chat_id),
}
def get_chat_members(chat_id):
try:
return SESSION.query(ChatMembers).filter(
ChatMembers.chat == str(chat_id)).all()
finally:
SESSION.close()
def get_connected_chat(user_id):
try:
return SESSION.query(Connection).get((int(user_id)))
finally:
SESSION.close()
def get_all_users():
try:
return SESSION.query(Users).all()
finally:
SESSION.close()
def num_logchannels():
try:
return SESSION.query(func.count(distinct(GroupLogs.chat_id))).scalar()
finally:
SESSION.close()
def num_chats():
try:
return SESSION.query(Chats).count()
finally:
SESSION.close()
def approve(chat_id, user_id):
with APPROVE_INSERTION_LOCK:
approve_user = Approvals(str(chat_id), user_id)
SESSION.add(approve_user)
SESSION.commit()
def ensure_bot_in_db():
with INSERTION_LOCK:
bot = Users(dispatcher.bot.id, dispatcher.bot.username)
SESSION.merge(bot)
SESSION.commit()
def num_stickers_filters():
try:
return SESSION.query(StickersFilters).count()
finally:
SESSION.close()
