(fix) Changed cookie expiration info parsing in the Kubernetes cookie…

…s assistant to tackle the suggestion in the PR review
InjectiveLabs · Feb 29, 2024 · 8cebde5 · 8cebde5
1 parent cee3bfb
commit 8cebde5
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/pyinjective/core/network.py b/pyinjective/core/network.py
@@ -86,12 +86,11 @@ def _is_cookie_expired(self, cookie_data: str) -> bool:
         cookie = SimpleCookie()
         cookie.load(cookie_data)
 
-        if "GCLB" not in cookie:
+        expiration_data: Optional[str] = cookie.get("GCLB", {}).get("expires", None)
+        if expiration_data is None:
             expiration_time = 0
         else:
-            expiration_time = datetime.datetime.strptime(
-                cookie["GCLB"]["expires"], "%a, %d-%b-%Y %H:%M:%S %Z"
-            ).timestamp()
+            expiration_time = datetime.datetime.strptime(expiration_data, "%a, %d-%b-%Y %H:%M:%S %Z").timestamp()
 
         timestamp_diff = expiration_time - time.time()
         return timestamp_diff < self.SESSION_RENEWAL_OFFSET