2021-01-30 17:40:40 -07:00

47 lines
1.6 KiB
Python

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)
# def duration(self):
# duration = self.clocked_out - self.clocked_in
# duration_in_s = duration.total_seconds()
# return divmod(duration_in_s, 3600)[0]
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 isinstance(self.clocked_out, datetime):
self.duration = self.clocked_out - self.clocked_in
super(Period, self).save(*args, **kwargs)