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 add_note_to_db(chat_id,
note_name,
note_data,
msgtype,
buttons=None,
file=None):
if not buttons:
buttons = []
with NOTES_INSERTION_LOCK:
prev = SESSION.query(Notes).get((str(chat_id), note_name))
if prev:
with BUTTONS_INSERTION_LOCK:
prev_buttons = SESSION.query(Buttons).filter(
Buttons.chat_id == str(chat_id),
Buttons.note_name == note_name).all()
for btn in prev_buttons:
SESSION.delete(btn)
SESSION.delete(prev)
note = Notes(str(chat_id),
note_name,
note_data or "",
msgtype=msgtype.value,
file=file)
SESSION.add(note)
SESSION.commit()
for b_name, url, same_line in buttons:
add_note_button_to_db(chat_id, note_name, b_name, url, same_line)
def migrate_chat(old_chat_id, new_chat_id):
with CHAT_LOCK:
chat_notes = SESSION.query(ReportingChatSettings).filter(
ReportingChatSettings.chat_id == str(old_chat_id)).all()
for note in chat_notes:
note.chat_id = str(new_chat_id)
SESSION.commit()
def get_buttons(chat_id, note_name):
try:
return SESSION.query(Buttons).filter(
Buttons.chat_id == str(chat_id),
Buttons.note_name == note_name).order_by(Buttons.id).all()
finally:
SESSION.close()
def migrate_chat(old_chat_id, new_chat_id):
with STICKERS_FILTER_INSERTION_LOCK:
chat_filters = SESSION.query(StickersFilters).filter(
StickersFilters.chat_id == str(old_chat_id)).all()
for filt in chat_filters:
filt.chat_id = str(new_chat_id)
SESSION.commit()
def __load_flood_settings():
global CHAT_FLOOD
try:
all_chats = SESSION.query(FloodControl).all()
CHAT_FLOOD = {chat.chat_id: (None, DEF_COUNT, chat.limit) for chat in all_chats}
finally:
SESSION.close()
def __load_log_channels():
global CHANNELS
try:
all_chats = SESSION.query(GroupLogs).all()
CHANNELS = {chat.chat_id: chat.log_channel for chat in all_chats}
finally:
SESSION.close()
def gkick_reset(user_id):
user = SESSION.query(GloballyKickedUsers).get(user_id)
if user:
user.times = 0
SESSION.delete(user)
SESSION.commit()
__load_gkick_userid_list()
def rem_chat(chat_id):
with INSERTION_LOCK:
autochat = SESSION.query(ChatbotChats).get(str(chat_id))
if autochat:
SESSION.delete(autochat)
SESSION.commit()
def get_user_com_chats(user_id):
try:
chat_members = SESSION.query(ChatMembers).filter(
ChatMembers.user == int(user_id)).all()
return [i.chat for i in chat_members]
finally:
SESSION.close()
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
SESSION.add(curr_perm)
SESSION.commit()
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
def get_buttons(chat_id, keyword):
try:
return SESSION.query(Buttons).filter(
Buttons.chat_id == str(chat_id),
Buttons.keyword == keyword).order_by(Buttons.id).all()
finally:
SESSION.close()
def chat_should_report(chat_id: Union[str, int]) -> bool:
try:
chat_setting = SESSION.query(ReportingChatSettings).get(str(chat_id))
if chat_setting:
return chat_setting.should_report
return False
finally:
SESSION.close()
def get_reason(user_id):
user = SESSION.query(BlacklistUsers).get(str(user_id))
rep = ""
if user:
rep = user.reason
SESSION.close()
return rep
def unblacklist_user(user_id):
with BLACKLIST_LOCK:
user = SESSION.query(BlacklistUsers).get(str(user_id))
if user:
SESSION.delete(user)
SESSION.commit()
__load_blacklist_userid_list()
def user_should_report(user_id: int) -> bool:
try:
user_setting = SESSION.query(ReportingUserSettings).get(user_id)
if user_setting:
return user_setting.should_report
return True
finally:
SESSION.close()
def migrate_chat(old_chat_id, new_chat_id):
with GMUTE_SETTING_LOCK:
chat = SESSION.query(GmuteSettings).get(str(old_chat_id))
if chat:
chat.chat_id = new_chat_id
SESSION.add(chat)
SESSION.commit()
def get_chat_filters(chat_id):
try:
return SESSION.query(CustomFilters).filter(
CustomFilters.chat_id == str(chat_id)).order_by(
func.length(CustomFilters.keyword).desc()).order_by(
CustomFilters.keyword.asc()).all()
finally:
SESSION.close()
def __load_all_feds_settings(): global FEDERATION_NOTIFICATION try: getuser = SESSION.query(FedsUserSettings).all() for x in getuser: FEDERATION_NOTIFICATION[str(x.user_id)] = x.should_report finally: SESSION.close()
def ungmute_user(user_id):
with GMUTED_USERS_LOCK:
user = SESSION.query(GloballyMutedUsers).get(user_id)
if user:
SESSION.delete(user)
SESSION.commit()
__load_gmuted_userid_list()
def get_rules(chat_id):
rules = SESSION.query(Rules).get(str(chat_id))
ret = ""
if rules:
ret = rules.rules
SESSION.close()
return ret
def ungban_user(user_id):
with GBANNED_USERS_LOCK:
user = SESSION.query(GloballyBannedUsers).get(user_id)
if user:
SESSION.delete(user)
SESSION.commit()
__load_gbanned_userid_list()
def remove_url(tg_chat_id, tg_feed_link):
with INSERTION_LOCK:
# this loops to delete any possible duplicates for the same TG User ID, TG Chat ID and link
for row in check_url_availability(tg_chat_id, tg_feed_link):
# add the action to the DB query
SESSION.delete(row)
SESSION.commit()
def update_url(row_id, new_entry_links):
with INSERTION_LOCK:
row = SESSION.query(RSS).get(row_id)
# set the new old_entry_link with the latest update from the RSS Feed
row.old_entry_link = new_entry_links[0]
# commit the changes to the DB
SESSION.commit()
def set_user_me_info(user_id, info):
with INSERTION_LOCK:
userinfo = SESSION.query(UserInfo).get(user_id)
if userinfo:
userinfo.info = info
else:
userinfo = UserInfo(user_id, info)
SESSION.add(userinfo)
SESSION.commit()
def gkick_setvalue(user_id, name, value):
user = SESSION.query(GloballyKickedUsers).get(user_id)
if user:
user.times = value
if not user:
user = GloballyKickedUsers(user_id, name, value)
SESSION.merge(user)
SESSION.commit()
__load_gkick_userid_list()
def migrate_chat(old_chat_id, new_chat_id):
with INSERTION_LOCK:
flood = SESSION.query(FloodControl).get(str(old_chat_id))
if flood:
CHAT_FLOOD[str(new_chat_id)] = CHAT_FLOOD.get(str(old_chat_id), DEF_OBJ)
flood.chat_id = str(new_chat_id)
SESSION.commit()
SESSION.close()
def __load_gmuted_userid_list():
global GMUTED_LIST
try:
GMUTED_LIST = {
x.user_id
for x in SESSION.query(GloballyMutedUsers).all()
}
finally:
SESSION.close()
def __load_gmute_stat_list():
global GMUTESTAT_LIST
try:
GMUTESTAT_LIST = {
x.chat_id
for x in SESSION.query(GmuteSettings).all() if not x.setting
}
finally:
SESSION.close()
