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