def where(self, treatment):
dh = DateHelper()
now = dh.now()
if treatment is None:
self.checked_in = False
self.location = ""
return
elif now < treatment.start + treatment.duration:
self.checked_in = True
self.location = "treatment"
return
elif now < treatment.start + treatment.duration + treatment.recovery:
self.checked_in = True
self.location = "recovery"
return
else:
self.checked_in = False
self.location = ""
return
def start_treatment(self, time=None):
if time is None:
dh = DateHelper()
self.start = dh.now()
else:
self.start = time