layout of CAF list

1 files changed, 4 insertions, 1 deletions

diff --git a/ctrack/caf/models.py b/ctrack/caf/models.py
index 5d7b5b0..0151d4b 100644
--- a/ctrack/caf/models.py
+++ b/ctrack/caf/models.py
@@ -92,7 +92,10 @@ class CAF(models.Model):
     def organisation(self):
         first_ass = ApplicableSystem.objects.filter(caf=self).first()
         return first_ass.organisation
-    
+
+    def sub_mode(self):
+        return self.organisation().submode
+
     def __str__(self):
         # Get the organisation and applicable system
         ass = ApplicableSystem.objects.filter(caf=self).first()