def add_note(chat_id, keyword, reply, f_mesg_id):
to_check = get_note(chat_id, keyword)
if not to_check:
adder = Notes(str(chat_id), keyword, reply, f_mesg_id)
SESSION.add(adder)
SESSION.commit()
return True
else:
rem = SESSION.query(Notes).get((str(chat_id), keyword))
SESSION.delete(rem)
SESSION.commit()
adder = Notes(str(chat_id), keyword, reply, f_mesg_id)
SESSION.add(adder)
SESSION.commit()
return False
def add_snip(keyword, reply, f_mesg_id):
to_check = get_snip(keyword)
if not to_check:
adder = Snips(keyword, reply, f_mesg_id)
SESSION.add(adder)
SESSION.commit()
return True
else:
rem = SESSION.query(Snips).filter(Snips.snip == keyword)
SESSION.delete(rem)
SESSION.commit()
adder = Snips(keyword, reply, f_mesg_id)
SESSION.add(adder)
SESSION.commit()
return False
def add_filter(chat_id, keyword, f_mesg_id):
to_check = get_filter(chat_id, keyword)
if not to_check:
adder = Filters(str(chat_id), keyword, f_mesg_id)
SESSION.add(adder)
SESSION.commit()
return True
else:
rem = SESSION.query(Filters).get((str(chat_id), keyword))
SESSION.delete(rem)
SESSION.commit()
adder = Filters(str(chat_id), keyword, f_mesg_id)
SESSION.add(adder)
SESSION.commit()
return False
def addecho(chat_id, user_id, chat_name, user_name, user_username, chat_type):
to_check = is_echo(chat_id, user_id)
if not to_check:
adder = Echos(str(chat_id), str(user_id), chat_name, user_name,
user_username, chat_type)
SESSION.add(adder)
SESSION.commit()
return True
rem = SESSION.query(Echos).get((str(chat_id), str(user_id)))
SESSION.delete(rem)
SESSION.commit()
adder = Echos(str(chat_id), str(user_id), chat_name, user_name,
user_username, chat_type)
SESSION.add(adder)
SESSION.commit()
return False
def add_user_to_bl(chat_id: int, first_name: str, username: str, reason: str,
date: str):
"""add the user to the blacklist"""
to_check = check_is_black_list(chat_id)
if not to_check:
__user = Bot_BlackList(str(chat_id), first_name, username, reason,
date)
SESSION.add(__user)
SESSION.commit()
rem = SESSION.query(Bot_BlackList).get(str(chat_id))
SESSION.delete(rem)
SESSION.commit()
user = Bot_BlackList(str(chat_id), first_name, username, reason, date)
SESSION.add(user)
SESSION.commit()
return True
def add_filter(chat_id, keyword, reply, snip_type, media_id, media_access_hash,
media_file_reference):
to_check = get_filter(chat_id, keyword)
if not to_check:
adder = Filters(str(chat_id), keyword, reply, snip_type, media_id,
media_access_hash, media_file_reference)
SESSION.add(adder)
SESSION.commit()
return True
else:
rem = SESSION.query(Filters).get((str(chat_id), keyword))
SESSION.delete(rem)
SESSION.commit()
adder = Filters(str(chat_id), keyword, reply, snip_type, media_id,
media_access_hash, media_file_reference)
SESSION.add(adder)
SESSION.commit()
return False
def add_snip(keyword, reply, snip_type, media_id, media_access_hash,
media_file_reference):
to_check = get_snip(keyword)
if not to_check:
adder = Snips(keyword, reply, snip_type, media_id, media_access_hash,
media_file_reference)
SESSION.add(adder)
SESSION.commit()
return True
else:
rem = SESSION.query(Snips).filter(Snips.snip == keyword)
SESSION.delete(rem)
SESSION.commit()
adder = Snips(keyword, reply, snip_type, media_id, media_access_hash,
media_file_reference)
SESSION.add(adder)
SESSION.commit()
return False
def add_starter_to_db(
user_id,
first_name,
date,
username,
):
to_check = get_starter_details(user_id)
if not to_check:
user = Bot_Starters(str(user_id), first_name, date, username)
SESSION.add(user)
SESSION.commit()
return True
rem = SESSION.query(Bot_Starters).get(str(user_id))
SESSION.delete(rem)
SESSION.commit()
user = Bot_Starters(str(user_id), first_name, date, username)
SESSION.add(user)
SESSION.commit()
return True
def approve(chat_id, reason):
adder = PMPermit(str(chat_id), str(reason))
SESSION.add(adder)
SESSION.commit()
def remove_phrase(resp, phrase):
rem = SESSION.query(QuickPhrase).get((str(resp),str(phrase)))
if rem:
SESSION.delete(rem)
SESSION.commit()
def update_previous_welcome(chat_id, previous_welcome):
row = SESSION.query(Welcome).get(str(chat_id))
row.previous_welcome = previous_welcome
# commit the changes to the DB
SESSION.commit()
def unmute(sender, chat_id):
rem = SESSION.query(Mute).get((str(sender), str(chat_id)))
if rem:
SESSION.delete(rem)
SESSION.commit()
def remove_all_filters(chat_id):
saved_filter = SESSION.query(Filters).filter(
Filters.chat_id == str(chat_id))
if saved_filter:
saved_filter.delete()
SESSION.commit()
def dissprove(chat_id):
rem = SESSION.query(PMPermit).get(str(chat_id))
if rem:
SESSION.delete(rem)
SESSION.commit()
def gmute(sender):
adder = GMute(str(sender))
SESSION.add(adder)
SESSION.commit()
def delgvar(variable):
rem = SESSION.query(Globals).filter(Globals.variable == str(variable))\
.delete(synchronize_session="fetch")
if rem:
SESSION.commit()
def rm_note(chat_id, keyword):
note = SESSION.query(Notes).filter(Notes.chat_id == str(chat_id), Notes.keyword == keyword)
if note:
note.delete()
SESSION.commit()
def ungmute(sender):
rem = SESSION.query(GMute).get((str(sender)))
if rem:
SESSION.delete(rem)
SESSION.commit()
def rm_all_notes(chat_id):
notes = SESSION.query(Notes).filter(Notes.chat_id == str(chat_id))
if notes:
notes.delete()
SESSION.commit()
def add_filter(chat_id, keyword, reply):
adder = Filters(str(chat_id), keyword, reply)
SESSION.add(adder)
SESSION.commit()
def del_keyword_broadcastlist(keywoard):
with CATBROADCAST_INSERTION_LOCK:
broadcast_group = (SESSION.query(CatBroadcast.keywoard).filter(
CatBroadcast.keywoard == keywoard).delete())
BROADCAST_SQL_.BROADCAST_CHANNELS.pop(keywoard)
SESSION.commit()
def remove_filter(chat_id, keyword):
rem = SESSION.query(Filters).get((str(chat_id), keyword))
if rem:
SESSION.delete(rem)
SESSION.commit()
def approve(chat_id):
adder = PMPermit(str(chat_id))
SESSION.add(adder)
SESSION.commit()
def addecho(user_id, chat_id):
adder = ECHOSQL(str(user_id), str(chat_id))
SESSION.add(adder)
SESSION.commit()
def mute(sender, chat_id):
adder = Mute(str(sender), str(chat_id))
SESSION.add(adder)
SESSION.commit()
def remove_echo(user_id, chat_id):
note = SESSION.query(ECHOSQL).get((str(user_id), str(chat_id)))
if note:
SESSION.delete(note)
SESSION.commit()
def remove_filter(chat_id, keyword):
saved_filter = SESSION.query(Filters).get((str(chat_id), keyword))
if saved_filter:
SESSION.delete(saved_filter)
SESSION.commit()
def fizgban(chat_id, reason):
adder = GBan(str(chat_id), str(reason))
SESSION.add(adder)
SESSION.commit()
def clear_credentials(user):
saved_credentials = SESSION.query(GoogleDriveCreds).get(user)
if saved_credentials:
SESSION.delete(saved_credentials)
SESSION.commit()
return True
def fizungban(chat_id):
rem = SESSION.query(GBan).get(str(chat_id))
if rem:
SESSION.delete(rem)
SESSION.commit()
