agi-med-common 3.5.3__py3-none-any.whl → 3.5.5__py3-none-any.whl

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/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/models/enums/action_enum.py

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

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.dist-info → agi_med_common-3.5.5.dist-info}/METADATA

@@ -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.dist-info → agi_med_common-3.5.5.dist-info}/RECORD

@@ -1,4 +1,5 @@
-agi_med_common/__init__.py,sha256=rKg4nEk2rydeu38borIZ2Mt77RM3X7lGvafWKIThA1A,511
+agi_med_common/__init__.py,sha256=vHBvgayIw-99w9ksvmaANazMbH2LZuv-T92HBbiNl7A,511
+agi_med_common/file_storage.py,sha256=W6MOWBrO20y-xnW3FIR8vABK2PazRk3-Jm853mVXttk,1115
 agi_med_common/utils.py,sha256=4GV1yNZiRG3KYOh859dvOZI899z7_s_YN3aXzthMgwU,399
 agi_med_common/validators.py,sha256=R678gjPp-5XbnocRuEtdOQgJyCCOurxwaOe2nT04kSg,705
 agi_med_common/xml_parser.py,sha256=VvLIX_XCZao9i0qqpTVx8nx0vbFXSe8pEbdJdXnj97g,568
@@ -15,15 +16,15 @@ agi_med_common/models/chat_item/inner_context_item.py,sha256=YXmlNLabKN3O84UWJEN
 agi_med_common/models/chat_item/outer_context_item.py,sha256=IzrwhqbLKR8ZF8-Sd2CyNSa0eqb21NQH7LUqkg9se2Y,1046
 agi_med_common/models/chat_item/replica_item.py,sha256=O4QJlXQ_HaxtH3CEAYdxULCaMbJ6nTHPEDsZ6ZeoDDw,1507
 agi_med_common/models/enums/__init__.py,sha256=wnI9ub2F1L-TTB2jIpVF3OTLFr0QzcQjW3pX4P45HXo,377
-agi_med_common/models/enums/action_enum.py,sha256=ZZtSBlOsQWvJ1UYqmsMsVLrzO1kJdImqJMqI0_KSV9M,287
+agi_med_common/models/enums/action_enum.py,sha256=-aFNyHQ1yyCJedSrFubcHuLAiLZLP6KSs_5ff9lrR3I,320
 agi_med_common/models/enums/diagnostics_xml_tag_enum.py,sha256=aiUJpe5joh5EOcJ5sk1Xs003XkRry2CBL6H-bIEiapA,278
 agi_med_common/models/enums/doctor_choice_xml_tag_enum.py,sha256=MsSFI0vFHJVujhoj7fNX-fcGVV3jMF_nEDOJnjzh4ac,291
 agi_med_common/models/enums/moderation_label_enum.py,sha256=lbGG4Pu7cQp57uEyQEptNeYsQszK6yyxJ5AEhq4SGkE,292
 agi_med_common/models/enums/mtrs_label_enum.py,sha256=6emBndt3SCsQVZZFKQYCV2_iyjjmZEhwejJKJu39ZAw,257
 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=YoP15BTGjaFDxCOtjsqfwuMTW-BhcaQyEdxLfK7RQUA,454
-agi_med_common-3.5.3.dist-info/METADATA,sha256=LEEWAye9xaHIacpIc7HkAvSXFsBQXu4LJADAEiLtLmA,520
-agi_med_common-3.5.3.dist-info/WHEEL,sha256=beeZ86-EfXScwlR_HKu4SllMC9wUEj_8Z_4FJ3egI2w,91
-agi_med_common-3.5.3.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
-agi_med_common-3.5.3.dist-info/RECORD,,
+agi_med_common/models/enums/track_id_enum.py,sha256=KM4DbJ6fzK4xPSghKBaTdAhz_SlTSLxJvnKJoMjc9QE,501
+agi_med_common-3.5.5.dist-info/METADATA,sha256=lVIYXrkwvbdsgSzL5IPKFK-ZahrgODJQXx828rXdgf0,520
+agi_med_common-3.5.5.dist-info/WHEEL,sha256=DK49LOLCYiurdXXOXwGJm6U4DkHkg4lcxjhqwRa0CP4,91
+agi_med_common-3.5.5.dist-info/top_level.txt,sha256=26o565jF_7wYQj7-YJfTedtT9yDxDcf8RNikOYuPq78,15
+agi_med_common-3.5.5.dist-info/RECORD,,

{agi_med_common-3.5.3.dist-info → agi_med_common-3.5.5.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (76.1.0)
+Generator: setuptools (78.0.2)
 Root-Is-Purelib: true
 Tag: py3-none-any