File size: 889 Bytes
44aada1 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 | try:
from telebot.plugins.sql_helper import SESSION, BASE
except ImportError:
raise Exception("Hello!")
from sqlalchemy import Column, String
class Mute(BASE):
__tablename__ = "mute"
sender = Column(String(14), primary_key=True)
chat_id = Column(String(14), primary_key=True)
def __init__(self, sender, chat_id):
self.sender = str(sender)
self.chat_id = str(chat_id)
Mute.__table__.create(checkfirst=True)
def is_muted(sender, chat_id):
user = SESSION.query(Mute).get((str(sender), str(chat_id)))
if user:
return True
else:
return False
def mute(sender, chat_id):
adder = Mute(str(sender), str(chat_id))
SESSION.add(adder)
SESSION.commit()
def unmute(sender, chat_id):
rem = SESSION.query(Mute).get((str(sender), str(chat_id)))
if rem:
SESSION.delete(rem)
SESSION.commit()
|