1 files changed, 7 insertions, 2 deletions
diff --git a/ctrack/users/models.py b/ctrack/users/models.py
index 052efd6..93c086a 100644
--- a/ctrack/users/models.py
+++ b/ctrack/users/models.py
@@ -3,11 +3,10 @@ from django.db import models
 from django.urls import reverse
 from django.utils.translation import ugettext_lazy as _
 
-from ctrack.organisations.models import Stakeholder
+from ctrack.organisations.models import Stakeholder, Submode, Organisation
 
 
 class User(AbstractUser):
-
     name = models.CharField(_("Name of User"), blank=True, max_length=255)
     stakeholder = models.OneToOneField(
         Stakeholder, on_delete=models.CASCADE, null=True, blank=True
@@ -26,3 +25,9 @@ class User(AbstractUser):
     def get_organisation_name(self):
         if self.is_stakeholder:
             return self.stakeholder.person.organisation.name
+
+    def get_lead_inspector_organisations(self):
+        return Organisation.objects.filter(lead_inspector=self)
+
+    def get_deputy_lead_inspector_organisations(self):
+        return Organisation.objects.filter(deputy_lead_inspector=self)