relib 1.3.3__py3-none-any.whl → 1.3.5__py3-none-any.whl

relib/dict_utils.py

@@ -2,8 +2,8 @@ from typing import Any, Callable, Iterable, overload
 from .type_utils import as_any
 
 __all__ = [
-  "deepen_dict", "dict_by", "dict_firsts",
-  "flatten_dict_inner", "flatten_dict",
+  "deep_dict_pairs", "deepen_dict", "dict_by", "dict_firsts",
+  "flatten_dict",
   "get_at", "group",
   "key_of",
   "map_dict", "merge_dicts",
@@ -62,15 +62,15 @@ 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 flatten_dict_inner(d, prefix=()):
+def deep_dict_pairs(d, prefix=()):
   for key, value in d.items():
     if not isinstance(value, dict) or value == {}:
       yield prefix + (key,), value
     else:
-      yield from flatten_dict_inner(value, prefix + (key,))
+      yield from deep_dict_pairs(value, prefix + (key,))
 
 def flatten_dict(deep_dict: dict, prefix=()) -> dict:
-  return dict(flatten_dict_inner(deep_dict, prefix))
+  return dict(deep_dict_pairs(deep_dict, prefix))
 
 @overload
 def deepen_dict[K1, U](d: dict[tuple[K1], U]) -> dict[K1, U]: ...

relib/io_utils.py

@@ -1,8 +1,10 @@
 import json
 from pathlib import Path
-from typing import Any
+from typing import Any, Iterable
 
 __all__ = [
+  "clear_directory",
+  "empty_dirs",
   "read_json",
   "write_json",
 ]
@@ -19,3 +21,21 @@ def write_json(path: Path, obj: object, indent: None | int = None) -> None:
   with path.open("w") as f:
     separators = (",", ":") if indent is None else None
     return json.dump(obj, f, indent=indent, separators=separators)
+
+def empty_dirs(path: Path) -> Iterable[Path]:
+  nonempty_count = 0
+  for child in path.iterdir():
+    nonempty_count += 1
+    if child.is_dir():
+      for grand_child in empty_dirs(child):
+        yield grand_child
+        nonempty_count -= child == grand_child
+  if nonempty_count == 0:
+    yield path
+
+def clear_directory(path: Path):
+  if path.is_dir():
+    for file in path.glob("**/.DS_Store"):
+      file.unlink()
+    for directory in empty_dirs(path):
+      directory.rmdir()

relib/iter_utils.py

@@ -1,9 +1,11 @@
 from contextlib import contextmanager
 from itertools import chain, islice
-from typing import Any, Iterable, Literal, Self, overload
+from typing import Any, Iterable, Literal, Self, Sequence, overload
 from .dict_utils import dict_firsts
 
 __all__ = [
+  "as_list",
+  "at",
   "chunked",
   "distinct_by", "distinct", "drop_none",
   "first", "flatten",
@@ -16,6 +18,15 @@ __all__ = [
   "transpose",
 ]
 
+def as_list[T](iterable: Iterable[T]) -> list[T]:
+  return iterable if isinstance(iterable, list) else list(iterable)
+
+def at[T, U](values: Sequence[T], index: int, default: U = None) -> T | U:
+  try:
+    return values[index]
+  except IndexError:
+    return default
+
 def first[T](iterable: Iterable[T]) -> T | None:
   return next(iter(iterable), None)
 
@@ -28,8 +39,8 @@ def distinct[T](iterable: Iterable[T]) -> list[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[Any, T]]) -> list[T]:
-  pairs = sorted(pairs, key=lambda p: p[0])
+def sort_by[T](pairs: Iterable[tuple[Any, T]], reverse=False) -> list[T]:
+  pairs = sorted(pairs, key=lambda p: p[0], reverse=reverse)
   return [v for _, v in pairs]
 
 def move_value[T](iterable: Iterable[T], from_i: int, to_i: int) -> list[T]:
@@ -37,7 +48,7 @@ def move_value[T](iterable: Iterable[T], from_i: int, to_i: int) -> list[T]:
   values.insert(to_i, values.pop(from_i))
   return values
 
-def reversed_enumerate[T](values: list[T] | tuple[T, ...]) -> Iterable[tuple[int, T]]:
+def reversed_enumerate[T](values: Sequence[T] | tuple[T, ...]) -> Iterable[tuple[int, T]]:
   return zip(range(len(values))[::-1], reversed(values))
 
 def intersect[T](*iterables: Iterable[T]) -> list[T]:
@@ -116,7 +127,7 @@ def chunked[T](values: Iterable[T], *, num_chunks: int, chunk_size=None) -> list
 @overload
 def chunked[T](values: Iterable[T], *, num_chunks=None, chunk_size: int) -> list[list[T]]: ...
 def chunked(values, *, num_chunks=None, chunk_size=None):
-  values = values if isinstance(values, list) else list(values)
+  values = as_list(values)
   if isinstance(num_chunks, int):
     chunk_size = (len(values) / num_chunks).__ceil__()
   elif isinstance(chunk_size, int):

relib/runtime_tools.py

@@ -6,6 +6,7 @@ from concurrent.futures import ThreadPoolExecutor
 from functools import partial, wraps
 from time import time
 from typing import Callable, Coroutine, Iterable, ParamSpec, TypeVar
+from .iter_utils import as_list
 from .processing_utils import noop
 
 __all__ = [
@@ -25,7 +26,7 @@ default_executor = ThreadPoolExecutor(max_workers=default_workers)
 
 def raise_if_interrupt():
   if sys.exc_info()[0] in (KeyboardInterrupt, SystemExit):
-     raise
+    raise
 
 def clear_console() -> None:
   os.system("cls" if os.name == "nt" else "clear")
@@ -45,7 +46,7 @@ async def roll_tasks[T](tasks: Iterable[Coro[T]], workers=default_workers, progr
     return await asyncio.gather(*[worker(task, semaphore) for task in tasks])
 
   from tqdm import tqdm
-  tasks = tasks if isinstance(tasks, list) else list(tasks)
+  tasks = as_list(tasks)
   with tqdm(total=len(tasks)) as pbar:
     update = partial(pbar.update, 1)
     return await asyncio.gather(*[worker(task, semaphore, update) for task in tasks])

{relib-1.3.3.dist-info → relib-1.3.5.dist-info}/METADATA

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

relib-1.3.5.dist-info/RECORD

@@ -0,0 +1,11 @@
+relib/__init__.py,sha256=WerjUaM_sNvudjXFudLRtXB7viZWEW1RSinkDjrh4nE,163
+relib/dict_utils.py,sha256=zuUkPI1uiElWUgt3ETdajG7S2grP5mX1qM0t2Cc1zy4,2968
+relib/io_utils.py,sha256=cUyUFhrMCUDfkINhYo32QPaVGz3chqDO3ElymSCoWEg,1086
+relib/iter_utils.py,sha256=xQOyNwYCc9fyKScyd56vx4CVCzZBDDROKVfnJuaInCM,5774
+relib/processing_utils.py,sha256=eMzjlxsEmfvtKafDITBWSp9D5RwegSWsUsvj1FpmBM0,1893
+relib/runtime_tools.py,sha256=50zfIXZwXXU6tGKM-1iS3FEax82vthgp-KNmvmIGlqE,2964
+relib/type_utils.py,sha256=oY96cAAux1JwhXgWFFyqEv_f-wwyPc_Hm6I9Yeisu_M,323
+relib-1.3.5.dist-info/METADATA,sha256=adnaXzA56Yff2cXRK79hAopNMnPs_rfwBBKs95l3neY,1295
+relib-1.3.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+relib-1.3.5.dist-info/licenses/LICENSE,sha256=9xVsdtv_-uSyY9Xl9yujwAPm4-mjcCLeVy-ljwXEWbo,1059
+relib-1.3.5.dist-info/RECORD,,

relib-1.3.3.dist-info/RECORD

@@ -1,11 +0,0 @@
-relib/__init__.py,sha256=WerjUaM_sNvudjXFudLRtXB7viZWEW1RSinkDjrh4nE,163
-relib/dict_utils.py,sha256=jqW6bYSaQMt2AC2KFzDJKyl88idyMttWxXDu3t-fA5I,2980
-relib/io_utils.py,sha256=EtnIGQmLXjoHUPFteB5yPXDD3wGLvH4O3CahlCebXDQ,555
-relib/iter_utils.py,sha256=r9tus9F_obwXsHE8Jk8H9_ZPdOgtBI6WnpDxI6La-to,5477
-relib/processing_utils.py,sha256=eMzjlxsEmfvtKafDITBWSp9D5RwegSWsUsvj1FpmBM0,1893
-relib/runtime_tools.py,sha256=9N8gn3WCfqyS6D3Qs4l2nDja_NxDVVF-7hHIVrXA7Fc,2968
-relib/type_utils.py,sha256=oY96cAAux1JwhXgWFFyqEv_f-wwyPc_Hm6I9Yeisu_M,323
-relib-1.3.3.dist-info/METADATA,sha256=3z1-dTnOlkjYiy54JQ2VQOs0zy5_kyP4Bk33l1YuAoQ,1295
-relib-1.3.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-relib-1.3.3.dist-info/licenses/LICENSE,sha256=9xVsdtv_-uSyY9Xl9yujwAPm4-mjcCLeVy-ljwXEWbo,1059
-relib-1.3.3.dist-info/RECORD,,