python装饰器

装饰器是程序开发中经常会用到的一个功能,功能就是在运行原来功能基础上,加上一些其它功能,比如权限的验证,比如日志的记录等等。不修改原来的代码,进行功能的扩展。
比如java中的动态代理,python的注解装饰器。
其实python的装饰器,是修改了代码。
装饰器的工作原理:

def outer(fn):
    print('...outer...')

    def inner():
        print('...inner...')
        fn()

    return inner


def test():
    print('...test...')

test = outer(test)
print('*'*15)
test()

结果如下:


装饰器原理结果.jpg

装饰器就是引用了闭包的概念。装饰器的工作原理如下:


装饰器原理.jpg

现在采用装饰器:
def outer(fn):
    print('...outer...')

    def inner():
        print('...inner...')
        fn()

    return inner


@outer                       #就相当于 test = outer(test)
def test():
    print('...test...')

test()

输出结果如下:


装饰器结果.jpg

多个装饰器:

def f1(fn):
    def f1_inner():
        return '<b>'+fn()+'</b>'
    return f1_inner

def f2(fn):
    def f2_inner():
        return '<i>'+fn()+'</i>'
    return f2_inner

@f1
@f2
def fun():
    return 'hello world!'

print(fun())

结果如下:

多个装饰器结果.jpg

装饰器主要的功能有:
引入日志
函数执行时间统计
执行函数前预备处理
执行函数后清理处理
权限检验等场景
缓存
之前演示的装饰器都是无参调用的,装饰器的传参是如何实现的:

def f1(fn):
    def f2(a,b):
        print(a,b)
        fn(a,b)
    return f2


@f1
def fun(a, b):
    print(a+b)

fun(10, 20)

输出结果如下:


装饰器传参结果.jpg

这样把参数写死显然不适合装饰器的普遍性,如果有不同参数的函数调用该装饰器,如何改写装饰器:

from time import ctime, sleep


def f1(fn):
    def f2(*args, **kwargs):
        print('%s call at %s'%(fn.__name__, ctime()))
        print(*args, **kwargs)
        fn(*args, **kwargs)
    return f2


@f1
def fun1(a, b):
    print(a+b)
@f1
def fun2(a, b, c):
    print(a+b+c)

fun1(10, 20)
sleep(2)
fun2(30, 40, 50)

结果如下:


不定长传参结果.jpg

以上举得被装饰器装饰过的函数都是直接打印,如果是想返回的值的函数使用该装饰器,那装饰器又不具备普遍性,修改装饰器如下:

from time import ctime, sleep


def f1(fn):
    def f2(*args, **kwargs):
        print('%s call at %s'%(fn.__name__, ctime()))
        return fn(*args, **kwargs)
    return f2


@f1
def fun1(a, b):
    print(a+b)
@f1
def fun2(a, b, c):
    return a+b+c

fun1(10, 20)
sleep(2)
print(fun2(30, 40, 50))

结果如下:

return装饰器结果.jpg

如果闭包中的f2没有return fn(args,kwargs)而是直接fn(args,**kwargs),最后打印得到的结果是None。
装饰器带参数,在原有装饰器的基础上,设置外部变量。

from time import ctime, sleep

def timefun_arg(pre="hello"):
    def timefun(func):
        def wrappedfunc():
            print("%s called at %s %s"%(func.__name__, ctime(), pre))
            return func()
        return wrappedfunc
    return timefun

@timefun_arg("wangcai")
def foo():
    print("I am foo")

@timefun_arg("python")
def too():
    print("I am too")

@timefun_arg()
def coo():
    print("I am coo")

foo()
sleep(2)

too()
sleep(2)

coo()

输出结果如下:

带参数的装饰器.jpg

装饰器函数其实是这样一个接口约束,它必须接受一个callable对象作为参数,然后返回一个callable对象。在Python中一般callable对象都是函数,但也有例外。只要某个对象重写了 call() 方法,那么这个对象就是callable的。
类装饰器:

class Dog():
    def __init__(self, fun):
        self.__fun = fun

    def __call__(self, *args, **kwargs):
        print('...call...')
        return self.__fun(*args, **kwargs)

@Dog                       #foo = Dog(foo)
def foo():
    print('...foo...')

@Dog
def foo1(a, b):
    print('...foo1...')
    return a+b


foo()
print(foo1(1,2))

结果如下:

类装饰器结果.jpg

当用Dog来装作装饰器对foo函数进行装饰的时候,首先会创建Dog的实例对象
并且会把foo这个函数名当做参数传递到init方法中,即在init方法中的fun变量指向了foo函数体,foo函数相当于指向了用Dog创建出来的实例对象。当在使用foo()进行调用时,就相当于让这个对象(),因此会调用这个对象的call方法。为了能够在call方法中调用原来foo指向的函数体,所以在init方法中就需要一个实例属性来保存这个函数体的引用,所以才有了self.__fun = fun这句代码,从而在调用call方法中能够调用到foo之前的函数体。

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

推荐阅读更多精彩内容