- todays_date = str(datetime.now())[:10]
- target_start_str = start_date_str if start_date_str else sorted(self.days.keys())[0]
- target_start = todays_date if target_start_str == 'today' else target_start_str
- target_end_str = end_date_str if end_date_str else sorted(self.days.keys())[-1]
- target_end = todays_date if target_end_str == 'today' else target_end_str
- start_date = datetime.strptime(target_start, DATE_FORMAT)
- end_date = datetime.strptime(target_end, DATE_FORMAT)
- for n in range(int((end_date - start_date).days) + 1):
- current_date_obj = start_date + timedelta(n)
- current_date = current_date_obj.strftime(DATE_FORMAT)
- if current_date not in self.days.keys():
- days_to_show[current_date] = self.add_day()
+ todays_date_str = str(datetime.now())[:10]
+ todays_date_obj = datetime.strptime(todays_date_str, DATE_FORMAT)
+ yesterdays_date_obj = todays_date_obj - timedelta(1)
+ yesterdays_date_str = yesterdays_date_obj.strftime(DATE_FORMAT)
+ start_date_obj = datetime.strptime(sorted(self.days.keys())[0], DATE_FORMAT)
+ if start_date_str and len(start_date_str) > 0:
+ if start_date_str in {'today', 'yesterday'}:
+ start_date_obj = todays_date_obj if start_date_str == 'today' else yesterdays_date_obj
+ else:
+ start_date_obj = datetime.strptime(start_date_str, DATE_FORMAT)
+ end_date_obj = datetime.strptime(sorted(self.days.keys())[-1], DATE_FORMAT)
+ if end_date_str and len(end_date_str) > 0:
+ if end_date_str in {'today', 'yesterday'}:
+ end_date_obj = todays_date_obj if end_date_str == 'today' else yesterdays_date_obj
+ else:
+ end_date_obj = datetime.strptime(start_date_str, DATE_FORMAT)
+ for n in range(int((end_date_obj - start_date_obj).days) + 1):
+ current_date_obj = start_date_obj + timedelta(n)
+ current_date_str = current_date_obj.strftime(DATE_FORMAT)
+ if current_date_str not in self.days.keys():
+ days_to_show[current_date_str] = self.add_day()