python - 用__call__ 实现装饰器功能

问题描述

问 题

用一个类的__call__ 来实现一个装饰器，如下代码

class Check(object):
    def __init__(self, name):
        self.name = name

    def __call__(self, func):

        print "1111111111"

        def _decorator(*args, **kwargs):
            print "2222222222"
            return func(*args, **kwargs)
          

如果把这个类当做装饰器修饰一个函数。

@Check("param")
def param_check(request):
    "python code....."
    return Response("ok")

问题来了。启动一个django或其他任意一个python进程。执行的时候这个param_check路由的时候，print "1111111111" 只是第一次打印，后面再调用的时候就不打印1111111111111。
而 print "2222222222" 每次都打印的。我的理解是第一次 是__call__ 在定义这个装饰器，所以只执行了一次，后面再次请求时已经定义好了，只执行函数体的部分，也就是 print 222222222的部分。 谁知道这个python装饰器的特性

解决方案

其实用类式装饰器时，有个需要注意的地方，有参数和无参数的类式装饰器其实是不一样的

不带参数的类式装饰器

如果创建了一个不带参数的装饰器，被装饰的方法会传递给装饰器的构造器(__init__)，然后在被装饰的函数被调用的时候，装饰器的__call__()方法就会执行。

class Check(object):

    def __init__(self, func):
        self.func = func

    def __call__(self, *args):

        print("111111")
        self.func(*args)
        print("222222")

@Check
def param_check(request):
    print(request)



param_check('hello')
param_check('world')

111111
hello
222222
111111
world
222222

需要注意，在装饰阶段，__init__ 函数执行，在被装饰的方法被调用的时候，__call__ 执行。

带参数的类式装饰器

class Check(object):
    def __init__(self, name):
        self.name = name

    def __call__(self, func):

        print ("1111111111")

        def decorator(*args, **kwargs):
            print ("2222222222")
            return func(*args, **kwargs)
        return decorator

@Check('parm')
def param_check():
    print('Hello')

param_check()
param_check()

输出：

1111111111
2222222222
Hello
2222222222
Hello

可以很明显的看出来， __call__只会在装饰阶段被调用一次

想了解的更详细看这儿： http://www.artima.com/weblogs...

这篇关于python - 用__call__ 实现装饰器功能的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！