from django.db import models
from django.conf import settings
import reversion

# Create your models here.


@reversion.register
class TrainingCategory(models.Model):
    category_number = models.PositiveSmallIntegerField()
    category_name = models.CharField(max_length=50)


@reversion.register
class TrainingItem(models.Model):
    category = models.ForeignKey(TrainingCategory)
    item_number = models.PositiveSmallIntegerField()
    item_name = models.CharField(max_length=100)
    training_records = models.ManyToManyField(settings.AUTH_USER_MODEL,
                                              through='TrainingRecord', through_fields=('training_item', 'trainee'))


@reversion.register
class TrainingRecord(models.Model):
    trainee = models.ForeignKey(settings.AUTH_USER_MODEL)
    training_item = models.ForeignKey(TrainingItem)

    started_date = models.DateField(blank=True, null=True)
    started_trainer = models.ForeignKey(settings.AUTH_USER_MODEL, related_name='trainingrecords_started')
    started_notes = models.TextField(blank=True, null=True)
    completed_date = models.DateField(blank=True, null=True)
    completed_trainer = models.ForeignKey(settings.AUTH_USER_MODEL, related_name='trainingrecords_completed')
    completed_notes = models.TextField(blank=True, null=True)
    assessed_date = models.DateField(blank=True, null=True)
    assessed_trainer = models.ForeignKey(settings.AUTH_USER_MODEL, related_name='trainingrecords_assessed')
    assessed_notes = models.TextField(blank=True, null=True)

    class Meta:
        unique_together = ('trainee', 'training_item')


@reversion.register
class TrainingLevelRecord(models.Model):
    trainee = models.ForeignKey(settings.AUTH_USER_MODEL)

    technical_assistant = models.DateField(blank=True, null=True)

    sound_technician = models.DateField(blank=True, null=True)
    sound_supervisor = models.DateField(blank=True, null=True)

    lighting_technician = models.DateField(blank=True, null=True)
    lighting_supervisor = models.DateField(blank=True, null=True)

    power_technician = models.DateField(blank=True, null=True)
    power_supervisor = models.DateField(blank=True, null=True)

    haulage_supervisor = models.DateField(blank=True, null=True)