From 8f1438964e1a336c44609cd4cc4c6aef61137011 Mon Sep 17 00:00:00 2001
From: Karan Jayachandra <karan.jayachandra@nxp.com>
Date: Thu, 17 Oct 2024 10:38:41 +0200
Subject: Added content to the README and reordered the code

---
 utilities.py | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)

(limited to 'utilities.py')

diff --git a/utilities.py b/utilities.py
index 1ad24b8..7ba140e 100644
--- a/utilities.py
+++ b/utilities.py
@@ -1,5 +1,9 @@
+from time import time
 from typing import Tuple
 from random import shuffle
+from dataclasses import dataclass
+
+SECS_IN_MIN = 60
 
 
 def _shuffle_two_lists(a: list, b: list) -> Tuple[list, list]:
@@ -21,3 +25,33 @@ def sample_list(indices: list, cost: list[int], count: int = None) -> list:
     indices, cost = _shuffle_two_lists(indices, cost)
     indices = _pick_from_list_after_sorting_other(indices, cost)[:count]
     return indices
+
+
+@dataclass
+class Timer:
+    ref: float = None
+    running: bool = False
+    max_mins: int = 20
+    default: str = ("00", "00")
+
+    def __post_init__(self):
+        self.max_seconds = self.max_mins * SECS_IN_MIN
+
+    def start(self):
+        self.ref = time() + self.max_seconds
+        self.running = True
+
+    def stop(self):
+        self.ref = None
+        self.running = False
+
+    def get(self):
+        if not self.running:
+            return self.default
+        total_seconds = self.ref - time()
+        if total_seconds < 0:
+            self.stop()
+            return self.default
+        mins = f"{int(total_seconds // SECS_IN_MIN):02d}"
+        secs = f"{int(total_seconds % SECS_IN_MIN):02d}"
+        return mins, secs
-- 
cgit v1.3.1