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 rm_note(chat_id, note_name):
with NOTES_INSERTION_LOCK:
note = SESSION.query(Notes).get((str(chat_id), note_name))
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:
r = SESSION.query(RulesF).get(fed_id)
if r:
print("fund prev")
SESSION.delete(r)
r = RulesF(str(fed_id), rules)
SESSION.add(r)
try:
SESSION.commit()
return r
except:
SESSION.rollback()
return False
finally:
SESSION.commit()
def fban_user(fed_id, user_id, reason):
with FEDS_LOCK:
r = SESSION.query(BansF).all()
for I in r:
print("1")
if I.fed_id == fed_id:
print("2")
print(I.user_id, user_id)
if int(I.user_id) == int(user_id):
print("fund prev")
SESSION.delete(I)
r = BansF(str(fed_id), user_id, reason)
SESSION.add(r)
SESSION.commit()
return r
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 set_cleanbt(chat_id, is_enable):
with INSERTION_LOCK:
curr = SESSION.query(CleanerBlueText).get(str(chat_id))
if curr:
SESSION.delete(curr)
curr = CleanerBlueText(str(chat_id), is_enable)
if is_enable:
if str(chat_id) not in CLEANER_BT_CHATS:
CLEANER_BT_CHATS.append(str(chat_id))
else:
if str(chat_id) in CLEANER_BT_CHATS:
CLEANER_BT_CHATS.remove(str(chat_id))
SESSION.add(curr)
SESSION.commit()
def disabledel_set(chat_id, is_enable):
with DISABLEDEL_INSERTION_LOCK:
curr = SESSION.query(DisableDelete).get(str(chat_id))
if curr:
SESSION.delete(curr)
curr = DisableDelete(str(chat_id), is_enable)
if is_enable:
if str(chat_id) not in DISABLEDEL:
DISABLEDEL.append(str(chat_id))
else:
if str(chat_id) in DISABLEDEL:
DISABLEDEL.remove(str(chat_id))
SESSION.add(curr)
SESSION.commit()
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_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) # 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, str(fed_members)) SESSION.merge(fed) SESSION.commit() return True
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 set_cleanbt(chat_id, is_enable):
with CLEANER_CHAT_SETTINGS:
curr = SESSION.query(CleanerBlueTextChatSettings).get(str(chat_id))
if not curr:
curr = CleanerBlueTextChatSettings(str(chat_id), is_enable)
else:
curr.is_enabled = is_enable
if str(chat_id) not in CLEANER_CHATS:
CLEANER_CHATS.setdefault(str(chat_id), {
"setting": False,
"commands": set()
})
CLEANER_CHATS[str(chat_id)]["setting"] = is_enable
SESSION.add(curr)
SESSION.commit()
def un_fban_user(fed_id, user_id):
with FEDS_LOCK:
r = SESSION.query(BansF).all()
for I in r:
print("1")
if I.fed_id == fed_id:
print("2")
print(I.user_id, user_id)
if int(I.user_id) == int(user_id):
print("fund prev")
SESSION.delete(I)
try:
SESSION.commit()
return I
except:
SESSION.rollback()
return False
finally:
SESSION.commit()
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 fban_user(fed_id, user_id, first_name, last_name, user_name, reason): 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) SESSION.add(r) try: SESSION.commit() except: SESSION.rollback() return False finally: SESSION.commit() __load_all_feds_banned() return r
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']
# 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, str({'owner': str(owner_id), 'members': str(members)}))
SESSION.merge(fed)
SESSION.commit()
__load_all_feds_chats()
return True
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 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_warn(user_id, chat_id):
with WARN_INSERTION_LOCK:
removed = False
warned_user = SESSION.query(Warns).get((user_id, str(chat_id)))
temp_reason = []
if warned_user and warned_user.num_warns > 0:
warned_user.num_warns -= 1
if warned_user and warned_user.reasons is not None:
for reason in warned_user.reasons:
temp_reason.append(reason)
del temp_reason[-1]
warned_user.reasons = temp_reason
SESSION.add(warned_user)
SESSION.commit()
removed = True
SESSION.close()
return removed
def add_filter(chat_id,
keyword,
reply,
is_sticker=False,
is_document=False,
is_image=False,
is_audio=False,
is_voice=False,
is_video=False,
buttons=None):
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, is_sticker,
is_document, is_image, is_audio, is_voice,
is_video, bool(buttons))
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 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 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 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 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 == "videonote":
curr_perm.videonote = 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 warn_user(user_id, chat_id, reason=None):
with WARN_INSERTION_LOCK:
warned_user = SESSION.query(Warns).get((user_id, str(chat_id)))
if not warned_user:
warned_user = Warns(user_id, str(chat_id))
warned_user.num_warns += 1
if reason is "":
reason = "No reason given."
if reason:
if warned_user.reasons is None:
warned_user.reasons = [reason]
else:
warned_user.reasons = warned_user.reasons + [reason] # TODO:: double check this Daan: Not really wizardry, it adds a new entry to a list/array which can be done this way, basically append equivalent
reasons = warned_user.reasons
num = warned_user.num_warns
SESSION.add(warned_user)
SESSION.commit()
return num, reasons
def migrate_chat(old_chat_id, new_chat_id):
with WARN_INSERTION_LOCK:
chat_notes = SESSION.query(Warns).filter(Warns.chat_id == str(old_chat_id)).all()
for note in chat_notes:
note.chat_id = str(new_chat_id)
SESSION.commit()
with WARN_FILTER_INSERTION_LOCK:
chat_filters = SESSION.query(WarnFilters).filter(WarnFilters.chat_id == str(old_chat_id)).all()
for filt in chat_filters:
filt.chat_id = str(new_chat_id)
SESSION.commit()
WARN_FILTERS[str(new_chat_id)] = WARN_FILTERS[str(old_chat_id)]
del WARN_FILTERS[str(old_chat_id)]
with WARN_SETTINGS_LOCK:
chat_settings = SESSION.query(WarnSettings).filter(WarnSettings.chat_id == str(old_chat_id)).all()
for setting in chat_settings:
setting.chat_id = str(new_chat_id)
SESSION.commit()
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 == 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()
FEDERATION_BANNED_USERID.pop(fed_id)
FEDERATION_BANNED_FULL.pop(fed_id)
# Delete from database
curr = SESSION.query(Federations).get(fed_id)
if curr:
SESSION.delete(curr)
SESSION.commit()
return True
def ensure_bot_in_db():
with INSERTION_LOCK:
bot = Users(BotID, BotUsername)
SESSION.merge(bot)
SESSION.commit()
def ensure_bot_in_db():
with INSERTION_LOCK:
bot = Users(dispatcher.bot.id, dispatcher.bot.username)
SESSION.merge(bot)
SESSION.commit()
def add_note_button_to_db(chat_id, note_name, b_name, url, same_line):
with BUTTONS_INSERTION_LOCK:
button = Buttons(chat_id, note_name, b_name, url, same_line)
SESSION.add(button)
SESSION.commit()
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()
