agi-med-common 3.5.3__tar.gz → 3.5.5__tar.gz

{agi_med_common-3.5.3 → agi_med_common-3.5.5}/PKG-INFO

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

{agi_med_common-3.5.3 → agi_med_common-3.5.5}/src/agi_med_common/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "3.5.3"
+__version__ = "3.5.5"
 
 from .logger import LogLevelEnum, logger_init
 from .models import (

agi_med_common-3.5.5/src/agi_med_common/file_storage.py

@@ -0,0 +1,39 @@
+import string
+from hashlib import md5
+from pathlib import Path
+
+ResourceId = str
+ASCII_DIGITS = set(string.ascii_lowercase + string.digits)
+
+
+def _validate_exist(files_dir):
+    if not files_dir.exists():
+        err = f"Failed to access file-storage directory: {files_dir}"
+        raise OSError(err)
+
+
+def _validate_dtype(dtype: str):
+    if all(map(ASCII_DIGITS.__contains__, dtype)):
+        return
+    raise ValueError(f'Bad dtype: {dtype}')
+
+
+class FileStorage:
+    def __init__(self, files_dir):
+        self.files_dir = Path(files_dir)
+        self.files_dir.mkdir(exist_ok=True)
+        _validate_exist(self.files_dir)
+
+    def upload(self, content: bytes, dtype: str) -> ResourceId | None:
+        _validate_dtype(dtype)
+        fname_hash = md5(content).hexdigest()
+        fname = f'{fname_hash}.{dtype}'
+        fpath = self.files_dir / fname
+        fpath.write_bytes(content)
+        return str(fpath)
+
+    def download(self, rid: ResourceId) -> bytes:
+        return Path(rid).read_bytes()
+
+    def is_valid(self, rid: ResourceId) -> bytes:
+        return Path(rid).exists() and Path(rid).is_file()

{agi_med_common-3.5.3 → agi_med_common-3.5.5}/src/agi_med_common/models/enums/action_enum.py

@@ -13,5 +13,5 @@ class ActionEnum(StrEnum):
 
     DIAGNOSIS = auto()
     REPORT = auto()
-
+    FINANCIAL_EXTRACTION = auto()
     FINAL = auto()

{agi_med_common-3.5.3 → agi_med_common-3.5.5}/src/agi_med_common/models/enums/track_id_enum.py

@@ -13,3 +13,5 @@ class TrackIdEnum(StrEnum):
     CONSULTATION = auto()
     DUMMY = auto()
     COMMON_CONSULTATION = auto()
+    FINANCIER = auto()
+    MULTIMODAL = auto()

{agi_med_common-3.5.3 → agi_med_common-3.5.5}/src/agi_med_common.egg-info/PKG-INFO

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

{agi_med_common-3.5.3 → agi_med_common-3.5.5}/src/agi_med_common.egg-info/SOURCES.txt

@@ -2,6 +2,7 @@ README.md
 pyproject.toml
 requirements.txt
 src/agi_med_common/__init__.py
+src/agi_med_common/file_storage.py
 src/agi_med_common/utils.py
 src/agi_med_common/validators.py
 src/agi_med_common/xml_parser.py