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
|
).distinct() # distinct() is often necessary with Q lookups
|
||||||
return queryset
|
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"))
|
GENDERS = (("M", "Male"), ("F", "Female"))
|
||||||
|
|
||||||
@ -85,18 +94,6 @@ class User(AbstractUser):
|
|||||||
full_name = self.first_name + " " + self.last_name
|
full_name = self.first_name + " " + self.last_name
|
||||||
return full_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):
|
def __str__(self):
|
||||||
return "{} ({})".format(self.username, self.get_full_name)
|
return "{} ({})".format(self.username, self.get_full_name)
|
||||||
|
|
||||||
@ -212,4 +209,4 @@ class DepartmentHead(models.Model):
|
|||||||
ordering = ("-user__date_joined",)
|
ordering = ("-user__date_joined",)
|
||||||
|
|
||||||
def __str__(self):
|
def __str__(self):
|
||||||
return "{}".format(self.user)
|
return "{}".format(self.user)
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user