thds.core 1.45.20250805195434__py3-none-any.whl → 1.45.20250805205302__py3-none-any.whl

thds/core/iterators.py

@@ -1,4 +1,6 @@
+import time
 import typing as ty
+from datetime import timedelta
 
 T = ty.TypeVar("T")
 
@@ -7,3 +9,20 @@ def null_safe_iter(it: ty.Optional[ty.Iterable[T]]) -> ty.Iterator[T]:
     """Iterate the iterable if it is not None"""
     if it is not None:
         yield from it
+
+
+_Frequency = ty.Union[timedelta, float]
+
+
+def _to_float(t: _Frequency) -> float:
+    return t.total_seconds() if isinstance(t, timedelta) else t
+
+
+def titrate(it: ty.Iterable[T], *, at_rate: _Frequency, until_nth: int) -> ty.Iterator[T]:
+    it_, freq = iter(it), _to_float(at_rate)
+
+    for i, x in enumerate(it_):
+        yield x
+
+        if i + 1 < until_nth:
+            time.sleep(freq)

thds/core/progress.py

@@ -1,5 +1,7 @@
+import concurrent.futures
 import math
 import typing as ty
+from contextlib import contextmanager
 from datetime import timedelta
 from functools import partial, wraps
 from timeit import default_timer
@@ -54,12 +56,19 @@ def _name(obj: ty.Any) -> str:
     return "item"
 
 
+_Frequency = ty.Union[timedelta, float]
+
+
+def _to_float(t: _Frequency) -> float:
+    return t if isinstance(t, float) else t.total_seconds()
+
+
 @_progress_scope.bound
 def report(
     it: ty.Iterable[T],
     *,
     name: str = "",
-    roughly_every_s: timedelta = timedelta(seconds=20),
+    roughly_every_s: _Frequency = timedelta(seconds=20),
 ) -> ty.Iterator[T]:
     """Report round-number progress roughly every so often..."""
     iterator = iter(it)
