arona/PyRIGS
1
0
Fork 0
mirror of https://github.com/nottinghamtec/PyRIGS.git synced 2026-01-20 06:52:15 +00:00
Code Issues Projects Releases Wiki Activity

mild polishing

Browse Source
This commit is contained in:
FreneticScribbler
2021-07-06 14:51:43 +01:00
parent 68a424d62b
commit e836195fef
5 changed files with 47 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
training/models.py
View File

@@ -71,8 +71,23 @@ class TrainingLevel(models.Model, RevisionMixin):
level = models.IntegerField(choices=CHOICES)
prerequisite_levels = models.ManyToManyField('self', related_name='prerequisites', symmetrical=False, blank=True)
def get_requirements_of_depth(self, depth):
return self.requirements.filter(depth=depth)
@property
def started_requirements(self):
return self.get_requirements_of_depth(TrainingItemQualification.STARTED)
@property
def complete_requirements(self):
return self.get_requirements_of_depth(TrainingItemQualification.COMPLETE)
@property
def passed_out_requirements(self):
return self.get_requirements_of_depth(TrainingItemQualification.PASSED_OUT)
def __str__(self):
if self.department is None: # TA
if self.department is None: # 2TA
return self.get_level_display()
else:
return "{} {}".format(self.get_department_display(), self.get_level_display())