在CMSUser模型上擴展判斷權限的功能 ``` # 注意roles是什么，是上面的反向這個表的，用于通過這個屬性綁定CMSRole模型 # 拿到用戶所有的權限 @property def permissions(self): # 沒有任何權限 if not self.roles: return 0 all_permissions = 0 # 遍歷權限 for role in self.roles: permissions = role.permissions all_permissions |= permissions return all_permissions # 判斷用戶權限 def has_permission(self,permission): # all_permissions = self.permissions # result = all_permissions & permission == permission # return result return self.permissions & permission == permission # 判斷是否為開發者 def is_developer(self): return self.has_permission(CMSPersmission.ALL_PERMISSION) ``` 定義為用戶添加權限的命令 ``` # 將用戶添加到訪問者權限中 # python manage.py add_user_to_role -e 1479852727@qq.com -n 訪問者 # 定義添加角色的命令，不然沒有訪問權限 # 將某用戶添加到某角色中 @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("沒有這個角色:{}".format(name)) else: print("{}郵箱沒有這個用戶".format(email)) # python manage.py test_permission @manager.command def test_permission(): # 查詢用戶 user = CMSUser.query.first() if user.is_developer(): print("擁有訪問者權限") else: print("無訪問者權限") ```