Fix encoding

commands/pygg.py

@@ -37,11 +37,12 @@ class FlareRequests(requests.Session):
                 raise requests.RequestException(response)
 
         url += "&" if len(url.split("?")) > 1 else "?"
+        url = f"{url}{urllib.parse.urlencode(params)}"
 
         post_data = {
             "cmd": f"request.{method.lower()}",
             "session": FLARESESSION,
-            "url": f"{url}{urllib.parse.urlencode(params)}",
+            "url": url,
         }
 
         if data:
@@ -56,12 +57,22 @@ class FlareRequests(requests.Session):
             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]
+
                 resolved = requests.Response()
 
-                resolved.raw = io.BytesIO(solution["solution"]["response"].encode())
                 resolved.status_code = solution["solution"]["status"]
                 resolved.headers = solution["solution"]["headers"]
+                resolved.raw = io.BytesIO(solution["solution"]["response"].encode())
                 resolved.url = url
+                resolved.encoding = encoding or None
                 resolved.reason = solution["status"]
                 resolved.cookies = solution["solution"]["cookies"]
 
@@ -128,7 +139,10 @@ def search_ygg(query, multi, full):
         ygg_params["option_langue:multiple[]"] = "4"
 
     req = session.get(f"{YGGTORRENT_BASE_URL}/engine/search", params=ygg_params)
-    html = bs4.BeautifulSoup(req.text, "html.parser")
+    html = bs4.BeautifulSoup(
+        req.text,
+        "html.parser",
+    )
     trs = html.select("table.table tr")
 
     if len(trs) > 1: