Soft delete assets from deleted facilities/hospitals

care/facility/models/facility.py

@@ -2,7 +2,7 @@
 from django.contrib.auth import get_user_model
 from django.contrib.postgres.fields import ArrayField
 from django.core.validators import MinValueValidator
-from django.db import models
+from django.db import models, transaction
 from django.db.models import IntegerChoices
 from django.db.models.constraints import CheckConstraint, UniqueConstraint
 from django.utils.translation import gettext_lazy as _
@@ -90,27 +90,27 @@
     (5, "Hotel"),
     (6, "Lodge"),
     (7, "TeleMedicine"),
     # (8, "Govt Hospital"), # Change from "Govt Hospital" to "Govt Medical College Hospitals"
     (9, "Govt Labs"),
     (10, "Private Labs"),
     # Use 8xx for Govt owned hospitals and health centres
     (800, "Primary Health Centres"),
     # (801, "24x7 Public Health Centres"), # Change from "24x7 Public Health Centres" to "Primary Health Centres"
     (802, "Family Health Centres"),
     (803, "Community Health Centres"),
     # (820, "Urban Primary Health Center"),   # Change from "Urban Primary Health Center" to "Primary Health Centres"
     (830, "Taluk Hospitals"),
     # (831, "Taluk Headquarters Hospitals"),     # Change from "Taluk Headquarters Hospitals" to "Taluk Hospitals"
     (840, "Women and Child Health Centres"),
     # (850, "General hospitals"),  # Change from "General hospitals" to "District Hospitals"
     (860, "District Hospitals"),
     (870, "Govt Medical College Hospitals"),
     (900, "Co-operative hospitals"),
     (910, "Autonomous healthcare facility"),
     # Use 9xx for Labs
     # (950, "Corona Testing Labs"),    # Change from "Corona Testing Labs" to "Govt Labs"
     # Use 10xx for Corona Care Center
     # (1000, "Corona Care Centre"),   # Change from "Corona Care Centre" to "Other"
     (1010, "COVID-19 Domiciliary Care Center"),
     # Use 11xx for First Line Treatment Centre
     (1100, "First Line Treatment Centre"),
@@ -207,7 +207,7 @@
         "hub_id", flat=True
     )
     if spoke_id in ancestors_of_base:
         raise serializers.ValidationError("This facility is already an ancestor hub")
     for ancestor in ancestors_of_base:
         check_if_spoke_is_not_ancestor(ancestor, spoke_id)
 
@@ -299,6 +299,18 @@
                 facility=self, user=self.created_by, created_by=self.created_by
             )
 
+    @transaction.atomic
+    def delete(self, *args):
+        from care.facility.models.asset import Asset, AssetLocation
+
+        AssetLocation.objects.filter(facility_id=self.id).update(deleted=True)
+        Asset.objects.filter(
+            current_location_id__in=AssetLocation._base_manager.filter(  # noqa: SLF001
+                facility_id=self.id
+            ).values_list("id", flat=True)
+        ).update(deleted=True)
+        return super().delete(*args)
+
     @property
     def get_features_display(self):
         if not self.features:
@@ -308,19 +320,19 @@
     def get_facility_flags(self):
         return FacilityFlag.get_all_flags(self.id)
 
     CSV_MAPPING = {
         "name": "Facility Name",
         "facility_type": "Facility Type",
         "address": "Address",
         "ward__name": "Ward Name",
         "ward__number": "Ward Number",
         "local_body__name": "Local Body",
         "district__name": "District",
         "state__name": "State",
         "phone_number": "Phone Number",
     }
 
     CSV_MAKE_PRETTY = {"facility_type": (lambda x: REVERSE_FACILITY_TYPES[x])}
 
 
 class FacilityHubSpoke(BaseModel, FacilityRelatedPermissionMixin):