wip: implementing EngagementStrategy

1 files changed, 16 insertions, 6 deletions

diff --git a/engagements/models.py b/engagements/models.py
index ef0aee3..8876d91 100644
--- a/engagements/models.py
+++ b/engagements/models.py
@@ -255,9 +255,19 @@ class RegulatoryCycle(Common):
         return f"Regulatory Cycle: {self.get_year()}"
 
 
-# class EngagementStrategy(Common):
-#     name = models.CharField(max_length=128, null=False, blank=False)
-#     start_date = models.DateField(null=False, blank=False)
-#     end_date
-#     description = models.TextField(max_length=1024)
-#     engagement = models.ForeignKey(Engagement, on_delete=models.CASCADE)
+class EngagementStrategy(Common):
+    STATUS = (
+        ("DRAFT", "Draft"),
+        ("SUBMITTED", "Submitted"),
+        ("APPROVED", "Approved"),
+        ("REJECTED", "Rejected"),
+    )
+
+    name = models.CharField(max_length=128, null=False, blank=False)
+    organisation = models.ForeignKey(Organisation, on_delete=models.CASCADE)
+    start_year = models.ForeignKey(RegulatoryCycle, on_delete=models.CASCADE, related_name="start_year")
+    end_year = models.ForeignKey(RegulatoryCycle, on_delete=models.CASCADE, related_name="end_year")
+    description = models.TextField(max_length=1024)
+    inspector_sign_off = models.DateField(null=True, blank=True)
+    management_sign_off = models.DateField(null=True, blank=True)
+    status = models.CharField(max_length=32, choices=STATUS, default=STATUS[0][0])