Spaces:
Paused
Paused
:zap: [Enhance] Enable create with new cookie item KievRPSSecAuth
Browse files- apis/chat_api.py +1 -0
- networks/cookies_constructor.py +12 -3
apis/chat_api.py
CHANGED
|
@@ -145,6 +145,7 @@ class ChatAPIApp:
|
|
| 145 |
sec_access_token=creator.sec_access_token,
|
| 146 |
client_id=creator.client_id,
|
| 147 |
conversation_id=creator.conversation_id,
|
|
|
|
| 148 |
invocation_id=0,
|
| 149 |
)
|
| 150 |
|
|
|
|
| 145 |
sec_access_token=creator.sec_access_token,
|
| 146 |
client_id=creator.client_id,
|
| 147 |
conversation_id=creator.conversation_id,
|
| 148 |
+
cookies=creator.request_cookies,
|
| 149 |
invocation_id=0,
|
| 150 |
)
|
| 151 |
|
networks/cookies_constructor.py
CHANGED
|
@@ -1,8 +1,10 @@
|
|
| 1 |
import json
|
| 2 |
import requests
|
|
|
|
|
|
|
|
|
|
| 3 |
from datetime import datetime
|
| 4 |
from pathlib import Path
|
| 5 |
-
|
| 6 |
from utils.logger import logger
|
| 7 |
|
| 8 |
|
|
@@ -14,6 +16,10 @@ class CookiesConstructor:
|
|
| 14 |
self.secrets_path = Path(__file__).parents[1] / "secrets.json"
|
| 15 |
self.created_datetime_format = "%Y-%m-%d %H:%M:%S"
|
| 16 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 17 |
def create_secrets_json(self):
|
| 18 |
if not self.secrets_path.exists():
|
| 19 |
self.secrets_path.parent.mkdir(parents=True, exist_ok=True)
|
|
@@ -54,6 +60,7 @@ class CookiesConstructor:
|
|
| 54 |
|
| 55 |
requests_body = {"cookies": ""}
|
| 56 |
try:
|
|
|
|
| 57 |
res = requests.post(
|
| 58 |
self.bypass_url,
|
| 59 |
json=requests_body,
|
|
@@ -61,7 +68,9 @@ class CookiesConstructor:
|
|
| 61 |
)
|
| 62 |
data = res.json()
|
| 63 |
cookies_str = data["result"]["cookies"]
|
| 64 |
-
|
|
|
|
|
|
|
| 65 |
logger.note(f"Get Cookies: {cookies_str}")
|
| 66 |
if cookies_str:
|
| 67 |
with open(self.secrets_path, "r") as rf:
|
|
@@ -71,7 +80,7 @@ class CookiesConstructor:
|
|
| 71 |
"created_time": datetime.now().strftime(
|
| 72 |
self.created_datetime_format
|
| 73 |
),
|
| 74 |
-
"
|
| 75 |
}
|
| 76 |
with open(self.secrets_path, "w") as wf:
|
| 77 |
json.dump(secrets, wf)
|
|
|
|
| 1 |
import json
|
| 2 |
import requests
|
| 3 |
+
import random
|
| 4 |
+
import string
|
| 5 |
+
|
| 6 |
from datetime import datetime
|
| 7 |
from pathlib import Path
|
|
|
|
| 8 |
from utils.logger import logger
|
| 9 |
|
| 10 |
|
|
|
|
| 16 |
self.secrets_path = Path(__file__).parents[1] / "secrets.json"
|
| 17 |
self.created_datetime_format = "%Y-%m-%d %H:%M:%S"
|
| 18 |
|
| 19 |
+
def generate_kiev_rps_sec_auth(self):
|
| 20 |
+
kiev = "".join(random.choices(string.ascii_uppercase + string.digits, k=32))
|
| 21 |
+
return kiev
|
| 22 |
+
|
| 23 |
def create_secrets_json(self):
|
| 24 |
if not self.secrets_path.exists():
|
| 25 |
self.secrets_path.parent.mkdir(parents=True, exist_ok=True)
|
|
|
|
| 60 |
|
| 61 |
requests_body = {"cookies": ""}
|
| 62 |
try:
|
| 63 |
+
logger.note(f"Requesting Cookies from: {self.bypass_url}")
|
| 64 |
res = requests.post(
|
| 65 |
self.bypass_url,
|
| 66 |
json=requests_body,
|
|
|
|
| 68 |
)
|
| 69 |
data = res.json()
|
| 70 |
cookies_str = data["result"]["cookies"]
|
| 71 |
+
cookies_screenshot = data["result"]["screenshot"]
|
| 72 |
+
kiev = self.generate_kiev_rps_sec_auth()
|
| 73 |
+
cookies_str = f"KievRPSSecAuth={kiev}; {cookies_str}"
|
| 74 |
logger.note(f"Get Cookies: {cookies_str}")
|
| 75 |
if cookies_str:
|
| 76 |
with open(self.secrets_path, "r") as rf:
|
|
|
|
| 80 |
"created_time": datetime.now().strftime(
|
| 81 |
self.created_datetime_format
|
| 82 |
),
|
| 83 |
+
"screenshot": self.bypass_url + cookies_screenshot,
|
| 84 |
}
|
| 85 |
with open(self.secrets_path, "w") as wf:
|
| 86 |
json.dump(secrets, wf)
|