@@ -67,7 +76,7 @@ def report(
     start = default_timer()
     report_every = 0
     last_report = 0
-    frequency = roughly_every_s.total_seconds()
+    frequency = _to_float(roughly_every_s)
 
     try:
         first_item = next(iterator)
@@ -110,3 +119,28 @@ def _report_gen(f: ty.Callable[P, ty.Iterator[T]], *args: P.args, **kwargs: P.kw
 
 def report_gen(f: ty.Callable[P, ty.Iterator[T]]) -> ty.Callable[P, ty.Iterator[T]]:
     return wraps(f)(partial(_report_gen, f))
+
+
+@contextmanager
+def report_still_alive(
+    roughly_every_s: _Frequency = timedelta(seconds=60), *, name: str = ""
+) -> ty.Iterator[None]:
+    start, frequency = default_timer(), _to_float(roughly_every_s)
+
+    sentinel: concurrent.futures.Future[bool] = concurrent.futures.Future()
+
+    def _report_still_alive():
+        while not sentinel.done():
+            try:
+                return sentinel.result(timeout=frequency)
+            except concurrent.futures.TimeoutError:
+                logger.info(
+                    f"Still working{f' on {name} ' if name else ' '}after {default_timer() - start:.3f} seconds..."
+                )
+
+    with concurrent.futures.ThreadPoolExecutor() as executor:
+        executor.submit(_report_still_alive)
+
+        yield
+
+        sentinel.set_result(True)

{thds_core-1.45.20250805195434.dist-info → thds_core-1.45.20250805205302.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: thds.core
-Version: 1.45.20250805195434
+Version: 1.45.20250805205302
 Summary: Core utilities.
 Author-email: Trilliant Health <info@trillianthealth.com>
 License: MIT

{thds_core-1.45.20250805195434.dist-info → thds_core-1.45.20250805205302.dist-info}/RECORD

@@ -23,7 +23,7 @@ thds/core/home.py,sha256=tTClL_AarIKeri1aNCpuIC6evD7qr83ESGD173B81hU,470
 thds/core/hostname.py,sha256=canFGr-JaaG7nUfsQlyL0JT-2tnZoT1BvXzyaOMK1vA,208
 thds/core/imports.py,sha256=0LVegY8I8_XKZPcqiIp2OVVzEDtyqYA3JETf9OAKNKs,568
 thds/core/inspect.py,sha256=3IY9CSa7zAcAVyBDOYfMtJ2QU5cRc98JaN91XAbaSok,2368
-thds/core/iterators.py,sha256=d3iTQDR0gCW1nMRmknQeodR_4THzR9Ajmp8F8KCCFgg,208
+thds/core/iterators.py,sha256=h0JBu2-rYhKMfJTDlZWfyHQWzgtIO8vp_Sp0gENFo7g,645
 thds/core/lazy.py,sha256=e1WvG4LsbEydV0igEr_Vl1cq05zlQNIE8MFYT90yglE,3289
 thds/core/link.py,sha256=4-9d22l_oSkKoSzlYEO-rwxO1hvvj6VETY7LwvGcX6M,5534
 thds/core/logical_root.py,sha256=gWkIYRv9kNQfzbpxJaYiwNXVz1neZ2NvnvProtOn9d8,1399
@@ -32,7 +32,7 @@ thds/core/meta.py,sha256=3oX7wTO_SmrVKABFPLHHIVyNBXEil1MdGfc5s88_Isk,12134
 thds/core/parallel.py,sha256=0A3JQx7aPNaFLh61gZog2E3k4C5qdr5st-xOPS-3j1Y,7612
 thds/core/pickle_visit.py,sha256=QNMWIi5buvk2zsvx1-D-FKL7tkrFUFDs387vxgGebgU,833
 thds/core/prof.py,sha256=5ViolfPsAPwUTHuhAe-bon7IArPGXydpGoB5uZmObDk,8264
-thds/core/progress.py,sha256=4YGbxliDl1i-k-88w4s86uy1E69eQ6xJySGPSkpH1QM,3358
+thds/core/progress.py,sha256=tY8tc_6CMnu_O8DVisnsRoDpFJOw5vqyYzLhQDxsLn8,4361
 thds/core/project_root.py,sha256=K18U3MLthZnzmdrWmKKtHLd6iu7am9b2vNAThqknpfo,891
 thds/core/protocols.py,sha256=4na2EeWUDWfLn5-SxfMmKegDSndJ5z-vwMhDavhCpEM,409
 thds/core/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -73,8 +73,8 @@ thds/core/sqlite/structured.py,sha256=SvZ67KcVcVdmpR52JSd52vMTW2ALUXmlHEeD-VrzWV
 thds/core/sqlite/types.py,sha256=oUkfoKRYNGDPZRk29s09rc9ha3SCk2SKr_K6WKebBFs,1308
 thds/core/sqlite/upsert.py,sha256=BmKK6fsGVedt43iY-Lp7dnAu8aJ1e9CYlPVEQR2pMj4,5827
 thds/core/sqlite/write.py,sha256=z0219vDkQDCnsV0WLvsj94keItr7H4j7Y_evbcoBrWU,3458
-thds_core-1.45.20250805195434.dist-info/METADATA,sha256=91Sly1y3tAKBag9aPdocjvr4bpvy317V9-O0RAB-eN0,2216
-thds_core-1.45.20250805195434.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-thds_core-1.45.20250805195434.dist-info/entry_points.txt,sha256=bOCOVhKZv7azF3FvaWX6uxE6yrjK6FcjqhtxXvLiFY8,161
-thds_core-1.45.20250805195434.dist-info/top_level.txt,sha256=LTZaE5SkWJwv9bwOlMbIhiS-JWQEEIcjVYnJrt-CriY,5
-thds_core-1.45.20250805195434.dist-info/RECORD,,
+thds_core-1.45.20250805205302.dist-info/METADATA,sha256=Sqmu1rzU2Q7vL6cGeaU9C9DzbH302MIkPGoFU-B3Ttg,2216
+thds_core-1.45.20250805205302.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+thds_core-1.45.20250805205302.dist-info/entry_points.txt,sha256=bOCOVhKZv7azF3FvaWX6uxE6yrjK6FcjqhtxXvLiFY8,161
+thds_core-1.45.20250805205302.dist-info/top_level.txt,sha256=LTZaE5SkWJwv9bwOlMbIhiS-JWQEEIcjVYnJrt-CriY,5
+thds_core-1.45.20250805205302.dist-info/RECORD,,