Swaped from CIS to NIST controls

backend/core/models.py

@@ -159,11 +159,17 @@ class Risk(models.Model):
 
 class Control(models.Model):
     id = models.AutoField(primary_key=True)
-    safeguard_id = models.CharField(max_length=10, unique=True, null=True, help_text="Unique identifier for the safeguard (e.g. '1.1', '4.6')")
-    name = models.CharField(max_length=255)
-    description = models.TextField(default=" ", help_text="Description of the control")
+
+    subcategory = models.CharField(max_length=64, unique=True, null=False, blank=False, help_text='NIST Subcategory')
+    function = models.TextField(null=True, blank=True, help_text="Function")
+    category = models.TextField(null=True, blank=True,help_text="Category")
+    implementation_examples = models.TextField(null=True, blank=True, help_text="Implementation Examples")
+    effectiveness_monitoring_examples = models.TextField(null=True, blank=True, help_text="Effectiveness Monitoring Examples")
+    documentation_score = models.IntegerField(null=True, blank=True, help_text="Documentation Score")
+    implementation_score = models.IntegerField(null=True, blank=True, help_text="Implementation Score")
+
     def __str__(self):
-        return f"{self.id} ({self.name})"
+        return f"{self.id} ({self.subcategory} - {self.function or self.category or ''})".rstrip(" -() ")
     
 class DocumentRiskControl(models.Model):
     document = models.ForeignKey(Document, on_delete=models.CASCADE)