执行脚本装饰器记录每次执行时间

Python脚本执行与性能记录

最新推荐文章于 2025-10-20 15:44:50 发布

原创最新推荐文章于 2025-10-20 15:44:50 发布 · 158 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #开发语言

本文介绍了一个Python脚本，该脚本能根据指定次数执行不同的命令，并记录每次执行的开始时间、结束时间及耗时。此外，脚本还会将这些性能数据输出到CSV文件中以供后续分析。

# encoding=utf-8

import os
import sys
import time
import csv

news_dict_list = []


def loging(func):
    def wrapper(times, *args, **kwargs):
        for i in range(times):
            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)

            res = func(*args, **kwargs)

            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,
                    }
            news_dict_list.append(news)

        # 输出csv
        headers = ['start_time', 'end_time', "num_round", "cost_time"]
        with open('pwr_log.csv', mode='w', encoding='utf-8', newline='') as f:
            csv_writer = csv.DictWriter(f, headers)
            csv_writer.writeheader()
            csv_writer.writerows(news_dict_list)

        return res

    return wrapper


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


@loging
def run_command(mode):
    # 针对不同模块取不同的值
    model_path = model_path_dict.get(mode)
    model_command = model_command_dict.get(mode)
    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)
    return mode


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