手把手教你实现一个 Python 计时器

为了更好地掌握 Python 计时器的应用，我们后面还补充了有关Python类、上下文管理器和装饰器的背景知识。因篇幅限制，其中利用上下文管理器和装饰器优化 Python 计时器，将在后续文章学习，不在本篇文章范围内。,,首先，​我们向某段代码中添加一个Python 计时器以监控其性能。,Python 中的内置time[1]模块中有几个可以测量时间的函数：,Python 3.7 引入了几个新函数，如thread_time()[2]，以及上述所有函数的纳秒版本，以_ns​后缀命名。例如，perf_counter_ns()​是perf_counter()的纳秒版本的。,首先，使用perf_counter()​创建一个 Python 计时器。将把它与其他 Python 计时器函数进行比较，看看 perf_counter() 的优势。,创建一个脚本，​定义一个简短的函数：从清华云​上下载一组数据。,我们可以使用 Python 计时器来监控该​脚本的性能。,现在使用函数time.perf_counter()​函数创建一个计时器，这是一个非常适合针对部分代码的性能计时的计数器。,perf_counter()​从某个未指定的时刻开始测量时间（以秒为单位），这意味着对该函数的单个调用的返回值没有用。但当查看对perf_counter()两次调用之间的差异时，可以计算出两次调用之间经过了多少秒。,在此示例中，两次调用 perf_counter() 相隔近 19 秒。可以通过计算两个输出之间的差异来确认这一点：413.31714087 - 394.540232282 = 18.78。,现在可以将 Python 计时器添加到示例代码中：,注意perf_counter()​通过计算两次调用之间的差异来打印整个程序运行所花费的时间。,print()​函数中 f​ 字符串前面的表示这是一个 f-string​ ，这是格式化文本字符串的较为便捷的​方式。:0.4f​是一个格式说明符，表示数字，toc - tic应打印为带有四位小数的十进制数。,运行程序可以看到程序经过的时间：,就是这么简单。接下来我们一起学习如何将 Python 计时器包装到一个类、一个上下文管理器和一个装饰器中（该系列后续两篇文章，待更新）​，这样可以更加一致和方便使用计时器。,这里我们至少需要一个变量​来存储 Python 计时器的状态。接下来​我们创建一个与手动调用 perf_counter() 相同的类，但更具可读性和一致性​。,创建和更新Timer类，​使用该类以多种不同方式对代码进行计时。,Class​类是面向对象编程的主要构建块。类本质上是一个模板，可以使用它来创建对象。,在 Python 中，当需要对需要跟踪特定状态的事物进行建模时，类非常有用。一般来说，类是属性的集合，称为属性，以及行为，称为方法。,类有利于跟踪状态。在Timer​类中，想要跟踪计时器何时开始以及已经多少时间。对于Timer​类的第一个实现，将添加一个._start_time​属性以及.start()​和.stop()​方法。将以下代码添加到名为 timer.py 的文件中：,这里我们需要花点时间仔细地浏览代码，会发现一些不同的事情。,首先定义了一个TimerError​ Python 类。该(Exception)​符号表示TimerError​ 继承自另一个名为Exception​的父类。使用这个内置类进行错误处理。不需要向TimerError​添加任何属性或方法，但自定义错误可以更灵活地处理Timer内部问题。,接下来自定义Timer​类。当从一个类创建或实例化一个对象时，代码会调用特殊方法.__init__()​初始化实例。在这里定义的第一个Timer​版本中，只需初始化._start_time​属性，将用它来跟踪 Python 计时器的状态，计时器未运行时它的值为None。计时器运行后，用它来跟踪计时器的启动时间。,当调用.start()​启动新的 Python 计时器时，首先检查计时器是否运行。然后将perf_counter()​的当前值存储在._start_time中。,另一方面，当调用.stop()​时，首先检查Python计时器是否正在运行。如果是，则将运行时间计算为perf_counter()​的当前值与存储在._start_time​中的值的差值。最后，重置._start_time，以便重新启动计时器，并打印运行时间。,以下是使用Timer方法：,将此示例与前面直接使用perf_counter()的示例进行比较。代码的结构相似，但现在代码更清晰了，这也是使用类的好处之一。通过仔细选择类、方法和属性名称，可以使你的代码非常具有描述性!,现在Timer​类中写入download_data.py。只需要对以前的代码进行一些更改：,注意，该代码与之前使用的代码非常相似。除了使代码更具可读性之外，Timer还负责将经过的时间打印到控制台，使得所用时间的记录更加一致。运行代码时，得到的输出几乎相同：,打印经过的时间Timer可能是一致的，但这种方法好像不是很灵活。下面我们添加一些更加灵活的东西到代码中。​,到目前为止，我们已经了解到类适用于我们想要封装状态并确保代码一致性的情况。在本节中，我们将一起给 Python 计时器加入更多便利性和灵活性，那​怎么做呢？,首先，自定义用于报告所用时间的文本。在前面的代码中，文本 f"Elapsed time: {elapsed_time:0.4f} seconds"​ 被生​硬编码到 .stop() ​中。如若想使得类代码更加灵活， 可以使用实例变量，其值通常作为参数传递给.__init__()​并存储到 self 属性。为方便起见，我们还可以提供合理的默认值。,要添加.text​为Timer​实例变量，可执行以下操作timer.py：,注意，默认文本"Elapsed time: {:0.4f} seconds"​是作为一个常规字符串给出的，而不是f-string​。这里不能使用f-string​，因为f-string会立即计算，当你实例化Timer时，你的代码还没有计算出消耗的时间。,在.stop()​中，.text​用作模板并使用.format()方法填充模板：,在此更新为timer.py之后，可以将文本更改如下：,接下来，我们不只是想将消息打印到控制台，还想保存时间测量结果，这样可以便于将它们存储在数据库中。可以通过从.stop()​返回elapsed_time的值来实现这一点。然后，调用代码可以选择忽略该返回值或保存它以供以后处理。,如果想要将Timer集成到日志logging中。要支持计时器的日志记录或其他输出，需要更改对print()的调用，以便用户可以提供自己的日志记录函数。这可以用类似于你之前定制的文本来完成:,不是直接使用print()​，而是创建另一个实例变量 self.logger​，引用一个接受字符串作为参数的函数。除此之外，还可以对文件对象使用logging.info()​或.write()​等函数。还要注意if中，它允许通过传递logger=None来完全关闭打印。,以下是两个示例，展示了新功能的实际应用：,接下来第三个改进是积累时间度量的能力。例如，在循环中调用一个慢速函数时，希望以命名计时器的形式添加更多的功能，并使用一个字典来跟踪代码中的每个Python计时器。,我们扩展download_data.py脚本。,这段代码的一个微妙问题是，不仅要测量下载数据所需的时间，还要测量 Python 存储数据到磁盘所花费的时间。这可能并重要，有时候这两者所花费的时间可以忽略不计。但还是希望有一种方法可以精确地计时没一个步骤，将会更好。,有几种方法可以在不改变Timer当前实现的情况下解决这个问题，且只需要几行代码即可实现。,首先，将引入一个名为.timers的字典作为Timer的类变量，此时Timer的所有实例将共享它。通过在任何方法之外定义它来实现它:,类变量可以直接在类上访问，也可以通过类的实例访问：,在这两种情况下，代码都返回相同的空类字典。,接下来向 Python 计时器添加可选名称。可以将该名称用于两种不同的目的：,要向Python计时器添加名称，需要对 timer.py​ 进行更改。首先，Timer 接受 name 参数。第二，当计时器停止时，运行时间应该添加到 .timers 中:,注意，在向.timers​中添加新的Python计时器时，使用了.setdefault()​方法。它只在没有在字典中定义name的情况下设置值，如果name已经在.timers中使用，那么该值将保持不变，此时可以积累几个计时器:,现在可以重新访问download_data.py并确保仅测量下载数据所花费的时间：,现在你有了一个非常简洁的版本，Timer它一致、灵活、方便且信息丰富！也可以将本节中所做的许多改进应用于项目中的其他类型的类。,最后一个改进Timer，以交互方式使用它时使其更具信息性。下面操作是实例化一个计时器类，并查看其信息：,最后一行是 Python 表示对象的默认方式。我们从这个结果中看到的信息，并不是很明确，我们接下来对其进行改进。,这里介绍一个 dataclasses 类，该类仅包含在 Python 3.7 及更高版本中。,可以使用@dataclass装饰器将 Python 计时器转换为数据类,此代码替换了之前的 .__init__() ​方法。请注意数据类如何使用类似于之前看到的用于定义所有变量的类变量语法的语法。事实上，.__init__()是根据类定义中的注释变量自动为数据类创建的。,如果需要注释变量以使用数据类。可以使用此注解向代码添加类型提示。如果不想使用类型提示，那么可以使用 Any 来注释所有变量。接下来我们很快就会学习如何将实际类型提示添加到我们的数据类中。,以下是有关 Timer 数据类的一些注意事项：,新 Timer 数据类与之前的常规类使用功能一样，但它现在有一个很好的信息表示：,现在我们有了一个非常简洁的 Timer 版本，它一致、灵活、方便且信息丰富！我们还可以将本文中所做的许多改进应用于项目中的其他类型的类。,现在我们访问当前的完整源代码Timer。会注意到在代码中添加了类型提示以获取额外的文档：,总结下： 使用类创建 Python 计时器有几个好处：,这个类非常灵活，几乎可以在任何需要监控代码运行时间的情况下使用它。但是，在接下来的部分中，云朵君将和大家一起了解如何使用上下文管理器和装饰器，这将更方便地对代码块和函数进行计时。