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

relib/__init__.py

@@ -1,46 +1,4 @@
-from .utils import (
-  noop,
-  non_none,
-  as_any,
-  list_split,
-  drop_none,
-  distinct,
-  first,
-  move_value,
-  transpose_dict,
-  make_combinations_by_dict,
-  merge_dicts,
-  intersect,
-  ensure_tuple,
-  key_of,
-  omit,
-  pick,
-  dict_by,
-  tuple_by,
-  flatten,
-  transpose,
-  map_dict,
-  deepen_dict,
-  flatten_dict_inner,
-  flatten_dict,
-  group,
-  reversed_enumerate,
-  get_at,
-  for_each,
-  sized_partitions,
-  num_partitions,
-  df_from_array,
-  StrFilter,
-  str_filterer,
-)
-from .system import (
-  read_json,
-  write_json,
-  clear_console,
-  console_link,
-  roll_tasks,
-  as_async,
-  async_limit,
-)
+from .utils import *
+from .system import *
 from .hashing import hash, hash_obj
 from .measure_duration import measure_duration

relib/system.py

@@ -8,20 +8,31 @@ from pathlib import Path
 from typing import Any, Awaitable, Callable, Iterable, ParamSpec, TypeVar
 from .utils import noop
 
+__all__ = [
+  "read_json",
+  "write_json",
+  "clear_console",
+  "console_link",
+  "roll_tasks",
+  "as_async",
+  "async_limit",
+]
+
 P = ParamSpec("P")
 R = TypeVar("R")
-default_num_workers = min(32, (os.cpu_count() or 1) + 4)
-_default_json: Any = {}
+default_workers = min(32, (os.cpu_count() or 1) + 4)
+default_sentinel = object()
 
-def read_json(path: Path, default=_default_json) -> Any:
-  if default is not _default_json and not path.exists():
+def read_json(path: Path, default=default_sentinel) -> Any:
+  if default is not default_sentinel and not path.exists():
     return default
   with path.open("r") as f:
     return json.load(f)
 
-def write_json(path: Path, obj: Any, indent: None | int = None) -> None:
+def write_json(path: Path, obj: object, indent: None | int = None) -> None:
   with path.open("w") as f:
-    return json.dump(obj, f, indent=indent)
+    separators = (",", ":") if indent is None else None
+    return json.dump(obj, f, indent=indent, separators=separators)
 
 def clear_console() -> None:
   os.system("cls" if os.name == "nt" else "clear")
