from django.db import models
from django.contrib.auth.models import User

SHIFT_CHOICES = (
    ('A', '7:00-15:00'),
    ('P', '15:00-23:00'),
    ('N', '23:00-7:00'),
)

class Employee(models.Model):
    user = models.OneToOneField(User, related_name='employeemodel', on_delete=models.CASCADE, blank=True, null=True)
    shift = models.CharField(max_length=11, choices=SHIFT_CHOICES, default='A')