Cleanup Timer

0x9900 · Jun 1, 2024 · 51d6093 · 51d6093
1 parent a71573f
commit 51d6093
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/dxcluster/_dxcluster.py b/dxcluster/_dxcluster.py
@@ -524,8 +524,7 @@ def run(self) -> None:
 
 class Timer:
   def __init__(self, max_time):
-    self.cur_time = int(time.time())
-    self.max_time = self.cur_time + max_time
+    self.max_time = self.current_time + max_time
 
   def __iter__(self):
     return self
@@ -534,11 +533,14 @@ def reduce(self):
     self.max_time /= 2
 
   def __next__(self):
-    self.cur_time = int(time.time())
-    if self.cur_time > self.max_time:
+    if self.current_time > self.max_time:
       raise StopIteration
     return True
 
+  @property
+  def current_time(self):
+    return time.time()
+
 
 class QueueIterator:
   def __init__(self, queue):