Fix KeyError: 'expires_in'

TeamUltroid · Jan 1, 2024 · bed02e4 · bed02e4
1 parent c643564
commit bed02e4
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/bot/onedrive.py b/bot/onedrive.py
@@ -95,8 +95,9 @@ async def get_access_token(self, code: str = None):
             headers={"Content-Type": "application/x-www-form-urlencoded"},
             data=data,
         ) as resp:
+            expiry_time = self.creds["expires_in"]
             self.creds = await resp.json()
-            self.creds["expires_at"] = time.time() + self.creds["expires_in"]
+            self.creds["expires_at"] = time.time() + expiry_time
             udB.set_key("OD_AUTH_TOKEN", self.creds)
             await resp.release()
             return self.creds
@@ -114,8 +115,9 @@ async def refresh_access_token(self):
             headers={"Content-Type": "application/x-www-form-urlencoded"},
             data=data,
         ) as resp:
+            expiry_time = self.creds["expires_in"]
             self.creds = await resp.json()
-            self.creds["expires_at"] = time.time() + self.creds["expires_in"]
+            self.creds["expires_at"] = time.time() + expiry_time
             await resp.release()
             udB.set_key("OD_AUTH_TOKEN", self.creds)
             return