relib 1.3.11__py3-none-any.whl → 1.3.13__py3-none-any.whl

relib/class_utils.py

@@ -0,0 +1,9 @@
+from typing import Callable, Generic, TypeVar
+
+Fn = TypeVar("Fn", bound=Callable)
+
+class slicer(Generic[Fn]):
+  def __init__(self, fn: Fn):
+    self.fn = fn
+
+  __getitem__: Fn = lambda self, key: self.fn(key)  # type: ignore

relib/iter_utils.py

@@ -1,7 +1,8 @@
 from __future__ import annotations
 from contextlib import contextmanager
-from itertools import chain, islice
+from itertools import chain, islice, tee
 from typing import Any, Generic, Iterable, Literal, Sequence, overload
+from .class_utils import slicer
 from .dict_utils import dict_firsts
 from .types import T1, T2, T3, T4, T5, T, U
 
@@ -17,6 +18,7 @@ __all__ = [
   "range_of", "reversed_enumerate",
   "seekable", "sort_by",
   "transpose",
+  "unzip_iterable",
 ]
 
 def as_list(iterable: Iterable[T]) -> list[T]:
@@ -81,6 +83,7 @@ class seekable(Generic[T]):
     self.index = 0
     self.source = iter(iterable)
     self.sink: list[T] = []
+    self.abs = slicer(self.abs_getitem)
 
   def __iter__(self):
     return self
@@ -97,15 +100,16 @@ class seekable(Generic[T]):
   def __bool__(self):
     return bool(self[:1])
 
-  def clear(self):
+  def clear(self) -> seekable[T]:
     self.sink[:self.index] = []
     self.index = 0
+    return self
 
   def seek(self, index: int) -> seekable[T]:
-    remainder = index - self.index
-    if remainder > 0:
+    index = max(0, index)
+    self.index = min(index, len(self.sink))
+    if (remainder := index - self.index) > 0:
       next(islice(self, remainder, remainder), None)
-    self.index = max(0, min(index, len(self.sink)))
     return self
 
   def step(self, count: int) -> seekable[T]:
@@ -130,10 +134,21 @@ class seekable(Generic[T]):
     with self.freeze():
       if isinstance(key, int):
         return self[key:key + 1][0]
-      parts = (key.start, key.stop, key.step)
-      if not all(x is None or x >= 0 for x in parts):
-        raise ValueError(f"Indices for seekable() must be positive int")
-      return list(islice(self, *parts))
+      start, stop, step = key.start, key.stop, key.step
+      delta = min(self.index, -min(0, start or 0))
+      start = None if start is None else max(0, start + delta)
+      stop = None if stop is None else max(0, stop + delta)
+      self.step(-delta)
+      return list(islice(self, start, stop, step))
+
+  @overload
+  def abs_getitem(self, key: int) -> T: ...
+  @overload
+  def abs_getitem(self, key: slice[int | None]) -> list[T]: ...
+  def abs_getitem(self, key: int | slice[int | None]):
+    with self.freeze():
+      self.seek(0)
+      return self[key]
 
   def consume(self) -> Iterable[T]:
     for value in self:
