Manage Policy model save

intuitem · Feb 14, 2024 · 2e927d8 · 2e927d8
1 parent 4a053dc
commit 2e927d8
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/backend/core/models.py b/backend/core/models.py
@@ -836,6 +836,10 @@ class Meta:
 
     objects = SecurityMeasure.objects.filter(category="policy")
 
+    def save(self, *args, **kwargs):
+        self.category = "policy"
+        super(Policy, self).save(*args, **kwargs)
+
 
 class RiskScenario(AbstractBaseModel, NameDescriptionMixin):
     TREATMENT_OPTIONS = [