diff --git a/panera_shifts/__init__.py b/panera_shifts/__init__.py index 5ca66c3..6e4a9fe 100644 --- a/panera_shifts/__init__.py +++ b/panera_shifts/__init__.py @@ -8,6 +8,7 @@ from panera_shifts.oauth_required import ( get_cafe_id, get_iBOH_id, get_shifts, + get_timezone, ) @@ -21,4 +22,7 @@ def get_shifts_between(session, start_date: datetime.date, end_date: datetime.da bearer_token = get_bearer_token(session) cafe_id = get_cafe_id(session) iBOH_id = get_iBOH_id(session, bearer_token) - return get_shifts(session, bearer_token, iBOH_id, cafe_id, start_date, end_date) + timezone_info = get_timezone(session, bearer_token, cafe_id) + ret = get_shifts(session, bearer_token, iBOH_id, cafe_id, start_date, end_date) + ret["data"]["timezone"] = timezone_info["timezone"] + return ret diff --git a/panera_shifts/oauth_required.py b/panera_shifts/oauth_required.py index ab1dd5a..7337352 100644 --- a/panera_shifts/oauth_required.py +++ b/panera_shifts/oauth_required.py @@ -25,6 +25,17 @@ def get_iBOH_id(session, bearer_token): return data["ibohid"] +def get_timezone(session, bearer_token, cafe_id): + headers = {"authorization": f"Bearer {bearer_token}"} + base_url = ( + "https://pantry.panerabread.com/apis/selfservice-ui-service/v1/migration/check" + ) + r = session.get( + base_url, params={"cafeId": cafe_id, "companyCode": "PANERA"}, headers=headers + ) + return json.loads(r.text) + + def get_shifts( session, bearer_token,