@@ -35,7 +46,7 @@ async def worker[T](task: Awaitable[T], semaphore: asyncio.Semaphore, update=noo
     update()
     return result
 
-async def roll_tasks[T](tasks: Iterable[Awaitable[T]], workers=default_num_workers, progress=False) -> list[T]:
+async def roll_tasks[T](tasks: Iterable[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])
@@ -46,7 +57,7 @@ async def roll_tasks[T](tasks: Iterable[Awaitable[T]], workers=default_num_worke
     update = functools.partial(pbar.update, 1)
     return await asyncio.gather(*[worker(task, semaphore, update) for task in tasks])
 
-def as_async(workers=default_num_workers) -> Callable[[Callable[P, R]], Callable[P, Awaitable[R]]]:
+def as_async(workers=default_workers) -> Callable[[Callable[P, R]], Callable[P, Awaitable[R]]]:
   executor = ThreadPoolExecutor(max_workers=workers)
 
   def on_fn(func: Callable[P, R]) -> Callable[P, Awaitable[R]]:
@@ -59,7 +70,7 @@ def as_async(workers=default_num_workers) -> Callable[[Callable[P, R]], Callable
     return wrapper
   return on_fn
 
-def async_limit(workers=default_num_workers) -> Callable[[Callable[P, Awaitable[R]]], Callable[P, Awaitable[R]]]:
+def async_limit(workers=default_workers) -> Callable[[Callable[P, Awaitable[R]]], Callable[P, Awaitable[R]]]:
   semaphore = asyncio.Semaphore(workers)
 
   def on_fn(func: Callable[P, Awaitable[R]]) -> Callable[P, Awaitable[R]]:

relib/utils.py

@@ -2,9 +2,56 @@ import re
 from itertools import chain
 from typing import Any, Callable, Iterable, overload
 
+__all__ = [
+  "noop",
+  "clamp",
+  "non_none",
+  "as_any",
+  "list_split",
+  "drop_none",
+  "distinct",
+  "dict_firsts",
+  "distinct_by",
+  "sort_by",
+  "first",
+  "move_value",
+  "transpose_dict",
+  "make_combinations_by_dict",
+  "merge_dicts",
+  "intersect",
+  "ensure_tuple",
+  "key_of",
+  "omit",
+  "pick",
+  "dict_by",
+  "tuple_by",
+  "flatten",
+  "transpose",
+  "map_dict",
+  "deepen_dict",
+  "flatten_dict_inner",
+  "flatten_dict",
+  "group",
+  "reversed_enumerate",
+  "get_at",
+  "for_each",
+  "sized_partitions",
+  "num_partitions",
+  "df_from_array",
+  "StrFilter",
+  "str_filterer",
+]
+
 def noop() -> None:
   pass
 
+@overload
+def clamp(value: int, low: int, high: int) -> int: ...
+@overload
+def clamp(value: float, low: float, high: float) -> float: ...
+def clamp(value: float, low: float, high: float) -> float:
+    return max(low, min(value, high))
+
 def non_none[T](obj: T | None) -> T:
   assert obj is not None
   return obj
@@ -12,28 +59,45 @@ def non_none[T](obj: T | None) -> T:
 def as_any(obj: Any) -> Any:
   return obj
 
-def list_split[T](l: list[T], sep: T) -> list[list[T]]:
-  l = [sep, *l, sep]
-  split_at = [i for i, x in enumerate(l) if x is sep]
+def list_split[T](iterable: Iterable[T], sep: T) -> list[list[T]]:
+  values = [sep, *iterable, sep]
+  split_at = [i for i, x in enumerate(values) if x is sep]
   ranges = list(zip(split_at[0:-1], split_at[1:]))
   return [
-    l[start + 1:end]
+    values[start + 1:end]
     for start, end in ranges
   ]
 
 def drop_none[T](iterable: Iterable[T | None]) -> list[T]:
   return [x for x in iterable if x is not None]
 
-def distinct[T](items: Iterable[T]) -> list[T]:
-  return list(dict.fromkeys(items))
+def distinct[T](iterable: Iterable[T]) -> list[T]:
+  return list(dict.fromkeys(iterable))
+
+def dict_firsts[T, K](pairs: Iterable[tuple[K, T]]) -> dict[K, T]:
+  result: dict[K, T] = {}
+  for key, item in pairs:
+    if key not in result:
+      result[key] = item
+  return result
+
+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 first[T](iterable: Iterable[T]) -> T | None:
   return next(iter(iterable), None)
 
 def move_value[T](iterable: Iterable[T], from_i: int, to_i: int) -> list[T]:
-  l = list(iterable)
-  l.insert(to_i, l.pop(from_i))
-  return l
+  values = list(iterable)
+  values.insert(to_i, values.pop(from_i))
+  return values
 
 def transpose_dict(des):
   if isinstance(des, list):
@@ -53,7 +117,7 @@ def transpose_dict(des):
   raise ValueError("transpose_dict only accepts dict or list")
 
 def make_combinations_by_dict(des, keys=None, pairs=[]):
-  keys = sorted(des.keys()) if keys == None else keys
+  keys = sorted(des.keys()) if keys is None else keys
   if len(keys) == 0:
     return [dict(pairs)]
   key = keys[0]
@@ -148,8 +212,8 @@ def group[T, K](pairs: Iterable[tuple[K, T]]) -> dict[K, list[T]]:
     values_by_key.setdefault(key, []).append(value)
   return values_by_key
 
-def reversed_enumerate[T](l: list[T] | tuple[T, ...]) -> Iterable[tuple[int, T]]:
-  return zip(reversed(range(len(l))), reversed(l))
+def reversed_enumerate[T](values: list[T] | tuple[T, ...]) -> Iterable[tuple[int, T]]:
+  return zip(reversed(range(len(values))), reversed(values))
 
 def get_at[T](d: dict, keys: Iterable[Any], default: T) -> T:
   try:

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

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

relib-1.2.10.dist-info/RECORD

@@ -0,0 +1,9 @@
+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-1.2.8.dist-info/RECORD

@@ -1,9 +0,0 @@
-relib/__init__.py,sha256=pypVcr0ctaSdFZmwMnZvtioTbms3MlWfAcrnPVdZxog,689
-relib/hashing.py,sha256=DB_fnkj0ls01FgZbf4nPFHl4EBU8X_0OrmDvty4HlRE,6020
-relib/measure_duration.py,sha256=LCTo_D_qReNprD3fhtJ0daeWycS6xQE_cwxeg2_h0xo,456
-relib/system.py,sha256=RO8pd5eUmucbAxgfFi7UWHH5Cdp0aM6O1oe5cylrc4k,2527
-relib/utils.py,sha256=4fM0zsAdfrM2w9Q1YKcTfFG5kDzoNgD2amkgfTG39OY,6933
-relib-1.2.8.dist-info/METADATA,sha256=b8RvpgbcT7Y9zq6r-kwnnCZgql_eAdG41Y83-q2YIXg,1295
-relib-1.2.8.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-relib-1.2.8.dist-info/licenses/LICENSE,sha256=9xVsdtv_-uSyY9Xl9yujwAPm4-mjcCLeVy-ljwXEWbo,1059
-relib-1.2.8.dist-info/RECORD,,