From 09203ab336a2e5687dfaba76ef67ea8b30b3b511 Mon Sep 17 00:00:00 2001 From: Raphael Roberts Date: Mon, 26 Jul 2021 21:34:39 -0500 Subject: [PATCH] Fix KeyError --- panera_shifts/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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