Python函数怎样用装饰器实现函数权限控制 Python函数权限验证装饰器的编写教程​

def check_permission(permission_required):
    def decorator(func):
        def wrapper(*args, **kwargs):
            # 假设有个函数可以检查用户权限
            if not has_permission(permission_required):
                return "权限不足" # 或者抛出一个异常
            return func(*args, **kwargs)
        return wrapper
    return decorator

# 假设的权限检查函数
def has_permission(permission):
    # 这里可以根据实际情况，查询数据库、读取配置文件等
    # 这里简化一下，假设只有'admin'权限才能通过
    current_user_permission = get_current_user_permission()
    return permission in current_user_permission

def get_current_user_permission():
    # 实际情况是从用户session或者其他地方获取
    return ['user'] # 模拟当前用户只有user权限

@check_permission('admin')
def sensitive_data():
    return "只有管理员才能看到的数据"

@check_permission('user')
def public_data():
    return "所有人都能看到的数据"

print(sensitive_data()) # 输出: 权限不足
print(public_data()) # 输出: 所有人都能看到的数据

PermissionDenied

class PermissionDenied(Exception):
    pass

def check_permission(permission_required):
    def decorator(func):
        def wrapper(*args, **kwargs):
            if not has_permission(permission_required):
                raise PermissionDenied("权限不足")
            return func(*args, **kwargs)
        return wrapper
    return decorator

@check_permission('admin')
def sensitive_data():
    return "只有管理员才能看到的数据"

try:
    print(sensitive_data())
except PermissionDenied as e:
    print(e) # 输出: 权限不足

functools.lru_cache