2 files changed, 28 insertions, 0 deletions
diff --git a/ctrack/caf/managers.py b/ctrack/caf/managers.py
new file mode 100644
index 0000000..3aea117
--- /dev/null
+++ b/ctrack/caf/managers.py
@@ -0,0 +1,25 @@
+from django.db import models
+
+from ctrack.organisations.models import Organisation, Person
+
+
+class ApplicableSystemManager(models.Manager):
+    def with_primary_contact(self):
+        """
+        Add in the name of the primary nis contact to the context.
+        Using Custom Managers Django docs for an example.
+        """
+        from django.db import connection
+        with connection.cursor() as cursor:
+            cursor.execute("""
+                SELECT a.id, a.name, o.id, c.id, sm.descriptor, p.id, o.name
+                FROM caf_applicablesystem a, organisations_organisation o, organisations_person p, caf_caf c, organisations_submode sm
+                WHERE a.organisation_id = o.id AND a.caf_id = c.id AND p.organisation_id = o.id AND o.submode_id = sm.id AND p.primary_nis_contact = False;
+            """)
+            result_list = []
+            for row in cursor.fetchall():
+                org = Organisation.objects.get(pk=row[2])
+                ass = self.model(id=row[0], name=row[1], organisation=org)
+                ass.nis_contact = Person.objects.get(pk=row[5])
+                result_list.append(ass)
+            return result_list
diff --git a/ctrack/caf/models.py b/ctrack/caf/models.py
index 2069612..a031657 100644
--- a/ctrack/caf/models.py
+++ b/ctrack/caf/models.py
@@ -1,5 +1,6 @@
 from django.db import models
 
+from ctrack.caf.managers import ApplicableSystemManager
 from ctrack.organisations.models import Organisation, Person
 
 
@@ -55,6 +56,8 @@ class ApplicableSystem(models.Model):
     class Meta:
         verbose_name = "Applicable System"
 
+    objects = ApplicableSystemManager()
+
     def __str__(self):
         return f"{self.organisation.name} | {self.name}"