Files
PyRIGS/training/models.py
2021-06-29 17:13:36 +01:00

62 lines
2.0 KiB
Python

from django.db import models
from RIGS.models import RevisionMixin, Profile
from reversion import revisions as reversion
# 'shim' overtop the profile model to neatly contain all training related fields etc
@reversion.register
class Trainee(Profile):
pass
# Items
class TrainingCategory(models.Model):
number = models.CharField(max_length=3) # Does this 1:1 correspond with a department? I think the answer is sometimes...
name = models.CharField(max_length=50)
class TrainingItem(models.Model):
category = models.ForeignKey('TrainingCategory', on_delete=models.CASCADE)
number = models.CharField(max_length=3)
name = models.CharField(max_length=50)
class TrainingItemInstance(models.Model):
item = models.ForeignKey('TrainingItem', on_delete=models.CASCADE)
trainee = models.ForeignKey('Trainee', related_name='items', on_delete=models.CASCADE)
training_started_on = models.DateField()
training_started_by = models.ForeignKey('Trainee', related_name='training_started', on_delete=models.CASCADE)
training_complete_on = models.DateField()
training_complete_by = models.ForeignKey('Trainee', related_name='training_complete', on_delete=models.CASCADE)
passed_out_on = models.DateField()
passed_out_by = models.ForeignKey('Trainee', related_name='passed_out', on_delete=models.CASCADE)
class Department(models.Model):
name = models.CharField(max_length=50)
# Levels
class TrainingLevel(models.Model, RevisionMixin):
requirements = models.ManyToManyField(TrainingItem)
class Meta:
abstract = True
@reversion.register
class TechnicalAssistant(TrainingLevel):
# department = models.ForeignKey('Department', on_delete=models.CASCADE)
pass
@reversion.register
class Technician(TrainingLevel):
department = models.ForeignKey('Department', on_delete=models.CASCADE)
@reversion.register
class Supervisor(TrainingLevel):
department = models.ForeignKey('Department', on_delete=models.CASCADE)