From 9c88fe241b7e69d10d02957a1b909b4c6fb8f1c0 Mon Sep 17 00:00:00 2001 From: papi Date: Sun, 31 Mar 2024 23:18:36 +0300 Subject: [PATCH] Fix: no attribute error --- accounts/models.py | 23 ++++++++++------------- 1 file changed, 10 insertions(+), 13 deletions(-) diff --git a/accounts/models.py b/accounts/models.py index c4196e8..90d9496 100644 --- a/accounts/models.py +++ b/accounts/models.py @@ -54,6 +54,15 @@ class CustomUserManager(UserManager): ).distinct() # distinct() is often necessary with Q lookups return queryset + def get_student_count(self): + return self.model.objects.filter(is_student=True).count() + + def get_lecturer_count(self): + return self.model.objects.filter(is_lecturer=True).count() + + def get_superuser_count(self): + return self.model.objects.filter(is_superuser=True).count() + GENDERS = (("M", "Male"), ("F", "Female")) @@ -85,18 +94,6 @@ 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) @@ -212,4 +209,4 @@ class DepartmentHead(models.Model): ordering = ("-user__date_joined",) def __str__(self): - return "{}".format(self.user) \ No newline at end of file + return "{}".format(self.user)