Python decorator 2: 类

最新推荐文章于 2021-02-02 22:34:13 发布

原创最新推荐文章于 2021-02-02 22:34:13 发布 · 351 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

Python 专栏收录该内容

26 篇文章

订阅专栏

class myDecorator(object):
 
    def __init__(self, fn):
        print "inside myDecorator.__init__()"
        self.fn = fn
 
    def __call__(self):
        self.fn()
        print "inside myDecorator.__call__()"
 
@myDecorator
def aFunction():
    print "inside aFunction()"
 
print "Finished decorating aFunction()"
 
aFunction()
 
# 输出：
# inside myDecorator.__init__()
# Finished decorating aFunction()
# inside aFunction()
# inside myDecorator.__call__()

上面这个示例展示了，用类的方式声明一个decorator。我们可以看到这个类中有两个成员：
1）一个是__init__()，这个方法是在我们给某个函数decorator时被调用，所以，需要有一个fn的参数，也就是被decorator的函数。
2）一个是__call__()，这个方法是在我们调用被decorator函数时被调用的。
上面输出可以看到整个程序的执行顺序。

__init()__可以看成函数式decorator中外层函数

__call()__可以看成函数式decorator中的内层函数

上面的class可以看成以下函数形式

def myDecorator(fn):
    print ("inside myDecorator.__init__()")
    def newDecorator():
        fn();
        print ("inside myDecorator.__call__()")
    return newDecorator

思考：

class makeHtmlTagClass(object):

    def __init__(self, tag, css_class=""):
        self._tag = tag
        self._css_class = " class='{0}'".format(css_class) \
                                       if css_class !="" else ""

    def __call__(self, fn):
        def wrapped(*args, **kwargs):
            return "<" + self._tag + self._css_class+">"  \
                       + fn(*args, **kwargs) + "</" + self._tag + ">"
        return wrapped

@makeHtmlTagClass(tag="b", css_class="bold_css")
@makeHtmlTagClass(tag="i", css_class="italic_css")
def hello(name):
    return "Hello, {}".format(name)

print (hello("Hao Chen"))

结果

<b class='bold_css'><i class='italic_css'>Hello, Hao Chen</i></b>