agi-med-common 3.5.13__py3-none-any.whl → 3.6.1__py3-none-any.whl

agi_med_common/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "3.5.13"
+__version__ = "3.6.1"
 
 from .logger import LogLevelEnum, logger_init
 from .models import (
@@ -18,3 +18,4 @@ from .models import DiagnosticsXMLTagEnum, MTRSXMLTagEnum, DoctorChoiceXMLTagEnu
 from .utils import make_session_id, read_json
 from .validators import is_file_exist, validate_prompt
 from .xml_parser import XMLParser
+from .parallel_map import parallel_map

agi_med_common/parallel_map.py

@@ -0,0 +1,31 @@
+from typing import Any, Callable, TypeVar
+import concurrent.futures
+from tqdm import tqdm
+
+
+X = TypeVar("X")
+
+
+def parallel_map(
+    func: Callable[..., X],
+    items: list[Any],
+    process: bool = False,
+    multiple_args: bool = False,
+    max_workers: int = 2,
+    show_tqdm: bool = False,
+    desc: str = "",
+) -> list[X]:
+    pool = (
+        concurrent.futures.ProcessPoolExecutor(max_workers=max_workers)
+        if process
+        else concurrent.futures.ThreadPoolExecutor(max_workers=max_workers)
+    )
+    with pool as executor:
+        futures = []
+        for item in items:
+            if multiple_args:
+                futures.append(executor.submit(func, *item))
+            else:
+                futures.append(executor.submit(func, item))
+        results: list[X] = [future.result() for future in tqdm(futures, disable=(not show_tqdm), desc=desc)]
+    return results

{agi_med_common-3.5.13.dist-info → agi_med_common-3.6.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agi_med_common
-Version: 3.5.13
+Version: 3.6.1
 Summary: Сommon for agi-med team
 Author: AGI-MED-TEAM
 Requires-Python: >=3.11

{agi_med_common-3.5.13.dist-info → agi_med_common-3.6.1.dist-info}/RECORD

@@ -1,5 +1,6 @@
-agi_med_common/__init__.py,sha256=fYfAcXePLHvwwFdAP5vbZX0L5HFZuZLKFXij2BLvCUc,537
+agi_med_common/__init__.py,sha256=qJk81VD0GhIv5kgOHlHaN0LMgw9GukSqiSB3fMJ4-I4,575
 agi_med_common/file_storage.py,sha256=GR6_jKZ0o9uwUkOKZd1XaCVOSw0YmFlCOlC6EX8zz0I,1247
+agi_med_common/parallel_map.py,sha256=bC7YGM5sWxSNvaA54Qpu8rlFJTOlRyMw0eWlttkTsW8,886
 agi_med_common/utils.py,sha256=5iurKl5d1zZ_cN4yqtc0_FhrwuNBcDurbWilu28_saE,325
 agi_med_common/validators.py,sha256=R678gjPp-5XbnocRuEtdOQgJyCCOurxwaOe2nT04kSg,705
 agi_med_common/xml_parser.py,sha256=VvLIX_XCZao9i0qqpTVx8nx0vbFXSe8pEbdJdXnj97g,568
@@ -24,7 +25,7 @@ agi_med_common/models/enums/mtrs_label_enum.py,sha256=6emBndt3SCsQVZZFKQYCV2_iyj
 agi_med_common/models/enums/mtrs_xml_tag_enum.py,sha256=6OxuRsrx4b2uBjfrBgm4Y789Ly337_mQXL9VPRCpLyg,273
 agi_med_common/models/enums/state_enum.py,sha256=p7XhDxBxPeeSgbLWitq6JpJLKntS3XMHon_EevbnhEs,163
 agi_med_common/models/enums/track_id_enum.py,sha256=N3KUd97a4xDet_NKZ5URm0qWAAv3ts1l_foOjdlGY4c,532
-agi_med_common-3.5.13.dist-info/METADATA,sha256=p3UYLkQfXfGzVZY8lmaO49NHWRgf_TkHOpJZNBMol2w,521
-agi_med_common-3.5.13.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-agi_med_common-3.5.13.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
-agi_med_common-3.5.13.dist-info/RECORD,,
+agi_med_common-3.6.1.dist-info/METADATA,sha256=zXIejfAPXm8Q1u_8OCjTP_v8bTSWCYExG0IumlienhI,520
+agi_med_common-3.6.1.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+agi_med_common-3.6.1.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
+agi_med_common-3.6.1.dist-info/RECORD,,