From 125d0bde2d4d7b0fc8cb8239fe4815272cd4236b Mon Sep 17 00:00:00 2001
From: Michel Roux <xefir@crystalyx.net>
Date: Thu, 5 Jan 2023 15:19:04 +0100
Subject: [PATCH] Use flaresolverr v3

---
 pynyaata/flarerequests.py | 71 +++++++++++++--------------------------
 1 file changed, 23 insertions(+), 48 deletions(-)

diff --git a/pynyaata/flarerequests.py b/pynyaata/flarerequests.py
index e4ea5ac..a94a7a8 100644
--- a/pynyaata/flarerequests.py
+++ b/pynyaata/flarerequests.py
@@ -1,6 +1,6 @@
 from io import BytesIO
 from urllib import parse
-from requests import RequestException, Response, Session, post
+from requests import Response, Session, post
 from .config import CLOUDPROXY_ENDPOINT
 
 
@@ -9,67 +9,42 @@ class FlareRequests(Session):
         if not CLOUDPROXY_ENDPOINT:
             return super().request(method, url, params, data, **kwargs)
 
-        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)
-
         if params:
             url += "&" if len(url.split("?")) > 1 else "?"
             url = f"{url}{parse.urlencode(params)}"
 
         post_data = {
             "cmd": f"request.{method.lower()}",
-            "session": FLARESESSION,
             "url": url,
         }
 
         if data:
             post_data["postData"] = parse.urlencode(data)
 
-        try:
-            response = post(
-                CLOUDPROXY_ENDPOINT,
-                json=post_data,
-            )
+        response = post(
+            CLOUDPROXY_ENDPOINT,
+            json=post_data,
+        )
 
-            solution = response.json()
+        solution = response.json()
 
-            if "solution" in solution:
-                if "content-type" in solution["solution"]["headers"]:
-                    content_type = solution["solution"]["headers"][
-                        "content-type"
-                    ].split(";")
-                    if len(content_type) > 1:
-                        charset = content_type[1].split("=")
-                        if len(charset) > 1:
-                            encoding = charset[1]
+        if "solution" in solution:
+            headers = solution["solution"]["headers"]
+            if "content-type" in headers:
+                content_type = headers["content-type"].split(";")
+                if len(content_type) > 1:
+                    charset = content_type[1].split("=")
+                    if len(charset) > 1:
+                        encoding = charset[1]
 
-                resolved = Response()
+            resolved = Response()
 
-                resolved.status_code = solution["solution"]["status"]
-                resolved.headers = solution["solution"]["headers"]
-                resolved.raw = BytesIO(solution["solution"]["response"].encode())
-                resolved.url = url
-                resolved.encoding = encoding or None
-                resolved.reason = solution["status"]
-                resolved.cookies = solution["solution"]["cookies"]
+            resolved.status_code = solution["solution"]["status"]
+            resolved.headers = headers
+            resolved.raw = BytesIO(solution["solution"]["response"].encode())
+            resolved.url = url
+            resolved.encoding = encoding or None
+            resolved.reason = solution["status"]
+            resolved.cookies = solution["solution"]["cookies"]
 
-                return resolved
-
-            raise RequestException(response)
-        except RequestException:
-            session = post(
-                CLOUDPROXY_ENDPOINT,
-                json={"cmd": "sessions.destroy", "session": FLARESESSION},
-            )
-
-            raise RequestException(solution)
+            return resolved