Патч

src/clients/vitacore/api.py

@@ -200,6 +200,7 @@ class VITACORE_API(AsyncClient):
                 raise e.UnknownException
 
     async def getHospExaminations(self, patId: str, examId: str):
+        patId = 'b66a85f1-4aaa-4db8-942a-2de44341824e'
         token = await self.get_token()
         req = await self.get(
             '/getHospExaminations',
@@ -210,6 +211,18 @@ class VITACORE_API(AsyncClient):
         match req.status_code:
             case st.HTTP_200_OK:
                 return s.HospExaminationsModel.model_validate(req.json())
+
+            case st.HTTP_206_PARTIAL_CONTENT:
+                error = s.ErrorModel.model_validate(req.json())
+
+                if error.error == 'Пациент не госпитализирован!':
+                    return s.HospExaminationsModel(
+                        EventID='none',
+                        EventDate=datetime.now(UTC),
+                        LpuName='fakeName',
+                        Examinations=[],
+                    )
+
             case _:
                 self.logger.error(req.json())
                 raise e.UnknownException
@@ -351,6 +364,13 @@ class VITACORE_API(AsyncClient):
         match req.status_code:
             case st.HTTP_200_OK:
                 return s.ELNsModel.model_validate(req.json())
+
+            case st.HTTP_206_PARTIAL_CONTENT:
+                error = s.ErrorModel.model_validate(req.json())
+
+                if error.error == 'Пациент не госпитализирован!':
+                    return s.ELNsModel(PatientELNs=[])
+
             case _:
                 self.logger.error(req.json())
                 raise e.UnknownException
@@ -369,3 +389,28 @@ class VITACORE_API(AsyncClient):
             case _:
                 self.logger.error(req.json())
                 raise e.UnknownException
+
+    async def getDiagResultFile(self, resultId: str):
+        token = await self.get_token()
+        req = await self.get(
+            '/getDiagResultFile',
+            params={'resultId': resultId},
+            headers={'Authorization': f'Bearer {token}'},
+        )
+
+        match req.status_code:
+            case st.HTTP_200_OK:
+                return s.DiagResultFileModel.model_validate(req.json())
+
+            case st.HTTP_206_PARTIAL_CONTENT:
+                error = s.ErrorModel.model_validate(req.json())
+
+                if (
+                    error.error == 'Не найдены проведенные исследования по '
+                    'данному идентификатору'
+                ):
+                    return s.DiagResultFileModel(content='')
+
+            case _:
+                self.logger.error(req.json())
+                raise e.UnknownException