This repository has been archived on 2023-10-01. You can view files and clone it, but cannot push or open issues or pull requests.
PyNyaaTa/pynyaata/flarerequests.py

52 lines
1.8 KiB
Python
Raw Normal View History

2022-12-21 14:53:50 +00:00
from urllib import parse
from requests import RequestException, Session, post
2022-12-21 15:14:43 +00:00
from .config import CLOUDPROXY_ENDPOINT, REQUESTS_TIMEOUT
2022-12-21 14:53:50 +00:00
class FlareRequests(Session):
2022-12-21 15:23:06 +00:00
def request(self, method, url, params=None, timeout=REQUESTS_TIMEOUT, **kwargs):
2022-12-21 14:53:50 +00:00
if not CLOUDPROXY_ENDPOINT:
2022-12-21 15:14:43 +00:00
return super().request(method, url, params, timeout=timeout, **kwargs)
2022-12-21 14:53:50 +00:00
sessions = post(CLOUDPROXY_ENDPOINT, json={"cmd": "sessions.list"}).json()
if "sessions" in sessions and len(sessions["sessions"]) > 0:
FLARESESSION = sessions["sessions"][0]
else:
response = post(CLOUDPROXY_ENDPOINT, json={"cmd": "sessions.create"})
session = response.json()
if "session" in session:
FLARESESSION = session["session"]
else:
raise RequestException(response)
try:
response = post(
CLOUDPROXY_ENDPOINT,
json={
"cmd": f"request.{method.lower()}",
"session": FLARESESSION,
"url": url,
"postData": parse.urlencode(params),
2022-12-21 15:14:43 +00:00
"maxTimeout": timeout,
2022-12-21 14:53:50 +00:00
},
)
solution = response.json()
if "solution" in solution:
response.cookies = solution["solution"]["cookies"]
response.headers = solution["solution"]["headers"]
response.text = solution["solution"]["response"]
return response
raise RequestException(response)
except RequestException:
session = post(
2022-12-21 15:14:43 +00:00
CLOUDPROXY_ENDPOINT,
{"cmd": "sessions.destroy", "session": FLARESESSION},
2022-12-21 14:53:50 +00:00
)
raise RequestException(solution)