记录脚本执行时间进行统计

# encoding=utf-8
# 只能统计总的时间
import os
import sys
import time
import csv

news_dict_list = []

# 修改为实际的脚本放置位置
model_path_dict = {'a': "C:\\CodeDir\\PyCode",
                   'b': "C:\\CodeDir\\",
                   "c": "C:\\Users\\Desktop\\Code",
                   }
# 修改为实际的脚本执行脚本
model_command_dict = {'a': "dir", 'b': "dir", "c": "dir"}


def run_command(all_num_cycles, model_name):
    # 针对不同模块取不同的值
    model_path = model_path_dict.get(model_name)
    model_command = model_command_dict.get(model_name)
    for i in range(0, int(all_num_cycles)):

        start_time = time.time()
        struct_start_time = time.localtime(start_time)
        start_struct_time = time.strftime("%Y%m%d%H%M%S", struct_start_time)
        pwd_before_changed = os.getcwd()

        # change dir
        os.chdir(model_path)
        pwd_after_changed = os.getcwd()
        # 目录判断，报错退出
        if pwd_after_changed != model_path:
            print("Error dir ")
            sys.exit()
        os.system(model_command)
        time.sleep(1)
        # 退回原目录
        os.chdir(pwd_before_changed)
        time.sleep(1)
        end_time = time.time()
        struct_end_time = time.localtime(end_time)
        end_struct_time = time.strftime("%Y%m%d%H%M%S", struct_end_time)

        news = {"start_time": start_struct_time,
                "end_time": end_struct_time, "num_round": i + 1,
                "cost_time": end_time - start_time,
                "function": model_name
                }
        news_dict_list.append(news)
    csv_name = model_name + '.csv'
    # 输出csv
    headers = ['start_time', 'end_time', "num_round", "cost_time", "function"]
    with open(csv_name, mode='a', encoding='utf-8', newline='') as f:
        csv_writer = csv.DictWriter(f, headers)
        csv_writer.writeheader()
        csv_writer.writerows(news_dict_list)

        return model_name


if __name__ == '__main__':
    n = int(input("输入执行次数:\n"))
    m = input('a b c:\n')
    run_command(n, m)