1.概述
当我们查询用户信息时候,需要多添加一个用户的权限
2.流程
2.1在用户返回数据成字典时候,添加一个role_name的字段
def to_dict(self):
return {
'id':self.id,
'name': self.name,
'nick_name':self.nick_name,
'phone':self.phone,
'email':self.email,
# 此时的role是根据下面backref的role定义的,获取到他的名称
# 防止角色为空
'role_name': self.role.name if self.role else ''
}
3.Postman测试
4.完整代码展示
# /flask_shop/models.py
# 此文件用于建立数据库表的模型
# 需要针对数据库的模型
from enum import unique
from flask_shop import db
# 对数据加密 检查密码
from werkzeug.security import generate_password_hash,check_password_hash
from datetime import datetime
# 创建公用数据库模型
class BaseModel:
# 记录创建的时间
create_time = db.Column(db.DateTime,default=datetime.now)
# 记录修改密码的时间
update_time = db.Column(db.DateTime, default=datetime.now, onupdate = datetime.now)
# 需要继承数据库中的模型
class User(db.Model,BaseModel):
__tablename__ = 't_user'
id = db.Column(db.Integer, primary_key = True)
name = db.Column(db.String(32), unique=True, nullable=False)
pwd = db.Column(db.String(128))
nick_name = db.Column(db.String(32))
phone = db.Column(db.String(11))
email = db.Column(db.String(32))
# 建立用户与角色的关系(多对一)
rid = db.Column(db.Integer, db.ForeignKey('t_role.id'))
# 定义两个装饰器,用于访问密码时使用
@property
def password(self):
return self.pwd
# 用户存储时使用,存的是加密过后的数据,若要访问,得通过property装饰器
# 用户先输入密码,传入到t_pwd,通过函数的加密返回给self.pwd;用户需要访问时,找到property
# self.pwd,是数据库中的密码(加密过后);t_pwd是用户输入的密码(真实密码)
@password.setter
def password(self,t_pwd):
self.pwd = generate_password_hash(t_pwd)
# 用户二次访问之后,需要把加密过后的密码转成真实密码,与用户输入的密码进行比对
def check_password(self,t_pwd):
return check_password_hash(self.pwd,t_pwd)
def to_dict(self):
return {
'id':self.id,
'name': self.name,
'nick_name':self.nick_name,
'phone':self.phone,
'email':self.email,
# 此时的role是根据下面backref的role定义的,获取到他的名称
# 防止角色为空
'role_name': self.role.name if self.role else ''
}
# 建立菜单的数据库
class Menu(db.Model):
__tablename__ = 't_menu'
id = db.Column(db.Integer, primary_key = True)
name = db.Column(db.String(32),unique = True, nullable = False)
level = db.Column(db.Integer)
path = db.Column(db.String(32))
# pid = db.Column(db.Integer)
pid = db.Column(db.Integer, db.ForeignKey('t_menu.id'))
# 建立一个自连接的数据,用于做子表
children = db.relationship('Menu')
# 由于前端的接收需要给出json数据,不可以直接去获取,所以得单独返回
def to_dict(self):
return {
'id':self.id,
'name': self.name,
'level':self.level,
'path':self.path,
'pid':self.pid,
# 子表中的内容需要单独返回,可以做一个递归或者循环
# 'children': self.get_child_list()
}
def get_child_list(self):
obj_child = self.children
data = []
for o in obj_child:
data.append(o.to_dict)
return data
# 角色管理
class Role(db.Model):
__tablename__ = 't_role'
id = db.Column(db.Integer, primary_key = True)
name = db.Column(db.String(32), unique=True, nullable = True)
desc = db.Column(db.String(32))
# backref反向关联,方便子表查询主表数据
users = db.relationship('User', backref='role')
# menu = db.Column('Menu', secondary = trm)
def to_dict(self):
return {
'id':self.id,
'name': self.name,
'desc':self.desc
}