python装饰器(Decorator)

装饰器是一种设计模式,它提供了强大的复用非业务逻辑的能力,经典的应用场景有日志记录、性能测试、事务处理等。
下面分几个部分逐步介绍python中的装饰器。

1、python如何理解函数

在介绍装饰器之前,先介绍一下函数。理解python中的函数,首先要建立这样认知:在python中,一切皆对象。变量、列表、字典,甚至函数、类、元类等都是对象。

这里讲的函数作为第一类对象,可以将它:

  • 赋值给一个变量或存入其他数据结构,如列表、字典等
  • 作为参数窜地给其他函数
  • 作为函数的返回值
>>> def foo():
...     i = 1
...     print i
... 
>>> a = foo
>>> a
<function foo at 0x100700320>
>>> b = [str, foo]
>>> b
[<type 'str'>, <function foo at 0x100700320>]
>>> def bar(func):
...     return func
... 
>>> c = bar(foo)
>>> c
<function foo at 0x100700320>
>>> c()
1
>>>

另外,实现了__call__方法类的实例也可以像函数一样使用。

>>> class Bar:
...     def __init__(self, i):
...         self.i = i
...     def __call__(self, j):
...         return self.i + j
... 
>>> foo = Bar(1)
>>> foo(2)
3

建立了以上的认知之后,我们来介绍装饰器。

2、装饰器其实是一种闭包

前面介绍过,闭包是一种引用了自由变量的函数。

>>> def print_msg(msg):
...     def printer():
...         print('start....')
...         print(msg)
...         print('end....')
...     return printer
... 
>>> another_print = print_msg('hello')
>>> another_print()
start....
hello
end....
>>> 

那装饰器呢?

>>> def print_msg(func):
...     def printer():
...         print('starting...')
...         func()
...         print('end....')
...     return printer
... 
>>> def foo():
...     print('hello')
... 
>>> another_print = print_msg(foo)
>>> another_print()
starting...
hello
end....

基本没有修改什么逻辑,只是把参数替换成了函数,就变成了一个简单的装饰器。所以,装饰器其实是一种以函数作为参数的闭包。这个print_msg就是装饰器,它封装一些可复用的非业务逻辑,而被装饰的foo可以专注于业务逻辑。这样整个设计就非常清晰,而且增加了可读性。

3、装饰器的两种类型

装饰器有两种类型:函数和类。
函数装饰器,就像上面例子中一样,是一个函数。但是装饰器不仅可以是函数,还可以是类。类装饰器在灵活性、内聚性、封装型等方面优点更未突出。
前面看过了函数装饰器的例子,下面看一个类装饰器的例子

>>> class Dec:
...     def __init__(self, func):
...         self._func = func
...     def __call__(self):
...         print('start....')
...         self._func()
...         print('end....')
... 
>>> def para():
...     print('hello')
... 
>>> a = Dec(para)
>>> a
<__main__.Dec instance at 0x10071c680>
>>> a()
start....
hello
end....

4、参数设置

4.1装饰器参数

装饰器的使用非常灵活,需要时可以设置必要的参数。

def outer(a):
    def print_msg(func):
        def printer():
            if a == 1:
                print('start...')
                func()
            else:
                func()
        return printer
    return print_msg

def foo():
    print('hello')

another_print = outer(1)
another_print(foo)()


start...
hello
4.2被装饰函数参数

如果被装饰函数需要传入参数,可以在执行该函数的嵌套函数中设置。上面例子中,可以在printer中设置需要的相关参数。

def printer(*args, **kwargs):
    if a == 1:
        print('start....')
        func(*args, **kwargs)
    else:
        func(*args, **kwargs)
        

5、@语法糖

@符号是装饰器的语法糖,将@符号和装饰器名称放在被装饰函数定义的地方,这样在使用函数时不用显示的调用装饰器, 而且这样并不会影响参数设置。

def outer(a):
    def print_msg(func):
        def printer():
            if a == 1:
                print(func.__name__)
                func()
            else:
                func()
        return printer
    return print_msg

@outer(a=1)
def foo():
    print('hello')

foo()


start...
hello

参考目录:

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 203,271评论 5 476
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,275评论 2 380
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 150,151评论 0 336
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,550评论 1 273
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,553评论 5 365
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,559评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,924评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,580评论 0 257
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,826评论 1 297
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,578评论 2 320
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,661评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,363评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,940评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,926评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,156评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,872评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,391评论 2 342

推荐阅读更多精彩内容