BBS论坛（九）

class CMSPermission(object): ALL_PERMISSION = 0b11111111 # 1.访问者的权限 VISITOR = 0b00000001 # 2.管理帖子的权限 POSTER = 0b00000010 # 3.管理评论的权限 COMMENTER = 0b00000100 # 4.管理板块的权限 BOARDER = 0b00001000 # 5.管理前台用户的权限 FRONTUSER = 0b00010000 # 6.管理后台用户的权限 CMSUSER = 0b00100000 # 7.管理后台管理员的权限 ADMINER = 0b01000000 cms_role_user = db.Table( 'cms_role_user', db.Column('cms_role_id',db.Integer,db.ForeignKey('cms_role.id'),primary_key=True), db.Column('cms_user_id',db.Integer,db.ForeignKey('cms_user.id'),primary_key=True) ) class CMSRole(db.Model): __tablename__ = 'cms_role' id = db.Column(db.Integer, primary_key=True, autoincrement=True) name = db.Column(db.String(50), nullable=False) desc = db.Column(db.String(200),nullable=True) create_time = db.Column(db.DateTime,default=datetime.now) permissions = db.Column(db.Integer,default=CMSPermission.VISITOR) users = db.relationship('CMSUser',secondary=cms_role_user,backref='roles')

python manage.py db migrate python manage.py db upgrade

CMSRole = cms_models.CMSRole CMSPermission = cms_models.CMSPermission @manager.command def create_role(): # 1.访问者（可以修改个人信息） visitor = CMSRole(name='访问者',desc='只能访问数据，不能修改') visitor.permissions = CMSPermission.VISITOR # 2.运营人员（修改个人信息，管理帖子，管理评论，管理前台用户） operator = CMSRole(name='运营',desc='管理帖子，管理评论，管理前台用户,') operator.permissions = CMSPermission.VISITOR|CMSPermission.POSTER\ |CMSPermission.COMMENTER|CMSPermission.FRONTUSER # 3.管理员（拥有所有权限） admin = CMSRole(name='管理员',desc='拥有本系统所有权限') admin.permissions = CMSPermission.VISITOR|CMSPermission.POSTER|CMSPermission.CMSUSER\ |CMSPermission.COMMENTER|CMSPermission.FRONTUSER|CMSPermission.BOARDER # 4.开发者 developer = CMSRole(name='开发者',desc='开发人员专用角色') developer.permissions = CMSPermission.ALL_PERMISSION db.session.add_all([visitor,operator,admin,developer]) db.session.commit()

python manage.py create_role

class CMSUser(db.Model): #....... @property def permissions(self): #用户拥有的权限 if not self.roles: return 0 all_permissions = 0 #用户所有的角色 for role in self.roles: #取出用户所有角色的所有权限 permissions = role.permissions #把所有权限通过“|=”整合到all_permissions all_permissions |= permissions return all_permissions def has_permission(self,permission): # 判断用户是否有‘xxx’权限 #通过与操作，判断用户是否有‘permission’；得到的结果相等返回true，不相等返回false return self.permissions&permission == permission @property def is_developer(self): #判断是不是开发者 return self.has_permission(CMSPermission.ALL_PERMISSION)

@manager.option('-e','--email',dest='email') #用户邮箱 @manager.option('-n','--name',dest='name') #角色名字 def add_user_to_role(email,name): '''添加用户到某个角色''' user = CMSUser.query.filter_by(email=email).first() if user: role = CMSRole.query.filter_by(name=name).first() if role: #把用户添加到角色里面 role.users.append(user) db.session.commit() print("用户添加到角色成功!") else: print("没有这个角色：%s" %role) else: print("%s邮箱没有这个用户!"%email) @manager.command def test_permission(): '''测试用户是否有xxx权限''' user = CMSUser.query.first() if user.has_permission(CMSPermission.VISITOR): print("这个用户有访问者权限") else: print("这个用户没有访问者权限")

#添加用户到角色 python manage.py add_user_to_role -e 1184405959@qq.com -n 访问者 #测试 python manage.py test_permission