pannabel/SkyLearn-Test
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix: no attribute error

Browse Source
This commit is contained in:
papi 2024-03-31 23:18:36 +03:00
parent 25180167bf
commit 9c88fe241b
1 changed files with 10 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
accounts/models.py
View File

@ -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)