Flask权限控制：如何在索引页仅展示当前用户可访问路由

当然可行！结合你已有的requires_admin装饰器，我们可以通过给路由标记权限属性的方式，轻松在索引页过滤出当前用户能访问的路由。下面以Python Web框架（比如Flask）为例，分享两种实用的实现思路：

方法1：手动给视图函数添加权限标记 #

在定义路由时，给需要管理员权限的视图函数添加一个自定义属性（比如is_admin_only），后续遍历路由时通过这个属性判断权限：

# 管理员专属路由示例
@app.route('/admin/settings')
@requires_admin
def admin_settings():
    # 给视图函数添加权限标记
    admin_settings.is_admin_only = True
    return "Admin Settings Page"

# 普通用户可访问的路由示例
@app.route('/user/profile')
def user_profile():
    return "User Profile Page"

然后在索引页的视图中，遍历所有注册路由，结合当前用户角色过滤：

@app.route('/')
def index():
    current_user_is_admin = current_user.is_admin  # 假设你有判断用户角色的逻辑
    accessible_routes = []
    
    # 遍历所有路由规则
    for rule in app.url_map.iter_rules():
        # 跳过静态文件路由（可选）
        if rule.rule.startswith('/static'):
            continue
        
        # 获取路由对应的视图函数
        view_func = app.view_functions[rule.endpoint]
        # 检查该路由是否需要管理员权限（默认非管理员路由）
        requires_admin_access = getattr(view_func, 'is_admin_only', False)
        
        # 根据用户角色筛选路由
        if not requires_admin_access or current_user_is_admin:
            accessible_routes.append({
                'path': rule.rule,
                'name': rule.endpoint.replace('_', ' ').title()
            })
    
    return render_template('index.html', routes=accessible_routes)

方法2：优化装饰器自动添加权限标记 #

手动加属性有点繁琐，我们可以修改requires_admin装饰器，让它自动给被装饰的视图函数打上权限标记，这样就不用每次定义路由都手动加属性了：

from functools import wraps
from flask import abort, current_user

def requires_admin(f):
    @wraps(f)
    def decorated_function(*args, **kwargs):
        if not current_user.is_admin:
            abort(403)  # 无权限时返回403或跳转页面
        return f(*args, **kwargs)
    
    # 自动给装饰后的函数添加管理员权限标记
    decorated_function.is_admin_only = True
    return decorated_function

之后所有用@requires_admin装饰的路由，都会自动带上is_admin_only=True的属性，索引页的过滤逻辑和方法1完全一致，不用做任何修改。

额外注意事项 #

• 动态路由（比如/user/<int:user_id>）可以直接按规则展示，或者根据需求格式化后再显示给用户；
• 如果你的项目用了权限管理扩展（比如Flask-Security），可以直接用扩展提供的API来获取权限对应的路由，不用自己实现标记逻辑；
• 记得过滤掉内部路由（比如Flask默认的静态文件路由），避免在仪表盘展示不必要的链接。

内容的提问来源于stack exchange，提问作者Laurent Meyer