Python 装饰器的典型使用场景（2）

最新推荐文章于 2025-10-28 13:29:00 发布

翻译最新推荐文章于 2025-10-28 13:29:00 发布 · 3.1k 阅读

文章标签：

#Python #装饰器 #decorator

Python 同时被 2 个专栏收录

13 篇文章

订阅专栏

装饰器

2 篇文章

订阅专栏

本文介绍了Python装饰器作为代理和上下文提供者的两种典型应用场景。代理装饰器用于控制函数访问权限，通过验证用户角色实现安全保护；上下文装饰器则确保函数在正确的环境中运行，并在前后执行必要的操作。

Python 装饰器的典型使用场景（2）

前一篇文章中（http://blog.youkuaiyun.com/qq_26886929/article/details/54091962），我们谈了装饰器的两方面用法，本文将讨论装饰器用作代理（Proxy）和上下文提供者（Context Provider）两个用途：

1. 代理（Proxy）装饰器

代理装饰器可被用作标记和注册函数的全局机制。例如，根据当前用户，保护对代码进行访问的安全层，可以使用一个关联着可调用对象所要求的权限的集中检测器来实现：

class User(object):
    def __init__(self, roles):
        self.roles = roles

class Unauthorized(Exception):
    pass

def protect(role):
    def _protect(function):
        def __protect(*args, **kw):
            user = globals().get('user')
            if user is None or role not in user.roles:
                raise Unauthorized("I won't tell you")
            return function(*args, **kw)
        return __protect
    return _protect

这种模式常常被用在 Python 网络框架中，来定义可发布类的安全。例如，Django 使用装饰器来提供函数访问的安全。

下面是个例子，在这个例子中当前用户被保存在一个全局变量中。当方法被调用时，装饰器会检查该用户的角色，以觉得该用户是否有访问权限：

>>> tarek = User(('admin', 'user'))
>>> bill = User(('user',))
>>> class MySecrets(object):
...     @protect('admin')
...     def waffle_recipe(self):
...         print('use tons of butter!')
...
>>> these_are = MySecrets()
>>> user = tarek
>>> these_are.waffle_recipe()
use tons of butter!
>>> user = bill
>>> these_are.waffle_recipe()
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "<stdin>", line 7, in wrap
__main__.Unauthorized: I won't tell you

2. 上下文提供者（Context Provider）

上下文装饰器被用来确保函数在正确的上下文环境运行，或者确保在函数运行前和运行后执行一些代码。换言之，该装饰器进行设置或取消设置某个特定的运行环境。例如，当一条数据要在多个线程之间共享时，就需要使用锁（lock）来确保数据被保护，以防出现并发访问。锁能够像下面这样放在装饰器中：

from threading import RLock
lock = RLock()
def synchronized(function):
    def _synchronized(*args, **kw):
        lock.acquire()
        try:
            return function(*args, **kw)
        finally:
            lock.release()
    return _synchronized

@synchronized
def thread_safe(): # make sure it locks the resource
    pass

上下文装饰器，通常被上下文管理器（with 语句）取代。