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

# Create your models here.
class Workday(models.Model):
	user = models.ForeignKey(User, on_delete=models.CASCADE)
	agency = models.ForeignKey(Agency, on_delete=models.CASCADE)
	breaks = models.ManyToManyField("Breaks", blank=True, related_name='breaks_at_day')
	start = models.DateTimeField(default=None, null=True, blank=True)
	end = models.DateTimeField(default=None, null=True, blank=True)

class Breaks(models.Model):
	user = models.ForeignKey(User, on_delete=models.CASCADE)
	agency = models.ForeignKey(Agency, on_delete=models.CASCADE)
	workday = models.ForeignKey("Workday", on_delete=models.CASCADE, related_name='workdayele')
	start = models.DateTimeField(default=None, null=True, blank=True)
	end = models.DateTimeField(default=None, null=True, blank=True)