relib 1.3.6__tar.gz → 1.3.8__tar.gz

{relib-1.3.6 → relib-1.3.8}/PKG-INFO

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

{relib-1.3.6 → relib-1.3.8}/pyproject.toml

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

{relib-1.3.6 → relib-1.3.8}/relib/dict_utils.py

@@ -1,6 +1,8 @@
 from typing import Any, Callable, Iterable, overload
 from .type_utils import as_any
-from .types import K1, K2, K3, K4, K5, K6, K, T, U
+from .types import K1, K2, K3, K4, K5, K6, T1, T2, K, T, U
+
+sentinel = object()
 
 __all__ = [
   "deep_dict_pairs", "deepen_dict", "dict_by", "dict_firsts",
@@ -10,6 +12,7 @@ __all__ = [
   "map_dict", "merge_dicts",
   "omit",
   "pick",
+  "remap",
   "tuple_by",
 ]
 
@@ -21,11 +24,14 @@ def merge_dicts(*dicts: dict[K, T]) -> dict[K, T]:
     result |= d
   return result
 
-def omit(d: dict[K, T], keys: Iterable[K]) -> dict[K, T]:
+def omit(d: dict[K, T], keys: Iterable[K], optional=False) -> dict[K, T]:
   if keys:
     d = dict(d)
     for key in keys:
-      del d[key]
+      try:
+        del d[key]
+      except KeyError if optional else ():
+        pass
   return d
 
 def pick(d: dict[K, T], keys: Iterable[K]) -> dict[K, T]:
@@ -37,9 +43,27 @@ def dict_by(keys: Iterable[K], values: Iterable[T]) -> dict[K, T]:
 def tuple_by(d: dict[K, T], keys: Iterable[K]) -> tuple[T, ...]:
   return tuple(d[key] for key in keys)
 
-def map_dict(fn: Callable[[T], U], d: dict[K, T]) -> dict[K, U]:
+def map_dict(fn: Callable[[T], T], d: dict[K, T]) -> dict[K, T]:
   return {key: fn(value) for key, value in d.items()}
 
+@overload
+def remap(d: dict[K1, T1]) -> dict[K1, T1]: ...
+@overload
+def remap(d: dict[K1, T1], *, keys: dict[K1, K2] = {}) -> dict[K2, T1]: ...
+@overload
+def remap(d: dict[K1, T1], *, values: dict[T1, T2] = {}) -> dict[K1, T2]: ...
+@overload
+def remap(d: dict[K1, T1], *, keys: dict[K1, K2], values: dict[T1, T2]) -> dict[K2, T2]: ...
+def remap(d: dict, *, keys=sentinel, values=sentinel) -> dict:
+  match (keys, values):
+    case (dict(), dict()):
+      return {keys[key]: values[value] for key, value in d.items()}
+    case (dict(), _):
+      return {keys[key]: value for key, value in d.items()}
+    case (_, dict()):
+      return {key: values[value] for key, value in d.items()}
+  return d
+
 def key_of(dicts: Iterable[dict[T, U]], key: T) -> list[U]:
   return [d[key] for d in dicts]
 

{relib-1.3.6 → relib-1.3.8}/relib/iter_utils.py

@@ -6,8 +6,7 @@ from .dict_utils import dict_firsts
 from .types import T1, T2, T3, T4, T5, T, U
 
 __all__ = [
-  "as_list",
-  "at",
+  "as_list", "at",
   "chunked",
   "distinct_by", "distinct", "drop_none",
   "first", "flatten",
@@ -15,7 +14,7 @@ __all__ = [
   "list_split",
   "move_value",
   "partition",
-  "reversed_enumerate",
+  "range_of", "reversed_enumerate",
   "seekable", "sort_by",
   "transpose",
 ]
@@ -74,6 +73,9 @@ def partition(iterable: Iterable[tuple[bool, T]]) -> tuple[list[T], list[T]]:
       false_values.append(value)
   return true_values, false_values
 
+def range_of(values: Sequence) -> range:
+  return range(len(values))
+
 class seekable(Generic[T]):
   def __init__(self, iterable: Iterable[T]):
     self.index = 0
@@ -124,6 +126,11 @@ class seekable(Generic[T]):
     with self.freeze():
       return list(islice(self, count))
 
+  def forgetful(self) -> Iterable[T]:
+    for value in self:
+      self.clear()
+      yield value
+
 @overload
 def chunked(values: Iterable[T], *, num_chunks: int, chunk_size=None) -> list[list[T]]: ...
 @overload

{relib-1.3.6 → relib-1.3.8}/uv.lock

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