装饰器记录函数执行时间-优快云博客

本文介绍了一种使用Python装饰器来记录函数开始和结束时间的方法，并展示了如何保持原始函数名称不变。通过不同装饰器的实现，可以灵活地在函数调用前后打印时间戳。

import time
import functools

def exec_time(func):
    def wrapper(*args, **kw):
        print(time.strftime('%Y-%m-%d %H:%M:%S',time.localtime(time.time())))
        return func(*args, **kw)
    return wrapper

@exec_time
def testfunc():
    print("test function...")

testfunc()
print(testfunc.__name__)

print("=====================")

def exec_time(text):
    def deractor(func):
        @functools.wraps(func)
        def wrapper(*args, **kw):
            print(time.strftime(text, time.localtime(time.time())))
            return func(*args, **kw)
        return wrapper
    return deractor

@exec_time('%Y-%m-%d')
def testfunc():
    print("test function...")

testfunc()
print(testfunc.__name__)

print("=====================")

def exec_time(text):
    def deractor(func):
        @functools.wraps(func)
        def wrapper(*args, **kw):
            print("begin time:", time.strftime(text, time.localtime(time.time())))
            t = func(*args, **kw)
            print("end time:", time.strftime(text, time.localtime(time.time())))
            return t;
        return wrapper
    return deractor

@exec_time('%Y-%m-%d %H:%M:%S')
def testfunc():
    print("test function...")

testfunc()
print(testfunc.__name__)

output:

2017-12-03 20:50:00
test function...
wrapper
=====================
2017-12-03
test function...
testfunc
=====================
begin time: 2017-12-03 20:50:00
test function...
end time: 2017-12-03 20:50:00
testfunc

转载于:https://www.cnblogs.com/hfultrastrong/p/7967214.html