From 0cbdf3ed4708c937679a7c8a9313b2eaafa11086 Mon Sep 17 00:00:00 2001
From: Zaki Benaissa <zakben45@gmail.com>
Date: Tue, 9 Jan 2024 19:22:13 +0100
Subject: [PATCH] Add count helper methods to User model

---
 accounts/models.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/accounts/models.py b/accounts/models.py
index 1baa3fd..2b92825 100644
--- a/accounts/models.py
+++ b/accounts/models.py
@@ -81,6 +81,18 @@ class User(AbstractUser):
             full_name = self.first_name + " " + self.last_name
         return full_name
 
+    @classmethod
+    def get_student_count(cls):
+        return cls.objects.filter(is_student=True).count()
+
+    @classmethod
+    def get_lecturer_count(cls):
+        return cls.objects.filter(is_lecturer=True).count()
+
+    @classmethod
+    def get_superuser_count(cls):
+        return cls.objects.filter(is_superuser=True).count()
+
     def __str__(self):
         return "{} ({})".format(self.username, self.get_full_name)