Data Migration

.gitignore

@@ -357,4 +357,4 @@ secrets.sh
 
 jwks.b64.txt
 
-care_db.dump
+care_db.dump*

care/emr/models/__init__.py

@@ -1,3 +1,4 @@
+# isort: skip_file
 from .allergy_intolerance import *  # noqa F403
 from .condition import *  # noqa F403
 from .observation import *  # noqa F403

care/emr/resources/medication/administration/spec.py

@@ -34,7 +34,6 @@ class MedicationAdministrationCategory(str, Enum):
     inpatient = "inpatient"
     outpatient = "outpatient"
     community = "community"
-    discharge = "discharge"
 
 
 class MedicationAdministrationPerformerFunction(str, Enum):

care/emr/resources/questionnaire_response/spec.py

@@ -41,7 +41,7 @@ class QuestionnaireResponseReadSpec(EMRResource):
     id: UUID4
     questionnaire: QuestionnaireReadSpec
     subject_id: str
-    responses: list
+    responses: list[QuestionnaireSubmitResult] = []
     encounter: str | None = None
     structured_responses: dict
     structured_response_type: str

care/facility/migrations/0477_historicalpatientregistration_migrated_emr_patient_id_and_more.py

@@ -0,0 +1,73 @@
+# Generated by Django 5.1.4 on 2025-01-14 09:20
+
+import django.db.models.deletion
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("contenttypes", "0002_remove_content_type_name"),
+        ("facility", "0476_facility_default_internal_organization_and_more"),
+        ("users", "0021_rename_gender_user_old_gender_user_geo_organization_and_more"),
+    ]
+
+    operations = [
+        migrations.DeleteModel(
+            name="HistoricalPatientRegistration",
+        ),
+        migrations.AddField(
+            model_name="patientconsultation",
+            name="migrated_emr_encounter_id",
+            field=models.BigIntegerField(blank=True, null=True),
+        ),
+        migrations.AddField(
+            model_name="patientregistration",
+            name="migrated_emr_patient_id",
+            field=models.BigIntegerField(blank=True, null=True),
+        ),
+        migrations.AddField(
+            model_name="assetlocation",
+            name="migrated_emr_location_id",
+            field=models.BigIntegerField(blank=True, default=None, null=True),
+        ),
+        migrations.AddField(
+            model_name="bed",
+            name="migrated_emr_bed_id",
+            field=models.BigIntegerField(blank=True, default=None, null=True),
+        ),
+        migrations.CreateModel(
+            name="MigrationTracking",
+            fields=[
+                (
+                    "id",
+                    models.BigAutoField(
+                        auto_created=True,
+                        primary_key=True,
+                        serialize=False,
+                        verbose_name="ID",
+                    ),
+                ),
+                ("old_model_obj_id", models.BigIntegerField()),
+                ("new_model_obj_id", models.BigIntegerField()),
+                ("field", models.CharField(max_length=255)),
+                ("data", models.TimeField()),
+                (
+                    "new_model",
+                    models.ForeignKey(
+                        on_delete=django.db.models.deletion.CASCADE,
+                        related_name="new_model",
+                        to="contenttypes.contenttype",
+                    ),
+                ),
+                (
+                    "old_model",
+                    models.ForeignKey(
+                        on_delete=django.db.models.deletion.CASCADE,
+                        related_name="old_model",
+                        to="contenttypes.contenttype",
+                    ),
+                ),
+            ],
+        ),
+    ]

care/facility/migrations/0478_migrate_organizations.py

