diff --git a/smash/web/models/holiday.py b/smash/web/models/holiday.py index 01cceaeadf3c442b6f6bb51874c47e682aae1072..f18d5a37a6b42234fd3b8bfdc8297e729302ae9d 100644 --- a/smash/web/models/holiday.py +++ b/smash/web/models/holiday.py @@ -1,6 +1,7 @@ # coding=utf-8 from django.db import models +from constants import AVAILABILITY_CHOICES class Holiday(models.Model): class Meta: @@ -21,6 +22,8 @@ class Holiday(models.Model): verbose_name='Comments' ) + kind = models.CharField(max_length=1, choices=AVAILABILITY_CHOICES, default='H', help_text='Defines the kind of availability. Either Holiday or Extra Availability.') + def __str__(self): return "%s %s" % (self.person.first_name, self.person.last_name)