Browse Source

Fix KeyError

master
Raphael Roberts 4 years ago
parent
commit
09203ab336
  1. 4
      panera_shifts/__init__.py

4
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
Loading…
Cancel
Save