python-count-time

python-count-time.py

#coding:utf-8
import time


__author = 'haifang'

def count_function_time():
    def wraps(function):
        function.CUR_TIME = time.time()
        def log_time(point):
            end_time = time.time()
            spend_time = end_time - function.CUR_TIME
            function.CUR_TIME = end_time
            print "{} {} {:.3f}".format(function.__name__, point, spend_time)
        function.log_time = log_time
        return function
    return wraps

@count_function_time()
def main():
    main.log_time("point 1")
    time.sleep(1)
    main.log_time("point 2")
    time.sleep(1.2)
    main.log_time("point 3")


if __name__ == "__main__":

    main()