python裝飾器計算程序運行時間

import time

def count_time(func):
    def int_time(*args, **kwargs):
        start_time = time.time()#這個是程序開始時間
        func()
        over_time = time.time()#這個是程序結束時間
        total_time =  over_time - start_time
        print('程序共計%s秒' % total_time)
    return int_time

@count_time
def main():
    print('>>>>開始計算函數運行時間')
    for i in range(1, 1000):    # 可以是任意函數  ， 這裏故意模擬函數的運行時間
        for j in range(i):
            print(j)

if __name__ == '__main__':
    main()