码迷,mamicode.com
首页 > 其他好文 > 详细

简单的装饰器

时间:2018-02-25 11:11:33      阅读:174      评论:0      收藏:0      [点我收藏+]

标签:str   tor   wrapper   import   需要   方法   class   print   instance   

import functools

def log(text):
    if isinstance(text, str):
        def decorator(func):
            @functools.wraps(func)
            def wrapper(*args, **kw):
                func(*args, **kw)
                print %s %s%(text, func.__name__)
            return wrapper
        return decorator
    else:
        @functools.wraps(text)
        def wrapper(*args, **kw):
            print %s %s%(call, text.__name__)
            return text(*args, **kw)
        return wrapper

@log(end)
def now1():
    print now

@log
def now2():
    print now

now1()
now2()

返回值:

now
end now1
call now2
now

 

几个重点:

一、带参数的装饰器需要写三层,而不带参数的需要两层。

二、这个装饰器既可以实现带参数,也可以实现不带参数。当传入了字符串时,函数会执行带参数的装饰器部分;没有任何参数时,会将text作为不带参数的装饰器中的‘func‘。

三、第一部分在原函数调用后打印,第二部分在原函数调用前打印。

四、需要用functool.wraps方法将原函数的函数名复制给装饰器,否则执行now1.__name__和now2.__name__会打印‘wrapper‘而不是‘now1‘和‘now2‘。

  

简单的装饰器

标签:str   tor   wrapper   import   需要   方法   class   print   instance   

原文地址:https://www.cnblogs.com/00986014w/p/8468639.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!