summaryrefslogtreecommitdiffstats
path: root/quora/api.py
diff options
context:
space:
mode:
authort.me/xtekky <98614666+xtekky@users.noreply.github.com>2023-04-27 14:29:33 +0200
committerGitHub <noreply@github.com>2023-04-27 14:29:33 +0200
commitd8ec09cebbe04268200cb2f53289525cb95d00bb (patch)
tree997609545bc922d63081a49ee356b747b4ecce68 /quora/api.py
parentMerge branch 'main' of https://github.com/Andrew-Tsegaye/gpt4free (diff)
parentMerge pull request #197 from AymaneHrouch/update_syspath (diff)
downloadgpt4free-d8ec09cebbe04268200cb2f53289525cb95d00bb.tar
gpt4free-d8ec09cebbe04268200cb2f53289525cb95d00bb.tar.gz
gpt4free-d8ec09cebbe04268200cb2f53289525cb95d00bb.tar.bz2
gpt4free-d8ec09cebbe04268200cb2f53289525cb95d00bb.tar.lz
gpt4free-d8ec09cebbe04268200cb2f53289525cb95d00bb.tar.xz
gpt4free-d8ec09cebbe04268200cb2f53289525cb95d00bb.tar.zst
gpt4free-d8ec09cebbe04268200cb2f53289525cb95d00bb.zip
Diffstat (limited to 'quora/api.py')
-rw-r--r--quora/api.py77
1 files changed, 22 insertions, 55 deletions
diff --git a/quora/api.py b/quora/api.py
index b28c124b..42814f2c 100644
--- a/quora/api.py
+++ b/quora/api.py
@@ -18,23 +18,21 @@
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.
-import requests
-import re
+import hashlib
import json
-import random
import logging
-import time
import queue
+import random
+import re
import threading
+import time
import traceback
-import hashlib
-import string
-import random
-import requests.adapters
-import websocket
from pathlib import Path
from urllib.parse import urlparse
+import requests
+import requests.adapters
+import websocket
parent_path = Path(__file__).resolve().parent
queries_path = parent_path / "graphql"
@@ -66,7 +64,7 @@ def request_with_retries(method, *args, **kwargs):
if r.status_code == 200:
return r
logger.warn(
- f"Server returned a status code of {r.status_code} while downloading {url}. Retrying ({i+1}/{attempts})..."
+ f"Server returned a status code of {r.status_code} while downloading {url}. Retrying ({i + 1}/{attempts})..."
)
raise RuntimeError(f"Failed to download {url} too many times.")
@@ -81,9 +79,7 @@ class Client:
def __init__(self, token, proxy=None):
self.proxy = proxy
self.session = requests.Session()
- self.adapter = requests.adapters.HTTPAdapter(
- pool_connections=100, pool_maxsize=100
- )
+ self.adapter = requests.adapters.HTTPAdapter(pool_connections=100, pool_maxsize=100)
self.session.mount("http://", self.adapter)
self.session.mount("https://", self.adapter)
@@ -139,9 +135,7 @@ class Client:
logger.info("Downloading next_data...")
r = request_with_retries(self.session.get, self.home_url)
- json_regex = (
- r'<script id="__NEXT_DATA__" type="application\/json">(.+?)</script>'
- )
+ json_regex = r'<script id="__NEXT_DATA__" type="application\/json">(.+?)</script>'
json_text = re.search(json_regex, r.text).group(1)
next_data = json.loads(json_text)
@@ -213,19 +207,14 @@ class Client:
if channel is None:
channel = self.channel
query = f'?min_seq={channel["minSeq"]}&channel={channel["channel"]}&hash={channel["channelHash"]}'
- return (
- f'wss://{self.ws_domain}.tch.{channel["baseHost"]}/up/{channel["boxName"]}/updates'
- + query
- )
+ return f'wss://{self.ws_domain}.tch.{channel["baseHost"]}/up/{channel["boxName"]}/updates' + query
def send_query(self, query_name, variables):
for i in range(20):
json_data = generate_payload(query_name, variables)
payload = json.dumps(json_data, separators=(",", ":"))
- base_string = (
- payload + self.gql_headers["poe-formkey"] + "WpuLMiXEKKE98j56k"
- )
+ base_string = payload + self.gql_headers["poe-formkey"] + "WpuLMiXEKKE98j56k"
headers = {
"content-type": "application/json",
@@ -233,15 +222,11 @@ class Client:
}
headers = {**self.gql_headers, **headers}
- r = request_with_retries(
- self.session.post, self.gql_url, data=payload, headers=headers
- )
+ r = request_with_retries(self.session.post, self.gql_url, data=payload, headers=headers)
data = r.json()
if data["data"] == None:
- logger.warn(
- f'{query_name} returned an error: {data["errors"][0]["message"]} | Retrying ({i+1}/20)'
- )
+ logger.warn(f'{query_name} returned an error: {data["errors"][0]["message"]} | Retrying ({i + 1}/20)')
time.sleep(2)
continue
@@ -304,9 +289,7 @@ class Client:
def on_ws_close(self, ws, close_status_code, close_message):
self.ws_connected = False
- logger.warn(
- f"Websocket closed with status {close_status_code}: {close_message}"
- )
+ logger.warn(f"Websocket closed with status {close_status_code}: {close_message}")
def on_ws_error(self, ws, error):
self.disconnect_ws()
@@ -333,11 +316,7 @@ class Client:
return
# indicate that the response id is tied to the human message id
- elif (
- key != "pending"
- and value == None
- and message["state"] != "complete"
- ):
+ elif key != "pending" and value == None and message["state"] != "complete":
self.active_messages[key] = message["messageId"]
self.message_queues[key].put(message)
return
@@ -381,9 +360,7 @@ class Client:
human_message = message_data["data"]["messageEdgeCreate"]["message"]
human_message_id = human_message["node"]["messageId"]
except TypeError:
- raise RuntimeError(
- f"An unknown error occurred. Raw response data: {message_data}"
- )
+ raise RuntimeError(f"An unknown error occurred. Raw response data: {message_data}")
# indicate that the current message is waiting for a response
self.active_messages[human_message_id] = None
@@ -418,9 +395,7 @@ class Client:
def send_chat_break(self, chatbot):
logger.info(f"Sending chat break to {chatbot}")
- result = self.send_query(
- "AddMessageBreakMutation", {"chatId": self.bots[chatbot]["chatId"]}
- )
+ result = self.send_query("AddMessageBreakMutation", {"chatId": self.bots[chatbot]["chatId"]})
return result["data"]["messageBreakCreate"]["message"]
def get_message_history(self, chatbot, count=25, cursor=None):
@@ -437,15 +412,11 @@ class Client:
cursor = str(cursor)
if count > 50:
- messages = (
- self.get_message_history(chatbot, count=50, cursor=cursor) + messages
- )
+ messages = self.get_message_history(chatbot, count=50, cursor=cursor) + messages
while count > 0:
count -= 50
new_cursor = messages[0]["cursor"]
- new_messages = self.get_message_history(
- chatbot, min(50, count), cursor=new_cursor
- )
+ new_messages = self.get_message_history(chatbot, min(50, count), cursor=new_cursor)
messages = new_messages + messages
return messages
elif count <= 0:
@@ -523,9 +494,7 @@ class Client:
data = result["data"]["poeBotCreate"]
if data["status"] != "success":
- raise RuntimeError(
- f"Poe returned an error while trying to create a bot: {data['status']}"
- )
+ raise RuntimeError(f"Poe returned an error while trying to create a bot: {data['status']}")
self.get_bots()
return data
@@ -568,9 +537,7 @@ class Client:
data = result["data"]["poeBotEdit"]
if data["status"] != "success":
- raise RuntimeError(
- f"Poe returned an error while trying to edit a bot: {data['status']}"
- )
+ raise RuntimeError(f"Poe returned an error while trying to edit a bot: {data['status']}")
self.get_bots()
return data