ramon
/
skillbox
1
0
Fork 0
Code Issues 1 Pull Requests 1 Packages Projects 1 Releases 1 Wiki Activity

Add helper functions to find the teacher of a user

This commit is contained in:
Ramon Wenger 2019-09-09 14:01:58 +02:00
parent 371c5cd276
commit 9e4784e5d8
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
server/users/models.py
View File

@ -37,6 +37,14 @@ class User(AbstractUser):
def users_in_same_school_class(self): def users_in_same_school_class(self):
return User.objects.filter(school_classes__users=self.pk) return User.objects.filter(school_classes__users=self.pk)
def get_teacher(self):
if self.user_roles.filter(role__key='teacher').exists():
return self
elif self.school_classes.count()>0:
return self.school_classes.first().get_teacher()
else:
return None
def selected_class(self): def selected_class(self):
try: try:
settings = UserSetting.objects.get(user=self) settings = UserSetting.objects.get(user=self)
@ -65,6 +73,9 @@ class SchoolClass(models.Model):
def is_user_in_schoolclass(self, user): def is_user_in_schoolclass(self, user):
return user.is_superuser or user.school_classes.filter(pk=self.id).count() > 0 return user.is_superuser or user.school_classes.filter(pk=self.id).count() > 0
def get_teacher(self):
return self.users.filter(user_roles__role__key='teacher').first()
class Role(models.Model): class Role(models.Model):
key = models.CharField(_('Key'), max_length=100, blank=False, null=False, unique=True) key = models.CharField(_('Key'), max_length=100, blank=False, null=False, unique=True)