Fix: no attribute error

This commit is contained in:
papi 2024-03-31 23:18:36 +03:00
parent 25180167bf
commit 9c88fe241b

View File

@ -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)
return "{}".format(self.user)