From 4f531fe4cb5fbfe6e772f1314c6b4625245e4afb Mon Sep 17 00:00:00 2001
From: Stefan Holst <holst@ci.kyutech.ac.jp>
Date: Wed, 6 Jan 2021 12:54:14 +0900
Subject: [PATCH] implement logging range

---
 src/kyupy/__init__.py | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/src/kyupy/__init__.py b/src/kyupy/__init__.py
index 389858e..ea054a6 100644
--- a/src/kyupy/__init__.py
+++ b/src/kyupy/__init__.py
@@ -29,6 +29,19 @@ def hr_bytes(nbytes):
     return f'{nbytes:.1f}{["", "ki", "Mi", "Gi", "Ti", "Pi"][multiplier]}B'
 
 
+def hr_time(seconds):
+    s = ''
+    if seconds >= 3600:
+        h = seconds // 3600
+        seconds -= h * 3600
+        s += f'{h}h'
+    if seconds >= 60 or len(s) > 0:
+        m = seconds // 60
+        seconds -= m * 60
+        s += f'{m}m'
+    return f'{s}{int(seconds)}s'
+
+
 class Log:
     def __init__(self):
         self.start = time.perf_counter()
@@ -48,6 +61,22 @@ class Log:
 
     def error(self, message): self.log('E', message)
 
+    def range(self, *args):
+        elems = len(range(*args))
+        start_time = time.perf_counter()
+        lastlog_time = start_time
+        log_interval = 5
+        for elem, i in enumerate(range(*args)):
+            yield i
+            current_time = time.perf_counter()
+            if current_time > lastlog_time + log_interval:
+                work_done = elem / elems
+                elapsed_time = current_time - start_time
+                total_time = elapsed_time / work_done
+                remaining_time = total_time - elapsed_time
+                self.log(':', f'{work_done*100:.0f}% done {hr_time(elapsed_time)} elapsed {hr_time(remaining_time)} remaining')
+                log_interval = min(600, int(log_interval*1.5))
+                lastlog_time = current_time
 
 log = Log()