@@ -185,3 +200,15 @@ def transpose(tuples: Iterable[tuple], default_num_returns=0) -> tuple[list, ...
   if not output:
     return ([],) * default_num_returns
   return tuple(map(list, output))
+
+@overload
+def unzip_iterable(iterable: Iterable[tuple[T1, T2]], n: Literal[2]) -> tuple[Iterable[T1], Iterable[T2]]: ...
+@overload
+def unzip_iterable(iterable: Iterable[tuple[T1, T2, T3]], n: Literal[3]) -> tuple[Iterable[T1], Iterable[T2], Iterable[T3]]: ...
+@overload
+def unzip_iterable(iterable: Iterable[tuple[T1, T2, T3, T4]], n: Literal[4]) -> tuple[Iterable[T1], Iterable[T2], Iterable[T3], Iterable[T4]]: ...
+@overload
+def unzip_iterable(iterable: Iterable[tuple[T1, T2, T3, T4, T5]], n: Literal[5]) -> tuple[Iterable[T1], Iterable[T2], Iterable[T3], Iterable[T4], Iterable[T5]]: ...
+def unzip_iterable(iterable: Iterable[tuple], n: int) -> tuple[Iterable, ...]:
+    iters = tee(iterable, n)
+    return tuple(map(lambda i, iter: (x[i] for x in iter), range(n), iters))

relib/runtime_tools.py

@@ -5,7 +5,7 @@ import sys
 from concurrent.futures import ThreadPoolExecutor
 from functools import partial, wraps
 from time import time
-from typing import Callable, Coroutine, Iterable, ParamSpec, TypeVar
+from typing import Awaitable, Callable, Coroutine, Iterable, ParamSpec, TypeVar
 from .iter_utils import as_list
 from .processing_utils import noop
 from .types import T
@@ -35,13 +35,13 @@ def clear_console() -> None:
 def console_link(text: str, url: str) -> str:
   return f"\033]8;;{url}\033\\{text}\033]8;;\033\\"
 
-async def worker(task: Coro[T], semaphore: asyncio.Semaphore, update=noop) -> T:
+async def worker(task: Coro[T] | Awaitable[T], semaphore: asyncio.Semaphore, update=noop) -> T:
   async with semaphore:
     result = await task
     update()
     return result
 
-async def roll_tasks(tasks: Iterable[Coro[T]], workers=default_workers, progress=False) -> list[T]:
+async def roll_tasks(tasks: Iterable[Coro[T] | Awaitable[T]], workers=default_workers, progress=False) -> list[T]:
   semaphore = asyncio.Semaphore(workers)
   if not progress:
     return await asyncio.gather(*[worker(task, semaphore) for task in tasks])

{relib-1.3.11.dist-info → relib-1.3.13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: relib
-Version: 1.3.11
+Version: 1.3.13
 Project-URL: Repository, https://github.com/Reddan/relib.git
 Author: Hampus Hallman
 License: Copyright 2018-2025 Hampus Hallman

{relib-1.3.11.dist-info → relib-1.3.13.dist-info}/RECORD

@@ -1,12 +1,13 @@
 relib/__init__.py,sha256=WerjUaM_sNvudjXFudLRtXB7viZWEW1RSinkDjrh4nE,163
+relib/class_utils.py,sha256=3z23lxhjFbn68IM_YnJPlvGG2pkybZuK3o_C-9b_LCc,225
 relib/dict_utils.py,sha256=ez8egTPvO05wVNWIyO-aJjtqaXUDYbzyw32M42q1D44,3677
 relib/io_utils.py,sha256=cUyUFhrMCUDfkINhYo32QPaVGz3chqDO3ElymSCoWEg,1086
-relib/iter_utils.py,sha256=iH8fu9FrNRurOsa--GqDbTzHdrV8eJOONvumMGr2pMA,6303
+relib/iter_utils.py,sha256=O5NAHUWLPzCDJiAUe1I5mghIV5T4Q2OcSKvs6OClC_w,7562
 relib/processing_utils.py,sha256=kFb4izrR8vec9BFWeGSb-TXR54bbA4X3dxAfIBZItuI,1911
-relib/runtime_tools.py,sha256=6HpRXmAHHLq4bNRj6fSiaLQeHR2h3OdxFZxaB93fXhI,2979
+relib/runtime_tools.py,sha256=jK4L4uE3eyGtmLNHEOSE651nrFdql00w7ctudGN6mb8,3020
 relib/type_utils.py,sha256=68yEhEIBCIKfsIT1u647avHVA_JjVvDYSdtLD3nCrJs,338
 relib/types.py,sha256=JzSGDl2L6tUWizGSsUpuyCKf-VKgtgfK_n21azSSMLM,288
-relib-1.3.11.dist-info/METADATA,sha256=YvXC6KVDCiYFMUUcbP8onzIcGzUzgyerJ4EBVLkDeBM,1296
-relib-1.3.11.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-relib-1.3.11.dist-info/licenses/LICENSE,sha256=9xVsdtv_-uSyY9Xl9yujwAPm4-mjcCLeVy-ljwXEWbo,1059
-relib-1.3.11.dist-info/RECORD,,
+relib-1.3.13.dist-info/METADATA,sha256=KYD1MwW_CYoc38hMwsjoi5-DZgE7MiH20Qn1q-wvdVs,1296
+relib-1.3.13.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+relib-1.3.13.dist-info/licenses/LICENSE,sha256=9xVsdtv_-uSyY9Xl9yujwAPm4-mjcCLeVy-ljwXEWbo,1059
+relib-1.3.13.dist-info/RECORD,,