X-Git-Url: https://wannabe.guru.org/gitweb/?a=blobdiff_plain;ds=sidebyside;f=stopwatch.py;h=4e69b6992b278df8eb8e725954d9c610468318aa;hb=532df2c5b57c7517dfb3dddd8c1358fbadf8baf3;hp=fa4f2b52f54998bb029f7e14433e5451abd67812;hpb=e8fbbb7306430478dec55d2c963eed116d8330cc;p=python_utils.git

diff --git a/stopwatch.py b/stopwatch.py
index fa4f2b5..4e69b69 100644
--- a/stopwatch.py
+++ b/stopwatch.py
@@ -1,12 +1,15 @@
 #!/usr/bin/env python3
 
+# © Copyright 2021-2022, Scott Gasch
+
 """A simple stopwatch decorator / context for timing things."""
 
+import contextlib
 import time
-from typing import Callable, Optional
+from typing import Callable, Literal
 
 
-class Timer(object):
+class Timer(contextlib.AbstractContextManager):
     """
     A stopwatch to time how long something takes (walltime).
 
@@ -31,6 +34,6 @@ class Timer(object):
         self.end = 0.0
         return lambda: self.end - self.start
 
-    def __exit__(self, *args) -> Optional[bool]:
+    def __exit__(self, *args) -> Literal[False]:
         self.end = time.perf_counter()
-        return None  # don't suppress exceptions
+        return False