class User(db.Model):
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
username = db.Column(db.String(20))
password_hash = db.Column(db.String(128))
phone = db.Column(db.String(20))
email = db.Column(db.String(50))
gender = db.Column(db.String(2))
role_id = db.Column(db.Integer)
class Role(db.Model):
__tablename__ = 'roles'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(40))
def __repr__(self):
return "<Role %s>" % self.name
class User(UserMixin, db.Model):
__tablename__ = 'users'
id = db.Column(db.Integer, primary_key=True)
email = db.Column(db.String(64))
username = db.Column(db.String(128), unique=True, index=True)
password_hash = db.Column(db.String(128))
#role_id = db.Column(db.Integer,db.ForeignKey('roles.id'))
#增加一个confirmed字段，如果为True，表示用户已经通过邮件验证
confirmed = db.Column(db.Boolean, default=False)
@property
def password(self):
raise AttributeError('password is not readable attribute')
@password.setter
def password(self, password):
self.password_hash = generate_password_hash(password)
def verify_password(self, password):
return check_password_hash(self.password_hash, password)
def __repr__(self):
return "<User %s>" % self.username
#定义用户账户确认令牌生成函数
def generate_confirmation_token(self, expiration=3600):
s = Serializer(current_app.config['SECRET_KEY'], expiration)
tmp_se = s.dumps({'confirm': self.id}) #生成一个加密的签名
return tmp_se.decode('utf-8') #因为dumps（）方法生成的签名类型为bytes，所以需要解码操作
#定义用户确认令牌函数
def confirm(self, token):
s = Serializer(current_app.config['SECRET_KEY'])
try:
to_json = s.loads(token.encode('utf-8'))
except:
return False
id = to_json.get('confirm')
if id != self.id:
return False
self.confirmed = True
db.session.add(self)
try:
db.session.commit()
except:
db.session.rollback()
return True
#定义重置密码令牌生成函数
def generate_reset_password_token(self, expiration=3600):
s = Serializer(current_app.config['SECRET_KEY'], expiration)
return s.dumps({'reset': self.id}).decode('utf-8')
@staticmethod
def reset_password(token, new_password):
s = Serializer(current_app.config['SECRET_KEY'])
#获取对应用户的id号
try:
data = s.loads(token.encode('utf-8'))
except:
return False
user = User.query.get(data.get('reset'))
if user is None:
return False
user.password = new_password
db.session.add(user)
return True
#定义用户更改邮件令牌函数
def generate_change_email_token(self, new_email, expiration=3600):
s = Serializer(current_app.config['SECRET_KEY'], expiration)
return s.dumps({
'change_email': self.id,
'new_email': new_email
}).decode('utf-8')
def change_email(self, token):
s = Serializer(current_app.config['SECRET_KEY'])
try:
data = s.loads(token.encode('utf-8'))
except:
return False
if data.get('change_email') != self.id:
return False
new_email = data.get('new_email')
if new_email is None:
return False
if self.query.filter_by(email=new_email).first() is not None:
return False
self.email = new_email
db.session.add(self)
return True
class Message(db.Model):
id = db.Column(db.Integer, primary_key=True)
body = db.Column(db.String(200))
name = db.Column(db.String(20))
timestamp = db.Column(db.DateTime, default=datetime.now, index=True)
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)
class Role(db.Model):
id = db.Column(db.Integer, primary_key=True)
rolename = db.Column(db.String(20))
menulist = db.Column(db.String(500))
class Menu(db.Model):
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(20))
parent_id = db.Column(db.Integer)
role_type = db.Column(db.Integer)