def get_snips(keyword):
try:
return SESSION.query(Snips).get(keyword)
except:
return None
finally:
SESSION.close()
def get_all_snips():
try:
return SESSION.query(Snips).all()
except:
return None
finally:
SESSION.close()
def in_channels(chat_id):
try:
return SESSION.query(fban).filter(fban.chat_id == str(chat_id)).one()
except BaseException:
return None
finally:
SESSION.close()
def already_added(chat_id):
try:
return SESSION.query(Moidata).filter(
Moidata.chat_id == str(chat_id)).one()
except:
return None
finally:
SESSION.close()
def add_snip(keyword, reply, snip_type, media_id, media_access_hash,
media_file_reference):
adder = SESSION.query(Snips).get(keyword)
if adder:
adder.reply = reply
adder.snip_type = snip_type
adder.media_id = media_id
adder.media_access_hash = media_access_hash
adder.media_file_reference = media_file_reference
else:
adder = Snips(keyword, reply, snip_type, media_id, media_access_hash,
media_file_reference)
SESSION.add(adder)
SESSION.commit()
def remove_snip(keyword):
note = SESSION.query(Snips).filter(Snips.snip == keyword)
if note:
note.delete()
SESSION.commit()
def get_all_users():
stark = SESSION.query(Moidata).all()
SESSION.close()
return stark
def add_usersid_in_db(chat_id: int):
id_user = Moidata(str(chat_id))
SESSION.add(id_user)
SESSION.commit()
def get_all_channels():
rem = SESSION.query(fban).all()
SESSION.close()
return rem
def rm_channel(chat_id):
rem = SESSION.query(fban).get(str(chat_id))
if rem:
SESSION.delete(rem)
SESSION.commit()
def add_channel(chat_id):
adder = fban(str(chat_id))
SESSION.add(adder)
SESSION.commit()