Fix: no attribute error
This commit is contained in:
parent
25180167bf
commit
9c88fe241b
@ -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)
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user