utg-base 1.6.1__py3-none-any.whl → 1.7.0__py3-none-any.whl

utg_base/utils/thread.py

@@ -0,0 +1,67 @@
+import multiprocessing
+from concurrent.futures import ThreadPoolExecutor, as_completed
+from typing import Callable, Any, Iterable, Tuple, Dict, Union
+
+
+class ThreadPoolException(Exception):
+    """Exception to wrap the error in each task."""
+
+    def __init__(self, func: str, exception: Exception):
+        self.func = func
+        self.exception = exception
+        super().__init__(f"Exception in {func}: {exception}")
+
+
+def parallel_execute(
+        *tasks: Union[
+            Callable,  # func
+            Tuple[Callable, Iterable],  # (func, args)
+            Tuple[Callable, Iterable, Dict[str, Any]]  # (func, args, kwargs)
+        ],
+        max_workers: int = None
+) -> list[Any]:
+    """
+    Executes multiple functions in parallel.
+    Each can be without arguments or with args/kwargs.
+
+    Example:
+        parallel_execute(func1, func2)
+        parallel_execute((func1, (1, 2)), (func2, (), {"x": 5}))
+
+    The sequence is preserved.
+
+    :return: Result list or ThreadPoolException objects
+    """
+    if not tasks:
+        return []
+
+    cpu_count = multiprocessing.cpu_count()
+    max_workers = max_workers or min(cpu_count * 10, len(tasks))
+    results = [None] * len(tasks)
+
+    with ThreadPoolExecutor(max_workers=max_workers) as executor:
+        futures = {}
+
+        for i, task in enumerate(tasks):
+            # Task unpacking
+            if callable(task):
+                func, args, kwargs = task, (), {}
+            elif isinstance(task, tuple):
+                func = task[0]
+                args = task[1] if len(task) > 1 else ()
+                kwargs = task[2] if len(task) > 2 else {}
+            else:
+                raise ValueError(f"Invalid task format: {task}")
+
+            futures[executor.submit(func, *args, **kwargs)] = i
+
+        for future in as_completed(futures):
+            idx = futures[future]
+            func = tasks[idx][0] if isinstance(tasks[idx], tuple) else tasks[idx]
+            func_name = getattr(func, "__name__", str(func))
+            try:
+                results[idx] = future.result()
+            except Exception as e:
+                results[idx] = ThreadPoolException(func=func_name, exception=e)
+
+    return results

{utg_base-1.6.1.dist-info → utg_base-1.7.0.dist-info}/METADATA

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: utg-base
-Version: 1.6.1
+Version: 1.7.0
 Summary: UTG Base Package
-Author: Rovshen
-Author-email: rovshenashirov1619@gmail.com
+Author: Olimboy
+Author-email: shavkatov.olimboy@mail.ru
 Requires-Python: >=3.14,<4.0
 Classifier: Programming Language :: Python :: 3
 Requires-Dist: celery (>=5.5.3,<6.0.0)

{utg_base-1.6.1.dist-info → utg_base-1.7.0.dist-info}/RECORD

@@ -49,7 +49,8 @@ utg_base/utils/date.py,sha256=thcbK6RgTUYZfs4_vW5ucuu2e8H0rei6tv7SEC72iwM,3612
 utg_base/utils/dict_util.py,sha256=ipdCZO8aTukGQ319OWHb2Ij5MNtV-FioJQ4qCX3Th48,758
 utg_base/utils/response_processors.py,sha256=WdZQL49wOJqCIY2MucAI6sez_llCqih0v_ltQa-mv7k,687
 utg_base/utils/sql.py,sha256=rqIWcSjdjIMszdRnsnhV5TTYB8W17RPOujIQA9rKC_Y,762
+utg_base/utils/thread.py,sha256=4RqRnwtyHymY-dNcuPrMSTamE2V7wCMVfzzyIb0P4TI,2191
 utg_base/utils/translation.py,sha256=GxJHUt0iar_0E7RWBPbeLFQ4DhgXBjffHCmxfKyjFtk,463
-utg_base-1.6.1.dist-info/METADATA,sha256=VFG9y4kzasChqU2btKkXSwCifVjnxEoaIFnIV9bAhlw,874
-utg_base-1.6.1.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-utg_base-1.6.1.dist-info/RECORD,,
+utg_base-1.7.0.dist-info/METADATA,sha256=1XwqfOqDxhuzkk0Rfk5ZRjasjzBHIvYxfA2PuhFO9pk,871
+utg_base-1.7.0.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+utg_base-1.7.0.dist-info/RECORD,,