updated quora(poe) (token creation patched atm)
This commit is contained in:
parent
6c64e459b3
commit
8c0f5e9634
5 changed files with 86 additions and 60 deletions
|
@ -8,6 +8,9 @@ from pathlib import Path
|
||||||
from random import choice, choices, randint
|
from random import choice, choices, randint
|
||||||
from string import ascii_letters, digits
|
from string import ascii_letters, digits
|
||||||
from urllib import parse
|
from urllib import parse
|
||||||
|
from os import urandom
|
||||||
|
from hashlib import md5
|
||||||
|
from json import dumps
|
||||||
|
|
||||||
class PoeResponse:
|
class PoeResponse:
|
||||||
|
|
||||||
|
@ -91,6 +94,7 @@ class Model:
|
||||||
"sec-ch-ua" : "\"Chromium\";v=\"112\", \"Google Chrome\";v=\"112\", \"Not:A-Brand\";v=\"99\"",
|
"sec-ch-ua" : "\"Chromium\";v=\"112\", \"Google Chrome\";v=\"112\", \"Not:A-Brand\";v=\"99\"",
|
||||||
"sec-ch-ua-mobile" : "?0",
|
"sec-ch-ua-mobile" : "?0",
|
||||||
"sec-ch-ua-platform": "\"macOS\"",
|
"sec-ch-ua-platform": "\"macOS\"",
|
||||||
|
"content-type" : "application/json",
|
||||||
"sec-fetch-site" : "same-origin",
|
"sec-fetch-site" : "same-origin",
|
||||||
"sec-fetch-mode" : "cors",
|
"sec-fetch-mode" : "cors",
|
||||||
"sec-fetch-dest" : "empty",
|
"sec-fetch-dest" : "empty",
|
||||||
|
@ -155,6 +159,7 @@ class Account:
|
||||||
"accept" : "text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.7",
|
"accept" : "text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.7",
|
||||||
"sec-fetch-site" : "same-origin",
|
"sec-fetch-site" : "same-origin",
|
||||||
"sec-fetch-mode" : "navigate",
|
"sec-fetch-mode" : "navigate",
|
||||||
|
"content-type" : "application/json",
|
||||||
"sec-fetch-user" : "?1",
|
"sec-fetch-user" : "?1",
|
||||||
"sec-fetch-dest" : "document",
|
"sec-fetch-dest" : "document",
|
||||||
"accept-encoding" : "gzip, deflate, br",
|
"accept-encoding" : "gzip, deflate, br",
|
||||||
|
@ -168,16 +173,20 @@ class Account:
|
||||||
client.headers["poe-formkey"] = next_data['props']['formkey']
|
client.headers["poe-formkey"] = next_data['props']['formkey']
|
||||||
client.headers["poe-tchannel"] = client.get('https://poe.com/api/settings').json()['tchannelData']['channel']
|
client.headers["poe-tchannel"] = client.get('https://poe.com/api/settings').json()['tchannelData']['channel']
|
||||||
|
|
||||||
payload = {
|
payload = dumps(separators = (',', ':'), obj = {
|
||||||
"queryName": "MainSignupLoginSection_sendVerificationCodeMutation_Mutation",
|
'queryName': 'MainSignupLoginSection_sendVerificationCodeMutation_Mutation',
|
||||||
"variables": {
|
'variables': {
|
||||||
"emailAddress": mail_address,
|
'emailAddress': mail_address,
|
||||||
"phoneNumber" : None
|
'phoneNumber': None,
|
||||||
|
'recaptchaToken': None,
|
||||||
},
|
},
|
||||||
"query": "mutation MainSignupLoginSection_sendVerificationCodeMutation_Mutation(\n $emailAddress: String\n $phoneNumber: String\n) {\n sendVerificationCode(verificationReason: login, emailAddress: $emailAddress, phoneNumber: $phoneNumber) {\n status\n errorMessage\n }\n}\n"
|
'query': 'mutation MainSignupLoginSection_sendVerificationCodeMutation_Mutation(\n $emailAddress: String\n $phoneNumber: String\n $recaptchaToken: String\n) {\n sendVerificationCode(verificationReason: login, emailAddress: $emailAddress, phoneNumber: $phoneNumber, recaptchaToken: $recaptchaToken) {\n status\n errorMessage\n }\n}\n',
|
||||||
}
|
})
|
||||||
|
|
||||||
response = client.post('https://poe.com/api/gql_POST', json=payload)
|
base_string = payload + client.headers["poe-formkey"] + 'WpuLMiXEKKE98j56k'
|
||||||
|
client.headers["poe-tag-id"] = md5(base_string.encode()).hexdigest()
|
||||||
|
|
||||||
|
response = client.post('https://poe.com/api/gql_POST', data=payload)
|
||||||
if 'Bad Request' in response.text:
|
if 'Bad Request' in response.text:
|
||||||
if logging: print('bad request, retrying...' , response.json())
|
if logging: print('bad request, retrying...' , response.json())
|
||||||
Account.create(proxy = proxy, logging = logging)
|
Account.create(proxy = proxy, logging = logging)
|
||||||
|
@ -197,7 +206,7 @@ class Account:
|
||||||
|
|
||||||
if logging: print('code', mail_token)
|
if logging: print('code', mail_token)
|
||||||
|
|
||||||
payload = {
|
payload = dumps(separators = (',', ':'), obj={
|
||||||
"queryName": "SignupOrLoginWithCodeSection_signupWithVerificationCodeMutation_Mutation",
|
"queryName": "SignupOrLoginWithCodeSection_signupWithVerificationCodeMutation_Mutation",
|
||||||
"variables": {
|
"variables": {
|
||||||
"verificationCode" : mail_token,
|
"verificationCode" : mail_token,
|
||||||
|
@ -205,9 +214,12 @@ class Account:
|
||||||
"phoneNumber" : None
|
"phoneNumber" : None
|
||||||
},
|
},
|
||||||
"query": "mutation SignupOrLoginWithCodeSection_signupWithVerificationCodeMutation_Mutation(\n $verificationCode: String!\n $emailAddress: String\n $phoneNumber: String\n) {\n signupWithVerificationCode(verificationCode: $verificationCode, emailAddress: $emailAddress, phoneNumber: $phoneNumber) {\n status\n errorMessage\n }\n}\n"
|
"query": "mutation SignupOrLoginWithCodeSection_signupWithVerificationCodeMutation_Mutation(\n $verificationCode: String!\n $emailAddress: String\n $phoneNumber: String\n) {\n signupWithVerificationCode(verificationCode: $verificationCode, emailAddress: $emailAddress, phoneNumber: $phoneNumber) {\n status\n errorMessage\n }\n}\n"
|
||||||
}
|
})
|
||||||
|
|
||||||
response = client.post('https://poe.com/api/gql_POST', json = payload)
|
base_string = payload + client.headers["poe-formkey"] + 'WpuLMiXEKKE98j56k'
|
||||||
|
client.headers["poe-tag-id"] = md5(base_string.encode()).hexdigest()
|
||||||
|
|
||||||
|
response = client.post('https://poe.com/api/gql_POST', data = payload)
|
||||||
if logging: print('verify_code', response.json())
|
if logging: print('verify_code', response.json())
|
||||||
|
|
||||||
token = parse.unquote(client.cookies.get_dict()['p-b'])
|
token = parse.unquote(client.cookies.get_dict()['p-b'])
|
||||||
|
@ -216,11 +228,17 @@ class Account:
|
||||||
f.write(f'{token}\n')
|
f.write(f'{token}\n')
|
||||||
|
|
||||||
if enable_bot_creation:
|
if enable_bot_creation:
|
||||||
client.post("https://poe.com/api/gql_POST", json = {
|
|
||||||
|
payload = {
|
||||||
"queryName": "UserProfileConfigurePreviewModal_markMultiplayerNuxCompleted_Mutation",
|
"queryName": "UserProfileConfigurePreviewModal_markMultiplayerNuxCompleted_Mutation",
|
||||||
"variables": {},
|
"variables": {},
|
||||||
"query": "mutation UserProfileConfigurePreviewModal_markMultiplayerNuxCompleted_Mutation {\n markMultiplayerNuxCompleted {\n viewer {\n hasCompletedMultiplayerNux\n id\n }\n }\n}\n"
|
"query": "mutation UserProfileConfigurePreviewModal_markMultiplayerNuxCompleted_Mutation {\n markMultiplayerNuxCompleted {\n viewer {\n hasCompletedMultiplayerNux\n id\n }\n }\n}\n"
|
||||||
})
|
}
|
||||||
|
|
||||||
|
base_string = dumps(payload, separators = (',', ':')) + client.headers["poe-formkey"] + 'WpuLMiXEKKE98j56k'
|
||||||
|
client.headers["poe-tag-id"] = md5(base_string.encode()).hexdigest()
|
||||||
|
|
||||||
|
client.post("https://poe.com/api/gql_POST", json = payload)
|
||||||
|
|
||||||
return token
|
return token
|
||||||
|
|
||||||
|
|
92
quora/api.py
92
quora/api.py
|
@ -1,19 +1,3 @@
|
||||||
# ading2210/poe-api: a reverse engineered Python API wrapepr for Quora's Poe
|
|
||||||
# Copyright (C) 2023 ading2210
|
|
||||||
|
|
||||||
# This program is free software: you can redistribute it and/or modify
|
|
||||||
# it under the terms of the GNU General Public License as published by
|
|
||||||
# the Free Software Foundation, either version 3 of the License, or
|
|
||||||
# (at your option) any later version.
|
|
||||||
|
|
||||||
# This program is distributed in the hope that it will be useful,
|
|
||||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
||||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
||||||
# GNU General Public License for more details.
|
|
||||||
|
|
||||||
# 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 requests
|
||||||
import re
|
import re
|
||||||
import json
|
import json
|
||||||
|
@ -71,14 +55,6 @@ class Client:
|
||||||
home_url = "https://poe.com"
|
home_url = "https://poe.com"
|
||||||
settings_url = "https://poe.com/api/settings"
|
settings_url = "https://poe.com/api/settings"
|
||||||
|
|
||||||
formkey = ""
|
|
||||||
next_data = {}
|
|
||||||
bots = {}
|
|
||||||
active_messages = {}
|
|
||||||
message_queues = {}
|
|
||||||
ws = None
|
|
||||||
ws_connected = False
|
|
||||||
|
|
||||||
def __init__(self, token, proxy=None):
|
def __init__(self, token, proxy=None):
|
||||||
self.proxy = proxy
|
self.proxy = proxy
|
||||||
self.session = requests.Session()
|
self.session = requests.Session()
|
||||||
|
@ -90,6 +66,9 @@ class Client:
|
||||||
}
|
}
|
||||||
logger.info(f"Proxy enabled: {self.proxy}")
|
logger.info(f"Proxy enabled: {self.proxy}")
|
||||||
|
|
||||||
|
self.active_messages = {}
|
||||||
|
self.message_queues = {}
|
||||||
|
|
||||||
self.session.cookies.set("p-b", token, domain="poe.com")
|
self.session.cookies.set("p-b", token, domain="poe.com")
|
||||||
self.headers = {
|
self.headers = {
|
||||||
"User-Agent": user_agent,
|
"User-Agent": user_agent,
|
||||||
|
@ -99,10 +78,10 @@ class Client:
|
||||||
self.ws_domain = f"tch{random.randint(1, 1e6)}"
|
self.ws_domain = f"tch{random.randint(1, 1e6)}"
|
||||||
|
|
||||||
self.session.headers.update(self.headers)
|
self.session.headers.update(self.headers)
|
||||||
self.next_data = self.get_next_data()
|
self.next_data = self.get_next_data(overwrite_vars=True)
|
||||||
self.channel = self.get_channel_data()
|
self.channel = self.get_channel_data()
|
||||||
self.connect_ws()
|
self.connect_ws()
|
||||||
self.bots = self.get_bots()
|
self.bots = self.get_bots(download_next_data=False)
|
||||||
self.bot_names = self.get_bot_names()
|
self.bot_names = self.get_bot_names()
|
||||||
|
|
||||||
self.gql_headers = {
|
self.gql_headers = {
|
||||||
|
@ -112,7 +91,7 @@ class Client:
|
||||||
self.gql_headers = {**self.gql_headers, **self.headers}
|
self.gql_headers = {**self.gql_headers, **self.headers}
|
||||||
self.subscribe()
|
self.subscribe()
|
||||||
|
|
||||||
def get_next_data(self):
|
def get_next_data(self, overwrite_vars=False):
|
||||||
logger.info("Downloading next_data...")
|
logger.info("Downloading next_data...")
|
||||||
|
|
||||||
r = request_with_retries(self.session.get, self.home_url)
|
r = request_with_retries(self.session.get, self.home_url)
|
||||||
|
@ -120,8 +99,9 @@ class Client:
|
||||||
json_text = re.search(json_regex, r.text).group(1)
|
json_text = re.search(json_regex, r.text).group(1)
|
||||||
next_data = json.loads(json_text)
|
next_data = json.loads(json_text)
|
||||||
|
|
||||||
self.formkey = next_data["props"]["formkey"]
|
if overwrite_vars:
|
||||||
self.viewer = next_data["props"]["pageProps"]["payload"]["viewer"]
|
self.formkey = next_data["props"]["formkey"]
|
||||||
|
self.viewer = next_data["props"]["pageProps"]["payload"]["viewer"]
|
||||||
|
|
||||||
return next_data
|
return next_data
|
||||||
|
|
||||||
|
@ -134,17 +114,23 @@ class Client:
|
||||||
chat_data = r.json()["pageProps"]["payload"]["chatOfBotDisplayName"]
|
chat_data = r.json()["pageProps"]["payload"]["chatOfBotDisplayName"]
|
||||||
return chat_data
|
return chat_data
|
||||||
|
|
||||||
def get_bots(self):
|
def get_bots(self, download_next_data=True):
|
||||||
viewer = self.next_data["props"]["pageProps"]["payload"]["viewer"]
|
if download_next_data:
|
||||||
if not "availableBots" in viewer:
|
next_data = self.get_next_data()
|
||||||
raise RuntimeError("Invalid token.")
|
else:
|
||||||
bot_list = viewer["availableBots"]
|
next_data = self.next_data
|
||||||
|
|
||||||
|
if not "availableBots" in self.viewer:
|
||||||
|
raise RuntimeError("Invalid token or no bots are available.")
|
||||||
|
bot_list = self.viewer["availableBots"]
|
||||||
|
|
||||||
bots = {}
|
bots = {}
|
||||||
for bot in bot_list:
|
for bot in bot_list:
|
||||||
chat_data = self.get_bot(bot["displayName"].lower())
|
chat_data = self.get_bot(bot["displayName"])
|
||||||
bots[chat_data["defaultBotObject"]["nickname"]] = chat_data
|
bots[chat_data["defaultBotObject"]["nickname"]] = chat_data
|
||||||
|
|
||||||
|
self.bots = bots
|
||||||
|
self.bot_names = self.get_bot_names()
|
||||||
return bots
|
return bots
|
||||||
|
|
||||||
def get_bot_names(self):
|
def get_bot_names(self):
|
||||||
|
@ -170,9 +156,20 @@ class Client:
|
||||||
|
|
||||||
def send_query(self, query_name, variables):
|
def send_query(self, query_name, variables):
|
||||||
for i in range(20):
|
for i in range(20):
|
||||||
payload = generate_payload(query_name, variables)
|
json_data = generate_payload(query_name, variables)
|
||||||
|
payload = json.dumps(json_data, separators=(',', ':'))
|
||||||
|
|
||||||
|
base_string = payload + self.gql_headers['poe-formkey'] + 'WpuLMiXEKKE98j56k'
|
||||||
|
|
||||||
|
from hashlib import md5
|
||||||
|
headers = self.gql_headers |{
|
||||||
|
"content-type": "application/json",
|
||||||
|
"poe-tag-id": md5(base_string.encode()).hexdigest()
|
||||||
|
}
|
||||||
|
|
||||||
r = request_with_retries(
|
r = request_with_retries(
|
||||||
self.session.post, self.gql_url, json=payload, headers=self.gql_headers)
|
self.session.post, self.gql_url, data=payload, headers=headers)
|
||||||
|
|
||||||
data = r.json()
|
data = r.json()
|
||||||
if data["data"] == None:
|
if data["data"] == None:
|
||||||
logger.warn(
|
logger.warn(
|
||||||
|
@ -212,6 +209,7 @@ class Client:
|
||||||
self.ws.run_forever(**kwargs)
|
self.ws.run_forever(**kwargs)
|
||||||
|
|
||||||
def connect_ws(self):
|
def connect_ws(self):
|
||||||
|
self.ws_connected = False
|
||||||
self.ws = websocket.WebSocketApp(
|
self.ws = websocket.WebSocketApp(
|
||||||
self.get_websocket_url(),
|
self.get_websocket_url(),
|
||||||
header={"User-Agent": user_agent},
|
header={"User-Agent": user_agent},
|
||||||
|
@ -233,9 +231,10 @@ class Client:
|
||||||
def on_ws_connect(self, ws):
|
def on_ws_connect(self, ws):
|
||||||
self.ws_connected = True
|
self.ws_connected = True
|
||||||
|
|
||||||
def on_ws_close(self, ws, close_status_code):
|
def on_ws_close(self, ws, close_status_code, close_message):
|
||||||
self.ws_connected = False
|
self.ws_connected = False
|
||||||
logger.warn(f"Websocket closed with status {close_status_code}")
|
logger.warn(
|
||||||
|
f"Websocket closed with status {close_status_code}: {close_message}")
|
||||||
|
|
||||||
def on_ws_error(self, ws, error):
|
def on_ws_error(self, ws, error):
|
||||||
self.disconnect_ws()
|
self.disconnect_ws()
|
||||||
|
@ -345,22 +344,27 @@ class Client:
|
||||||
def get_message_history(self, chatbot, count=25, cursor=None):
|
def get_message_history(self, chatbot, count=25, cursor=None):
|
||||||
logger.info(f"Downloading {count} messages from {chatbot}")
|
logger.info(f"Downloading {count} messages from {chatbot}")
|
||||||
|
|
||||||
|
messages = []
|
||||||
if cursor == None:
|
if cursor == None:
|
||||||
chat_data = self.get_bot(self.bot_names[chatbot])
|
chat_data = self.get_bot(self.bot_names[chatbot])
|
||||||
if not chat_data["messagesConnection"]["edges"]:
|
if not chat_data["messagesConnection"]["edges"]:
|
||||||
return []
|
return []
|
||||||
cursor = chat_data["messagesConnection"]["edges"][-1]["cursor"]
|
messages = chat_data["messagesConnection"]["edges"][:count]
|
||||||
|
cursor = chat_data["messagesConnection"]["pageInfo"]["startCursor"]
|
||||||
|
count -= len(messages)
|
||||||
|
|
||||||
cursor = str(cursor)
|
cursor = str(cursor)
|
||||||
if count > 50:
|
if count > 50:
|
||||||
messages = self.get_message_history(
|
messages = self.get_message_history(
|
||||||
chatbot, count=50, cursor=cursor)
|
chatbot, count=50, cursor=cursor) + messages
|
||||||
while count > 0:
|
while count > 0:
|
||||||
|
count -= 50
|
||||||
new_cursor = messages[0]["cursor"]
|
new_cursor = messages[0]["cursor"]
|
||||||
new_messages = self.get_message_history(
|
new_messages = self.get_message_history(
|
||||||
chatbot, min(50, count), cursor=new_cursor)
|
chatbot, min(50, count), cursor=new_cursor)
|
||||||
messages = new_messages + messages
|
messages = new_messages + messages
|
||||||
count -= 50
|
return messages
|
||||||
|
elif count <= 0:
|
||||||
return messages
|
return messages
|
||||||
|
|
||||||
result = self.send_query("ChatListPaginationQuery", {
|
result = self.send_query("ChatListPaginationQuery", {
|
||||||
|
@ -368,7 +372,9 @@ class Client:
|
||||||
"cursor": cursor,
|
"cursor": cursor,
|
||||||
"id": self.bots[chatbot]["id"]
|
"id": self.bots[chatbot]["id"]
|
||||||
})
|
})
|
||||||
return result["data"]["node"]["messagesConnection"]["edges"]
|
query_messages = result["data"]["node"]["messagesConnection"]["edges"]
|
||||||
|
messages = query_messages + messages
|
||||||
|
return messages
|
||||||
|
|
||||||
def delete_message(self, message_ids):
|
def delete_message(self, message_ids):
|
||||||
logger.info(f"Deleting messages: {message_ids}")
|
logger.info(f"Deleting messages: {message_ids}")
|
||||||
|
|
|
@ -13,3 +13,5 @@ UP2wQVds17VFHh6IfCQFrA==
|
||||||
FNgKEpc2r-XqWe0rHBfYpg==
|
FNgKEpc2r-XqWe0rHBfYpg==
|
||||||
juCAh6kB0sUpXHvKik2woA==
|
juCAh6kB0sUpXHvKik2woA==
|
||||||
nBvuNYRLaE4xE4HuzBPiIQ==
|
nBvuNYRLaE4xE4HuzBPiIQ==
|
||||||
|
oyae3iClomSrk6RJywZ4iw==
|
||||||
|
1Z27Ul8BTdNOhncT5H6wdg==
|
||||||
|
|
|
@ -6,7 +6,7 @@ print('token', token)
|
||||||
|
|
||||||
sleep(2)
|
sleep(2)
|
||||||
|
|
||||||
for response in quora.StreamingCompletion.create(model = 'gpt-4',
|
for response in quora.StreamingCompletion.create(model = 'gpt-3.5-turbo',
|
||||||
prompt = 'hello world',
|
prompt = 'hello world',
|
||||||
token = token):
|
token = token):
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue