relib 1.3.11__tar.gz → 1.3.12__tar.gz

{relib-1.3.11 → relib-1.3.12}/PKG-INFO

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

{relib-1.3.11 → relib-1.3.12}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "relib"
-version = "1.3.11"
+version = "1.3.12"
 requires-python = ">=3.10"
 dependencies = []
 authors = [

relib-1.3.12/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-1.3.11 → relib-1.3.12}/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
@@ -102,10 +105,10 @@ class seekable(Generic[T]):
     self.index = 0
 
   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 +133,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 +199,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:
+    iters = tee(iterable, n)
+    return tuple(map(lambda i, iter: (x[i] for x in iter), range(n), iters))

{relib-1.3.11 → relib-1.3.12}/uv.lock

@@ -161,7 +161,7 @@ wheels = [
 
 [[package]]
 name = "relib"
-version = "1.3.11"
+version = "1.3.12"
 source = { editable = "." }
 
 [package.dev-dependencies]