minor changes

nbdaaron · Jun 29, 2023 · 98bc47c · 98bc47c
1 parent 6954a1b
commit 98bc47c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/poe-api/src/poe.py b/poe-api/src/poe.py
@@ -195,7 +195,7 @@ def get_next_data(self, overwrite_vars=False):
 
     if overwrite_vars:
       self.formkey = self.extract_formkey(r.text)
-      if "payload" in next_data["props"]["pageProps"].keys():
+      if "payload" in next_data["props"]["pageProps"]:
         self.viewer = next_data["props"]["pageProps"]["payload"]["viewer"]
       else:
         self.viewer = next_data["props"]["pageProps"]["data"]["viewer"]
@@ -207,11 +207,11 @@ def get_next_data(self, overwrite_vars=False):
   def get_bot(self, display_name):
     url = f'https://poe.com/_next/data/{self.next_data["buildId"]}/{display_name}.json'
 
-    r = request_with_retries(self.session.get, url).json()
-    if "payload" in r["pageProps"]:
-      chat_data = r["pageProps"]["payload"]["chatOfBotDisplayName"]
+    data = request_with_retries(self.session.get, url).json()
+    if "payload" in data["pageProps"]:
+      chat_data = data["pageProps"]["payload"]["chatOfBotDisplayName"]
     else:
-      chat_data = r["pageProps"]["data"]["chatOfBotDisplayName"]
+      chat_data = data["pageProps"]["data"]["chatOfBotDisplayName"]
     return chat_data
 
   def get_bots(self, download_next_data=True):

diff --git a/setup.py b/setup.py
@@ -9,7 +9,7 @@
 
 setuptools.setup(
   name="poe-api",
-  version="0.4.6",
+  version="0.4.7",
   author="ading2210",
   license="GPLv3",
   description="A reverse engineered API wrapper for Quora's Poe",