diff --git a/panera_shifts/__init__.py b/panera_shifts/__init__.py index ee62e1c..04f398a 100644 --- a/panera_shifts/__init__.py +++ b/panera_shifts/__init__.py @@ -40,8 +40,8 @@ def get_shifts_between(session, start_date: datetime.date, end_date: datetime.da timezone_info = get_timezone(session, bearer_token, cafe_id) logger.debug(f"Timezone: {timezone_info}") - ret = get_shifts(session, bearer_token, employee_id, start_date, end_date) - ret["data"]["timezone"] = timezone_info["timezone"] + ret: dict = get_shifts(session, bearer_token, employee_id, start_date, end_date) + ret.setdefault("data", dict())["timezone"] = timezone_info["timezone"] logger.info("Ended get_shifts_between") logger.debug(pprint.pformat(ret)) return ret