relib 1.2.10__py3-none-any.whl → 1.2.11__py3-none-any.whl

relib/utils.py

@@ -1,6 +1,6 @@
 import re
 from itertools import chain
-from typing import Any, Callable, Iterable, overload
+from typing import Any, Callable, Iterable, Literal, overload
 
 __all__ = [
   "noop",
@@ -84,12 +84,9 @@ def dict_firsts[T, K](pairs: Iterable[tuple[K, T]]) -> dict[K, T]:
 def distinct_by[T](pairs: Iterable[tuple[object, T]]) -> list[T]:
   return list(dict_firsts(pairs).values())
 
-def sort_by[T](pairs: Iterable[tuple[object, T]]) -> list[T]:
-  pair_list: list[Any] = list(pairs)
-  pair_list.sort(key=lambda p: p[0])
-  for i in range(len(pair_list)):
-    pair_list[i] = pair_list[i][1]
-  return pair_list
+def sort_by[T](pairs: Iterable[tuple[Any, T]]) -> list[T]:
+  pairs = sorted(pairs, key=lambda p: p[0])
+  return [v for _, v in pairs]
 
 def first[T](iterable: Iterable[T]) -> T | None:
   return next(iter(iterable), None)
@@ -161,10 +158,36 @@ def dict_by[T, K](keys: Iterable[K], values: Iterable[T]) -> dict[K, T]:
 def tuple_by[T, K](d: dict[K, T], keys: Iterable[K]) -> tuple[T, ...]:
   return tuple(d[key] for key in keys)
 
-def flatten[T](iterable: Iterable[Iterable[T]]) -> list[T]:
-  return list(chain.from_iterable(iterable))
+@overload
+def flatten[T](iterable: Iterable[T], depth: Literal[0]) -> list[T]: ...
+@overload
+def flatten[T](iterable: Iterable[Iterable[T]], depth: Literal[1] = 1) -> list[T]: ...
+@overload
+def flatten[T](iterable: Iterable[Iterable[Iterable[T]]], depth: Literal[2]) -> list[T]: ...
+@overload
+def flatten[T](iterable: Iterable[Iterable[Iterable[Iterable[T]]]], depth: Literal[3]) -> list[T]: ...
+@overload
+def flatten[T](iterable: Iterable[Iterable[Iterable[Iterable[Iterable[T]]]]], depth: Literal[4]) -> list[T]: ...
+@overload
+def flatten(iterable: Iterable, depth: int) -> list: ...
+
+def flatten(iterable: Iterable, depth: int = 1) -> list:
+  for _ in range(depth):
+    iterable = chain.from_iterable(iterable)
+  return list(iterable)
+
+@overload
+def transpose[T1, T2](tuples: Iterable[tuple[T1, T2]], default_num_returns: int = 0) -> tuple[list[T1], list[T2]]: ...
+@overload
+def transpose[T1, T2, T3](tuples: Iterable[tuple[T1, T2, T3]], default_num_returns: int = 0) -> tuple[list[T1], list[T2], list[T3]]: ...
+@overload
+def transpose[T1, T2, T3, T4](tuples: Iterable[tuple[T1, T2, T3, T4]], default_num_returns: int = 0) -> tuple[list[T1], list[T2], list[T3], list[T4]]: ...
+@overload
+def transpose[T1, T2, T3, T4, T5](tuples: Iterable[tuple[T1, T2, T3, T4, T5]], default_num_returns: int = 0) -> tuple[list[T1], list[T2], list[T3], list[T4], list[T5]]: ...
+@overload
+def transpose(tuples: Iterable[tuple], default_num_returns: int = 0) -> tuple[list, ...]: ...
 
-def transpose(tuples, default_num_returns=0):
+def transpose(tuples: Iterable[tuple], default_num_returns=0) -> tuple[list, ...]:
   output = tuple(zip(*tuples))
   if not output:
     return ([],) * default_num_returns

{relib-1.2.10.dist-info → relib-1.2.11.dist-info}/METADATA

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

{relib-1.2.10.dist-info → relib-1.2.11.dist-info}/RECORD

@@ -2,8 +2,8 @@ relib/__init__.py,sha256=4_nmex7mRhCwdtLF8k0XLbxxPs-UeN2sP-EEImm5JGs,126
 relib/hashing.py,sha256=DB_fnkj0ls01FgZbf4nPFHl4EBU8X_0OrmDvty4HlRE,6020
 relib/measure_duration.py,sha256=LCTo_D_qReNprD3fhtJ0daeWycS6xQE_cwxeg2_h0xo,456
 relib/system.py,sha256=3RWmSweTCQtB1wzsgpUqcAsMo6TIhVRq2oSt28Ul_1E,2733
-relib/utils.py,sha256=A6GenRxjzI-1R80QD3G8Xq-E72LezIbItuHeBSaYnHg,8362
-relib-1.2.10.dist-info/METADATA,sha256=q2wmDYkUk7f_D3WxRQloVtqjOpI6_Z3h5AFt9qXFEto,1296
-relib-1.2.10.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-relib-1.2.10.dist-info/licenses/LICENSE,sha256=9xVsdtv_-uSyY9Xl9yujwAPm4-mjcCLeVy-ljwXEWbo,1059
-relib-1.2.10.dist-info/RECORD,,
+relib/utils.py,sha256=bwOJXfsNx5nZf1fZxWtT_hLwjTbWo2cP9qfsqca1aHI,9680
+relib-1.2.11.dist-info/METADATA,sha256=WxEfPUA2Ox2t1ulJZFIXX5ynIOf1jw1u6iWbvtcx9HI,1296
+relib-1.2.11.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+relib-1.2.11.dist-info/licenses/LICENSE,sha256=9xVsdtv_-uSyY9Xl9yujwAPm4-mjcCLeVy-ljwXEWbo,1059
+relib-1.2.11.dist-info/RECORD,,