from datetime import datetime
from django.db import models
from django.urls import reverse
from accounts.models import Student


class Code(models.Model):
    student = models.ForeignKey(Student, on_delete=models.CASCADE)
    station_number = models.IntegerField()

    created = models.DateTimeField(auto_now_add=True)
    modified = models.DateTimeField(auto_now=True)

    def qr_code_str(self):
        return f'{self.student.student_number}:{self.station_number}'

    def get_absolute_url(self):
        return reverse('code-detail', kwargs={'pk': self.pk})

    def __str__(self):
        return f'{self.student.student_number}:{self.station_number}'

class Period(models.Model):
    student = models.ForeignKey(Student, on_delete=models.CASCADE)
    clocked_in = models.DateTimeField()
    clocked_out = models.DateTimeField(blank=True, null=True)
    station_number = models.IntegerField()
    duration = models.DurationField(blank=True, null=True)

    created = models.DateTimeField(auto_now_add=True)
    modified = models.DateTimeField(auto_now=True)

    class Meta:
        ordering = ['-clocked_in']

    @property
    def p_duration(self):
        return round((self.clocked_out - self.clocked_in).seconds / 3600, 2)

    def get_absolute_url(self):
        return reverse('period-detail', kwargs={'pk': self.pk})

    def __str__(self):
        return f'{self.clocked_in}: {self.student.user.first_name} {self.student.user.last_name}'

    def save(self, *args, **kwargs):
        if self.clocked_out != None:
            self.duration = self.clocked_out - self.clocked_in
        super(Period, self).save(*args, **kwargs)