Return timezone with shifts

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

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,