Discuss / Python / 参考了楼上的

参考了楼上的

Topic source

LesLieM樂

#1 Created at ... [Delete] [Delete and Lock User]
#-*- coding: utf-8 -*-
import functools

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


@log('heihei')
def now1():
    print('2017-01-11')

@log
def now2():
    print('2017-01-10')

  • 1

Reply