@@ -0,0 +1,164 @@
+# Generated by Django 5.1.4 on 2025-01-21 08:28
+
+import logging
+from django.db import migrations
+
+logger = logging.getLogger(__name__)
+
+MIGRATION_ID = 158445695201
+COUNTRY = "India"
+
+
+def migrate_organizations(apps, schema_editor):
+    from care.emr.models import Organization
+
+    State = apps.get_model("users", "State")
+    District = apps.get_model("users", "District")
+    LocalBody = apps.get_model("users", "LocalBody")
+    Ward = apps.get_model("users", "Ward")
+
+    local_body_type_map = {
+        1: "grama_panchayath",
+        2: "block_panchayath",
+        3: "district_panchayath",
+        4: "nagar_panchayath",
+        10: "municipality",
+        20: "corporation",
+        50: "others",
+    }
+
+    logger.debug("Migrating Organization")
+    state_metadata = {
+        "country": COUNTRY,
+        "govt_org_type": "state",
+        "govt_org_children_type": "district",
+    }
+    for state_obj in State.objects.all():
+        state_org, _ = Organization.objects.filter(
+            name__iexact=state_obj.name,
+            org_type="govt",
+            parent=None,
+            metadata=state_metadata,
+        ).get_or_create(
+            defaults={
+                "name": state_obj.name,
+                "org_type": "govt",
+                "system_generated": True,
+                "metadata": state_metadata,
+                "meta": {"migration_id": MIGRATION_ID},
+            }
+        )
+        logger.debug(f"Created State: {state_org.name=}")
+        district_metadata = {
+            "country": COUNTRY,
+            "govt_org_type": "district",
+            "govt_org_children_type": "local_body",
+        }
+        for district_obj in District.objects.filter(state=state_obj):
+            district_org, created = Organization.objects.filter(
+                name__iexact=district_obj.name,
+                root_org=state_org,
+                parent=state_org,
+                org_type="govt",
+                metadata=district_metadata,
+            ).get_or_create(
+                defaults={
+                    "name": district_obj.name,
+                    "org_type": "govt",
+                    "root_org": state_org,
+                    "parent": state_org,
+                    "system_generated": True,
+                    "metadata": district_metadata,
+                    "meta": {"migration_id": MIGRATION_ID},
+                },
+            )
+            logger.debug(f"Created District: {state_org.name=}, {district_org.name=}")
+            local_body_metadata = {
+                "country": COUNTRY,
+                "govt_org_type": "local_body",
+                "govt_org_children_type": "ward",
+            }
+            for local_body_obj in LocalBody.objects.filter(district=district_obj):
+                lb_type = local_body_type_map.get(local_body_obj.body_type, "others")
+                local_body_org, created = Organization.objects.filter(
+                    name__iexact=local_body_obj.name,
+                    root_org=state_org,
+                    parent=district_org,
+                    org_type="govt",
+                    metadata={
+                        "govt_org_local_body_type": lb_type,
+                        **local_body_metadata,
+                    },
+                ).get_or_create(
+                    defaults={
+                        "name": local_body_obj.name,
+                        "org_type": "govt",
+                        "root_org": state_org,
+                        "parent": district_org,
+                        "system_generated": True,
+                        "metadata": {
+                            "govt_org_local_body_type": lb_type,
+                            **local_body_metadata,
+                        },
+                        "meta": {"migration_id": MIGRATION_ID},
+                    },
+                )
+                logger.debug(
+                    f"Created Local Body: {state_org.name=}, \
+                             {district_org.name=}, {local_body_org.name=}"
+                )
+                ward_metadata = {
+                    "country": COUNTRY,
+                    "govt_org_type": "ward",
+                }
+                for ward_obj in Ward.objects.filter(local_body=local_body_obj):
+                    ward_org, created = Organization.objects.filter(
+                        name__iexact=ward_obj.name,
+                        root_org=state_org,
+                        parent=local_body_org,
+                        org_type="govt",
+                        metadata={
+                            "govt_org_ward_number": ward_obj.number,
+                            **ward_metadata,
+                        },
+                    ).get_or_create(
+                        defaults={
+                            "name": ward_obj.name,
+                            "org_type": "govt",
+                            "root_org": state_org,
+                            "parent": local_body_org,
+                            "system_generated": True,
+                            "metadata": {
+                                "govt_org_ward_number": ward_obj.number,
+                                **ward_metadata,
+                            },
+                            "meta": {"migration_id": MIGRATION_ID},
+                        },
+                    )
+                    logger.debug(
+                        f"Created Ward: {state_org.name=}, \
+                                 {district_org.name=}, {local_body_org.name=}, \
+                                    {ward_org.name=}"
+                    )
+
+
+def reverse_migrate_organizations(apps, schema_editor):
+    logger.debug("Reversing Migration Organization")
+    schema_editor.execute(
+        "DELETE FROM emr_organization WHERE meta->>'migration_id' = %s",
+        [str(MIGRATION_ID)],
+    )
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        (
+            "facility",
+            "0477_historicalpatientregistration_migrated_emr_patient_id_and_more",
+        ),
+    ]
+
+    operations = [
+        migrations.RunPython(migrate_organizations, reverse_migrate_organizations),
+    ]