python装饰器

为函数动态地添加功能,却又不想改变原函数的定义:
那么就可以用一个函数B接受函数A为参数,然后B再返回一个函数。A是被修饰的函数,返回的函数是修饰过后的函数,我们通过调用修饰后的函数就可以实现动态添加功能。

def log(func):
    def wrapper(*args, **kw):
        print 'call %s():' % func.__name__
        return func(*args, **kw)
    return wrapper
def now():
    print '2013-12-25'
log(now)()

log(now)是修饰后的函数,加上()就是对wrapper的调用,输出结果为

call now():
2013-12-25

上面是一种基本的装饰器实现方式,但是仔细想想,我希望每次调用now都给它动态添加功能,难道我每次调用now都得像log(now)()这么写?其实python含有对装饰器的语法级支持:

def log(func):
    def wrapper(*args, **kw):
        print 'call %s():' % func.__name__      
    return func(*args, **kw)    
return wrapper

@log
def now():
    print '2013-12-25'

now()

输出和上面是一样的。
在now函数的定义前加上@log标记,now变量就是指向了修饰后的函数,即相当于执行了now=log(now),now变量不再指向原函数对象

上面的是不接受参数的装饰器,下面来看接受参数的装饰器:

def log(text):
    def decorator(func):
        def wrapper(*args, **kw):
            print '%s %s():' % (text, func.__name__)
            return func(*args, **kw)
        return wrapper
    return decorator

@log('execute')
def now():
    print '2013-12-25'

now()

输出为:

execute now():
2013-12-25

没有语法级支持的话,本来应该是这么写得:

def now():
    print '2013-12-25'
def log(text):
    def decorator(func):
        def wrapper(*args, **kw):
            print '%s %s():' % (text, func.__name__)
            return func(*args, **kw)
        return wrapper
    return decorator


log('execute')(now)()

上面的两种装饰器的定义都没错,但是还差一步。上面那样经过修饰的函数,它们的name属性已经从’now’变成了’wrapper’.可能会影响一些依赖函数签名的代码执行。
无参的decorator:

import functools
def log(func):
    @functools.wraps(func)
    def wrapper(*args, **kw):
        print 'call %s():' % func.__name__
        return func(*args, **kw)
return wrapper

或者有参的decorator:


import functoolsdef log(text):
    def decorator(func):
        @functools.wraps(func)
        def wrapper(*args, **kw):
            print '%s %s():' % (text, func.__name__)
            return func(*args, **kw)
    return wrapper
return decorator

本文是初学廖雪峰python教程,所写笔记,代码示例是廖大的。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值