《每天5分钟用Flask搭建一个管理系统》第8章：权限管理

第8章：权限管理

8.1 权限管理的概念

权限管理是控制用户对应用内不同资源访问的机制。它确保用户只能访问他们被授权的资源。

8.2 Flask-Principal扩展的使用

Flask-Principal是一个提供权限管理功能的Flask扩展，它允许您定义角色和权限，并在视图级别进行访问控制。

示例代码：安装Flask-Principal

pip install flask-principal

示例代码：配置Flask-Principal

from flask_principal import Principal, Permission, RoleNeed

principals = Principal(app)

# 定义角色需要
admin_permission = Permission(RoleNeed('admin'))
user_permission = Permission(RoleNeed('user'))

8.3 角色和权限的设置

角色是一组权限的集合。您可以为用户分配角色，从而间接地赋予他们权限。

示例代码：分配角色

from flask_login import current_user

# 假设User模型有一个角色属性
if hasattr(current_user, 'roles'):
    for role in current_user.roles:
        principals.add_permission(RoleNeed(role), current_user)

8.4 访问控制的实现

您可以使用Permission对象来保护路由，确保只有具有相应权限的用户才能访问。

示例代码：使用Permission保护路由

from flask_principal import identity_loaded, UserNeed

@identity_loaded.connect_via(app)
def on_identity_loaded(sender, identity):
    # 假设用户ID存储在session中
    identity.user = current_user.id

    if hasattr(current_user, 'id'):
        identity.provides.add(UserNeed(current_user.id))

    if hasattr(current_user, 'roles'):
        for role in current_user.roles:
            identity.provides.add(RoleNeed(role))

# 保护路由示例
@app.route('/admin')
@admin_permission.require(http_exception=403)
def admin():
    return 'Admin Dashboard'

8.5 动态权限检查

Flask-Principal允许您在模板中执行动态权限检查，显示或隐藏基于用户权限的内容。

示例代码：在模板中使用Permission

{% if current_user.can(admin_permission) %}
    <a href="{{ url_for('admin') }}">Admin Dashboard</a>
{% endif %}

8.6 总结

本章介绍了权限管理的概念，以及如何使用Flask-Principal扩展来设置角色和权限，并实现访问控制。