feat: add UpdateDiagnosis command SDK definition

canvas-medical · Sep 27, 2024 · f751d07 · f751d07
1 parent ad56f11
commit f751d07
Show file tree

Hide file tree

Showing 5 changed files with 47 additions and 2 deletions.
diff --git a/canvas_generated/messages/effects_pb2.py b/canvas_generated/messages/effects_pb2.py
diff --git a/canvas_generated/messages/effects_pb2.pyi b/canvas_generated/messages/effects_pb2.pyi
@@ -126,6 +126,11 @@ class EffectType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
     DELETE_VITALS_COMMAND: _ClassVar[EffectType]
     COMMIT_VITALS_COMMAND: _ClassVar[EffectType]
     ENTER_IN_ERROR_VITALS_COMMAND: _ClassVar[EffectType]
+    ORIGINATE_UPDATE_DIAGNOSIS_COMMAND: _ClassVar[EffectType]
+    EDIT_UPDATE_DIAGNOSIS_COMMAND: _ClassVar[EffectType]
+    DELETE_UPDATE_DIAGNOSIS_COMMAND: _ClassVar[EffectType]
+    COMMIT_UPDATE_DIAGNOSIS_COMMAND: _ClassVar[EffectType]
+    ENTER_IN_ERROR_UPDATE_DIAGNOSIS_COMMAND: _ClassVar[EffectType]
     ANNOTATE_PATIENT_CHART_CONDITION_RESULTS: _ClassVar[EffectType]
     ANNOTATE_CLAIM_CONDITION_RESULTS: _ClassVar[EffectType]
 UNKNOWN_EFFECT: EffectType
@@ -247,6 +252,11 @@ EDIT_VITALS_COMMAND: EffectType
 DELETE_VITALS_COMMAND: EffectType
 COMMIT_VITALS_COMMAND: EffectType
 ENTER_IN_ERROR_VITALS_COMMAND: EffectType
+ORIGINATE_UPDATE_DIAGNOSIS_COMMAND: EffectType
+EDIT_UPDATE_DIAGNOSIS_COMMAND: EffectType
+DELETE_UPDATE_DIAGNOSIS_COMMAND: EffectType
+COMMIT_UPDATE_DIAGNOSIS_COMMAND: EffectType
+ENTER_IN_ERROR_UPDATE_DIAGNOSIS_COMMAND: EffectType
 ANNOTATE_PATIENT_CHART_CONDITION_RESULTS: EffectType
 ANNOTATE_CLAIM_CONDITION_RESULTS: EffectType
 

diff --git a/canvas_sdk/commands/__init__.py b/canvas_sdk/commands/__init__.py
@@ -19,6 +19,7 @@
 from canvas_sdk.commands.commands.remove_allergy import RemoveAllergyCommand
 from canvas_sdk.commands.commands.stop_medication import StopMedicationCommand
 from canvas_sdk.commands.commands.task import TaskCommand
+from canvas_sdk.commands.commands.update_diagnosis import UpdateDiagnosisCommand
 from canvas_sdk.commands.commands.update_goal import UpdateGoalCommand
 from canvas_sdk.commands.commands.vitals import VitalsCommand
 
@@ -42,6 +43,7 @@
     "RemoveAllergyCommand",
     "StopMedicationCommand",
     "TaskCommand",
+    "UpdateDiagnosisCommand",
     "UpdateGoalCommand",
     "VitalsCommand",
 )
diff --git a/canvas_sdk/commands/commands/update_diagnosis.py b/canvas_sdk/commands/commands/update_diagnosis.py
@@ -0,0 +1,27 @@
+from canvas_sdk.commands.base import _BaseCommand as BaseCommand
+
+
+class UpdateDiagnosisCommand(BaseCommand):
+    """A class for managing an Update Diagnosis command within a specific note."""
+
+    class Meta:
+        key = "updateDiagnosis"
+        commit_required_fields = (
+            "condition_code",
+            "new_condition_code",
+        )
+
+    condition_code: str | None = None
+    new_condition_code: str | None = None
+    background: str | None = None
+    narrative: str | None = None
+
+    @property
+    def values(self) -> dict:
+        """The Update Diagnosis command's field values."""
+        return {
+            "condition_code": self.condition_code,
+            "new_condition_code": self.new_condition_code,
+            "background": self.background,
+            "narrative": self.narrative,
+        }
diff --git a/protobufs/canvas_generated/messages/effects.proto b/protobufs/canvas_generated/messages/effects.proto
@@ -149,6 +149,12 @@ enum EffectType {
   COMMIT_VITALS_COMMAND = 121;
   ENTER_IN_ERROR_VITALS_COMMAND = 122;
 
+  ORIGINATE_UPDATE_DIAGNOSIS_COMMAND = 123;
+  EDIT_UPDATE_DIAGNOSIS_COMMAND = 124;
+  DELETE_UPDATE_DIAGNOSIS_COMMAND = 125;
+  COMMIT_UPDATE_DIAGNOSIS_COMMAND = 126;
+  ENTER_IN_ERROR_UPDATE_DIAGNOSIS_COMMAND = 127;
+
   ANNOTATE_PATIENT_CHART_CONDITION_RESULTS = 200;
 
   ANNOTATE_CLAIM_CONDITION_RESULTS = 300;