class User(db.Model, UserMixin):
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
username = db.Column(db.String(30), nullable=False)
password = db.Column(db.String(30), nullable=False)
icons = db.Column(db.String(100), nullable=True)
___tablename__ = 'user'
class Raca(db.Model):
idRaca = db.Column(db.Integer(), primary_key=True)
nomeRaca = db.Column(db.String())
idStatus = db.Column(db.Integer(), ForeignKey('Status.idStatus'))
descricao = db.Column(db.String())
def __init__(self, **kwargs):
super().__init__(**kwargs)
self.nomeRaca = kwargs.pop('nomeRaca')
self.idStatus = kwargs.pop('idStatus')
self.descricao = kwargs.pop('descricao')
class User(db.Model):
__tablename__ = 'usertable'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(15), unique=True)
username = db.Column(db.String(15), unique=True)
email = db.Column(db.String(50), unique=True)
password = db.Column(db.String(256), unique=True)
class Eleicao(db.Model):
__tablename__ = "Eleicao"
idEleicao = db.Column(db.Integer, primary_key=True)
eleicao = db.Column(db.String())
dataInicio = db.Column(db.String())
dataFinal = db.Column(db.String())
statusEleicao = db.Column(db.Integer())
apuracao = db.Column(db.Integer())
login = db.Column(db.String(),ForeignKey('Pessoa.login'))
pessoa = relationship(Pessoa)
def __init__(self, **kwargs):
super().__init__(**kwargs)
self.eleicao = kwargs.pop('eleicao')
self.dataInicio = (str(datetime.now())).split('.')[0]
self.dataFinal = '0'
self.statusEleicao = 0
self.apuracao = 0
self.login = kwargs.pop('login')
class Pessoa(db.Model):
__tablename__ = "Pessoa"
login = db.Column(db.String(), primary_key=True)
nomePessoa = db.Column(db.String())
tipo = db.Column(db.Integer())
senha = db.Column(db.String(128))
def __init__(self, **kwargs):
super().__init__(**kwargs)
self.login = kwargs.pop('login')
self.nomePessoa = kwargs.pop('nomePessoa')
self.tipo = kwargs.pop('tipo')
self.senha = generate_password_hash(kwargs.pop('senha'))
def set_password(self, password):
self.senha = generate_password_hash(password)
def check_password(self, password):
return check_password_hash(self.senha, password)
def __repr__(self):
return '<User {}>'.format(self.login)
class Resposta(db.Model):
__tablename__ = "Resposta"
idResposta = db.Column(db.Integer, primary_key=True)
resposta = db.Column(db.String())
contadorResposta = db.Column(db.Integer())
idPergunta = db.Column(db.Integer(), ForeignKey('Pergunta.idPergunta'), primary_key=True)
perg = relationship(Pergunta)
def __init__(self, **kwargs):
super().__init__(**kwargs)
self.resposta = kwargs.pop('resposta')
self.contadorResposta = 0
self.idPergunta = kwargs.pop('idPergunta')
class Eleitor(db.Model):
__tablename__ = "Eleitor"
statusVoto = db.Column(db.Integer())
idEleicao = db.Column(db.Integer(), ForeignKey('Eleicao.idEleicao'), primary_key=True)
eleicao = relationship(Eleicao)
login = db.Column(db.String(),ForeignKey('Pessoa.login'),primary_key=True)
pessoa = relationship(Pessoa)
def __init__(self, **kwargs):
super().__init__(**kwargs)
self.statusVoto = 0
self.idEleicao = kwargs.pop('idEleicao')
self.login = kwargs.pop('login')
class User(db.Model, UserMixin):
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(20), unique=True, nullable=False)
email = db.Column(db.String(120), unique=True, nullable=False)
password = db.Column(db.String(60), nullable=False)
def repr(self):
return f"User1('{self.username}','{self.email}')"
def get_reset_token(self, expires_sec=1800):
s = Serializer(app.config['SECRET_KEY'], expires_sec)
return s.dumps({'user_id': self.id}).decode('utf-8')
@staticmethod
def verify_reset_token(token):
s = Serializer(app.config['SECRET_KEY'])
try:
user_id = s.loads(token)['user_id']
except:
return None
return User.query.get(user_id)
class Pergunta(db.Model):
__tablename__ = "Pergunta"
idPergunta = db.Column(db.Integer, primary_key=True)
pergunta = db.Column(db.String())
numMinResposta = db.Column(db.Integer())
numMaxResposta = db.Column(db.Integer())
idEleicao = db.Column(db.Integer(), ForeignKey('Eleicao.idEleicao'))
eleicao = relationship(Eleicao)
def __init__(self, **kwargs):
super().__init__(**kwargs)
self.pergunta = kwargs.pop('pergunta')
self.numMinResposta = kwargs.pop('numMinResposta')
self.numMaxResposta = kwargs.pop('numMaxResposta')
self.idEleicao = kwargs.pop('idEleicao')
class Article(db.Model):
__tablename__ = 'article' # 如果不指定表名，会默认以这个类名的小写为表名
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
title = db.Column(db.String(100), nullable=False)
content = db.Column(db.Text, nullable=False)
