Merge pull request #213 from praekeltfoundation/add-clinic-lookup

Add import clinics management command

Author: erikh360

momconnect/management/commands/import_clinics.py

@@ -0,0 +1,52 @@
+import csv
+from pathlib import Path
+
+from django.core.management.base import BaseCommand, CommandError
+
+from momconnect.models import Clinic
+
+
+class Command(BaseCommand):
+    help = (
+        "Import clinics from a CSV file and update or create them in the Clinic model"
+    )
+
+    def add_arguments(self, parser):
+        parser.add_argument(
+            "csv_file", type=str, help="Path to the CSV file containing clinic data"
+        )
+
+    def handle(self, *args, **options):
+        csv_file = options["csv_file"]
+        try:
+            with Path(csv_file).open(newline="", encoding="utf-8") as file:
+                reader = csv.DictReader(file)
+                for row in reader:
+                    clinic, created = Clinic.objects.update_or_create(
+                        value=row["value"],
+                        defaults={
+                            "code": row["code"],
+                            "uid": row["uid"],
+                            "name": row["name"],
+                            "province": row["province"],
+                            "location": row["location"],
+                            "area_type": row["area_type"],
+                            "unit_type": row["unit_type"],
+                            "district": row["district"],
+                            "municipality": row["municipality"],
+                        },
+                    )
+                    if created:
+                        self.stdout.write(
+                            self.style.SUCCESS(f"Created clinic: {clinic.name}")
+                        )
+                    else:
+                        self.stdout.write(
+                            self.style.SUCCESS(f"Updated clinic: {clinic.name}")
+                        )
+        except FileNotFoundError as e:
+            raise CommandError(f"File '{csv_file}' does not exist.") from e
+        except KeyError as e:
+            raise CommandError(f"Missing required column in CSV: {e}") from e
+        except Exception as e:
+            raise CommandError(f"An error occurred: {e}") from e

momconnect/tests/clinic-data/import_invalid.csv

@@ -0,0 +1,2 @@
+uid,code,name,province,location,area_type,unit_type,district
+123,123456,Clinic One,GP,Location One,Urban,Type A,District A

momconnect/tests/clinic-data/import_valid.csv

@@ -0,0 +1,3 @@
+uid,value,code,name,province,location,area_type,unit_type,district,municipality
+123,123456,123456,Clinic One,GP,Location One,Urban,Type A,District A,Municipality A
+456,123457,123457,Clinic Two,WC,Location Two,Rural,Type B,District B,Municipality B

momconnect/tests/test_import_clinics.py

@@ -0,0 +1,63 @@
+from django.core.management import CommandError, call_command
+from django.test import TestCase
+
+from momconnect.models import Clinic
+
+
+class ImportClinicsCommandTest(TestCase):
+    def get_file_path(self, name):
+        return f"./momconnect/tests/clinic-data/{name}.csv"
+
+    def test_import_clinics_success(self):
+        call_command("import_clinics", self.get_file_path("import_valid"))
+
+        self.assertEqual(Clinic.objects.count(), 2)
+        clinic1 = Clinic.objects.get(uid="123")
+        self.assertEqual(clinic1.name, "Clinic One")
+        self.assertEqual(clinic1.province, "GP")
+        self.assertEqual(clinic1.value, "123456")
+        self.assertEqual(clinic1.code, "123456")
+        self.assertEqual(clinic1.location, "Location One")
+        self.assertEqual(clinic1.area_type, "Urban")
+        self.assertEqual(clinic1.unit_type, "Type A")
+        self.assertEqual(clinic1.district, "District A")
+        self.assertEqual(clinic1.municipality, "Municipality A")
+
+        clinic2 = Clinic.objects.get(uid="456")
+        self.assertEqual(clinic2.name, "Clinic Two")
+        self.assertEqual(clinic2.province, "WC")
+        self.assertEqual(clinic2.value, "123457")
+        self.assertEqual(clinic2.code, "123457")
+        self.assertEqual(clinic2.location, "Location Two")
+        self.assertEqual(clinic2.area_type, "Rural")
+        self.assertEqual(clinic2.unit_type, "Type B")
+        self.assertEqual(clinic2.district, "District B")
+        self.assertEqual(clinic2.municipality, "Municipality B")
+
+    def test_import_clinics_update_existing(self):
+        Clinic.objects.create(
+            uid="123", value="123456", code="123456", name="Old Clinic"
+        )
+        call_command("import_clinics", self.get_file_path("import_valid"))
+
+        clinic = Clinic.objects.get(uid="123")
+        self.assertEqual(clinic.name, "Clinic One")
+        self.assertEqual(clinic.province, "GP")
+        self.assertEqual(clinic.value, "123456")
+        self.assertEqual(clinic.code, "123456")
+        self.assertEqual(clinic.location, "Location One")
+        self.assertEqual(clinic.area_type, "Urban")
+        self.assertEqual(clinic.unit_type, "Type A")
+        self.assertEqual(clinic.district, "District A")
+        self.assertEqual(clinic.municipality, "Municipality A")
+        self.assertEqual(Clinic.objects.count(), 2)
+
+    def test_import_clinics_missing_column(self):
+        with self.assertRaises(CommandError) as context:
+            call_command("import_clinics", self.get_file_path("import_invalid"))
+        self.assertIn("Missing required column", str(context.exception))
+
+    def test_import_clinics_file_not_found(self):
+        with self.assertRaises(CommandError) as context:
+            call_command("import_clinics", "nonexistent.csv")
+        self.assertIn("does not exist", str(context.exception))