relib 1.2.5__tar.gz → 1.2.7__tar.gz

{relib-1.2.5 → relib-1.2.7}/PKG-INFO

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

{relib-1.2.5 → relib-1.2.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "relib"
-version = "1.2.5"
+version = "1.2.7"
 requires-python = ">=3.12"
 dependencies = []
 authors = [

{relib-1.2.5 → relib-1.2.7}/relib/system.py

@@ -16,9 +16,9 @@ def read_json(path: Path) -> Any:
   with path.open("r") as f:
     return json.load(f)
 
-def write_json(path: Path, obj: Any) -> None:
+def write_json(path: Path, obj: Any, indent: None | int = None) -> None:
   with path.open("w") as f:
-    return json.dump(obj, f)
+    return json.dump(obj, f, indent=indent)
 
 def clear_console() -> None:
   os.system("cls" if os.name == "nt" else "clear")
@@ -32,18 +32,19 @@ 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: int, progress=False) -> list[T]:
+async def roll_tasks[T](tasks: Iterable[Awaitable[T]], workers=default_num_workers, progress=False) -> list[T]:
   semaphore = asyncio.Semaphore(workers)
   if not progress:
     return await asyncio.gather(*(worker(task, semaphore) for task in tasks))
+
   from tqdm import tqdm
-  assert isinstance(tasks, (list, tuple, set, frozenset, dict))
+  tasks = tasks if isinstance(tasks, list) else list(tasks)
   with tqdm(total=len(tasks)) as pbar:
-    update = lambda: pbar.update(1)
+    update = functools.partial(pbar.update, 1)
     return await asyncio.gather(*(worker(task, semaphore, update) for task in tasks))
 
-def as_async(num_workers=default_num_workers) -> Callable[[Callable[P, R]], Callable[P, Awaitable[R]]]:
-  executor = ThreadPoolExecutor(max_workers=num_workers)
+def as_async(workers=default_num_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]]:
     @functools.wraps(func)
@@ -53,5 +54,15 @@ def as_async(num_workers=default_num_workers) -> Callable[[Callable[P, R]], Call
       fn_call = functools.partial(ctx.run, func, *args, **kwargs)
       return await loop.run_in_executor(executor, fn_call)
     return wrapper
+  return on_fn
+
+def async_limit(workers=default_num_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]]:
+    @functools.wraps(func)
+    async def wrapper(*args: P.args, **kwargs: P.kwargs) -> R:
+      async with semaphore:
+        return await func(*args, **kwargs)
+    return wrapper
   return on_fn

{relib-1.2.5 → relib-1.2.7}/uv.lock

@@ -106,7 +106,7 @@ wheels = [
 
 [[package]]
 name = "relib"
-version = "1.2.5"
+version = "1.2.7"
 source = { editable = "." }
 
 [package.dev-dependencies]