AD/main/models.py

from django.db import models
from django.utils.translation import gettext_lazy as _
# Create your models here.


class Employee(models.Model):
    name = models.CharField(max_length=100)
    daily_workhours = models.PositiveIntegerField()
    info = models.TextField(blank=True, null=True)

    def __str__(self):
        return self.name


class Shift(models.Model):
    class ShiftType(models.TextChoices):
        NONE = 'N', _('None')  # Placeholder for "None" if you're using it
        VACATION = 'U', _('Urlaub')
        CATERING = 'C', _('Catering')
        SICK = 'K', _('Krank')
        RB = 'RB', _('Rufbereitschaft')
        SD = 'SO', _('Sonderdienst')
        KK = 'KKH', _('KKH')
        ST = 'ST', _('Stapler')
        AH = 'AH', _('Ausser Haus')

    date = models.DateField()
    start = models.TimeField(null=True, blank=True)
    end = models.TimeField(null=True, blank=True)
    shifttype = models.CharField(
        max_length=3,
        choices=ShiftType.choices,
        default=ShiftType.NONE,
        null=True,
    )
    employee = models.ForeignKey(Employee, on_delete=models.CASCADE)

    def __str__(self):
        return f"{self.date} - {self.get_shifttype_display()}"