diff --git a/AKModel/models.py b/AKModel/models.py index 45319e302bf3d2ef0553372ab20d996d86fe6598..c6c2ffef71ea586cf6bf49895c960e1eeabb60de 100644 --- a/AKModel/models.py +++ b/AKModel/models.py @@ -1,4 +1,5 @@ # Create your models here. +import datetime import itertools from django.db import models @@ -284,3 +285,10 @@ class AKSlot(models.Model): if self.start is None: return _("Not scheduled yet") return self.start.strftime('%a %H:%M') + + @property + def end(self): + """ + Retrieve end time of the AK slot + """ + return self.start + datetime.timedelta(hours=float(self.duration))