python时间记录器

最新推荐文章于 2024-07-30 17:23:09 发布

howtoshare

最新推荐文章于 2024-07-30 17:23:09 发布

阅读量326

点赞数 6

文章标签： python 开发语言

本文链接：https://blog.csdn.net/qq_34045261/article/details/136284595

版权

文章介绍了一个Python类Timer，用于测量代码执行时间并记录详细的子任务时间，支持自定义日志输出。Timer类还提供了子类TimerSubcategory，用于组织和跟踪不同层次的子操作。用户可以通过with语句轻松使用这些功能进行性能监控。

摘要由CSDN通过智能技术生成

借鉴1stable diffusion webui

import time
import argparse
class TimerSubcategory:
    def __init__(self, timer, category):
        self.timer = timer
        self.category = category
        self.start = None
        self.original_base_category = timer.base_category

    def __enter__(self):
        self.start = time.time()
        self.timer.base_category = self.original_base_category + self.category + "/"
        self.timer.subcategory_level += 1

        if self.timer.print_log:
            print(f"{'  ' * self.timer.subcategory_level}{self.category}:")

    def __exit__(self, exc_type, exc_val, exc_tb):
        elapsed_for_subcategroy = time.time() - self.start
        self.timer.base_category = self.original_base_category
        self.timer.add_time_to_record(self.original_base_category + self.category, elapsed_for_subcategroy)
        self.timer.subcategory_level -= 1
        self.timer.record(self.category, disable_log=True)


class Timer:
    def __init__(self, print_log=False):
        self.start = time.time()
        self.records = {}
        self.total = 0
        self.base_category = ''
        self.print_log = print_log
        self.subcategory_level = 0

    def elapsed(self):
        end = time.time()
        res = end - self.start
        self.start = end
        return res

    def add_time_to_record(self, category, amount):
        if category not in self.records:
            self.records[category] = 0

        self.records[category] += amount

    def record(self, category, extra_time=0, disable_log=False):
        e = self.elapsed()

        self.add_time_to_record(self.base_category + category, e + extra_time)

        self.total += e + extra_time

        if self.print_log and not disable_log:
            print(f"{'  ' * self.subcategory_level}{category}: done in {e + extra_time:.3f}s")

    def subcategory(self, name):
        
        self.elapsed()

        subcat = TimerSubcategory(self, name)
        return subcat

    def summary(self):
        res = f"{self.total:.1f}s"

        additions = [(category, time_taken) for category, time_taken in self.records.items() if time_taken >= 0.1 and '/' not in category]
        if not additions:
            return res

        res += " ("
        res += ", ".join([f"{category}: {time_taken:.1f}s" for category, time_taken in additions])
        res += ")"

        return res

    def dump(self):
        return {'total': self.total, 'records': self.records}

    def reset(self):
        self.__init__()


parser = argparse.ArgumentParser(add_help=False)
parser.add_argument("--log-startup", action='store_true', help="print a detailed log of what's happening at startup")
args = parser.parse_known_args()[0]

startup_timer = Timer(print_log=args.log_startup)

startup_record = None

稍微讲解一下
Timer类实例的时候开始计时，然后通过elapsed 方法计算到这个方法的时间，record 通过他把时间，和对应的内容记录一下。后期自己修改记录到日志中，summary 输出结构化的日志记录
subcategory子类，通过with 来使用子类记录

使用

timer = Timer()
timer.record(“operation_name”)
你也可以使用 subcategory 方法来创建子任务，这个方法返回一个上下文管理器，可以用在 with 语句中：
with timer.subcategory(“sub_operation_name”):
# Your sub operation here

print(timer.summary())