quasarr 2.4.8__py3-none-any.whl → 2.4.9__py3-none-any.whl

quasarr/search/sources/wd.py

@@ -12,9 +12,9 @@ from urllib.parse import quote, quote_plus
 import requests
 from bs4 import BeautifulSoup
 
-from quasarr.providers.hostname_issues import mark_hostname_issue, clear_hostname_issue
+from quasarr.providers.hostname_issues import clear_hostname_issue, mark_hostname_issue
 from quasarr.providers.imdb_metadata import get_localized_title
-from quasarr.providers.log import info, debug
+from quasarr.providers.log import debug, info
 
 hostname = "wd"
 supported_mirrors = ["rapidgator", "ddownload", "katfile", "fikper", "turbobit"]
@@ -47,15 +47,15 @@ def extract_size(text):
 
 
 def _parse_rows(
-        soup,
-        shared_state,
-        url_base,
-        password,
-        mirror_filter,
-        request_from=None,
-        search_string=None,
-        season=None,
-        episode=None
+    soup,
+    shared_state,
+    url_base,
+    password,
+    mirror_filter,
+    request_from=None,
+    search_string=None,
+    season=None,
+    episode=None,
 ):
     """
     Walk the <table> rows, extract one release per row.
@@ -75,7 +75,7 @@ def _parse_rows(
     releases = []
     is_search = search_string is not None
 
-    one_hour_ago = (datetime.now() - timedelta(hours=1)).strftime('%Y-%m-%d %H:%M:%S')
+    one_hour_ago = (datetime.now() - timedelta(hours=1)).strftime("%Y-%m-%d %H:%M:%S")
 
     for tr in soup.select("table.table tbody tr.lh-sm"):
         try:
@@ -93,28 +93,30 @@ def _parse_rows(
 
             # search context contains non-video releases (ebooks, games, etc.)
             if is_search:
-                if not shared_state.is_valid_release(title,
-                                                     request_from,
-                                                     search_string,
-                                                     season,
-                                                     episode):
+                if not shared_state.is_valid_release(
+                    title, request_from, search_string, season, episode
+                ):
                     continue
 
-                if 'lazylibrarian' in request_from.lower():
+                if "lazylibrarian" in request_from.lower():
                     # lazylibrarian can only detect specific date formats / issue numbering for magazines
                     title = shared_state.normalize_magazine_title(title)
                 else:
                     # drop .XXX. unless user explicitly searched xxx
-                    if XXX_REGEX.search(title) and 'xxx' not in search_string.lower():
+                    if XXX_REGEX.search(title) and "xxx" not in search_string.lower():
                         continue
                     # require resolution/codec
-                    if not (RESOLUTION_REGEX.search(title) or CODEC_REGEX.search(title)):
+                    if not (
+                        RESOLUTION_REGEX.search(title) or CODEC_REGEX.search(title)
+                    ):
                         continue
                     # require no spaces in title
                     if " " in title:
                         continue
 
-            hoster_names = tr.find("span", class_="button-warezkorb")["data-hoster-names"]
+            hoster_names = tr.find("span", class_="button-warezkorb")[
+                "data-hoster-names"
+            ]
             mirrors = [m.strip().lower() for m in hoster_names.split(",")]
             valid = [m for m in mirrors if m in supported_mirrors]
             if not valid or (mirror_filter and mirror_filter not in valid):
@@ -131,21 +133,25 @@ def _parse_rows(
             payload = urlsafe_b64encode(
                 f"{title}|{source}|{mirror_filter}|{mb}|{password}|{imdb_id}|{hostname}".encode()
             ).decode()
-            download_link = f"{shared_state.values['internal_address']}/download/?payload={payload}"
-
-            releases.append({
-                "details": {
-                    "title": title,
-                    "hostname": hostname,
-                    "imdb_id": imdb_id,
-                    "link": download_link,
-                    "mirror": mirror_filter,
-                    "size": size_bytes,
-                    "date": published,
-                    "source": source
-                },
-                "type": "protected"
-            })
+            download_link = (
+                f"{shared_state.values['internal_address']}/download/?payload={payload}"
+            )
+
+            releases.append(
+                {
+                    "details": {
+                        "title": title,
+                        "hostname": hostname,
+                        "imdb_id": imdb_id,
+                        "link": download_link,
+                        "mirror": mirror_filter,
+                        "size": size_bytes,
+                        "date": published,
+                        "source": source,
+                    },
+                    "type": "protected",
+                }
+            )
         except Exception as e:
             debug(f"Error parsing {hostname.upper()} row: {e}")
             continue
@@ -164,7 +170,7 @@ def wd_feed(shared_state, start_time, request_from, mirror=None):
         feed_type = "Serien"
 
     url = f"https://{wd}/{feed_type}"
-    headers = {'User-Agent': shared_state.values["user_agent"]}
+    headers = {"User-Agent": shared_state.values["user_agent"]}
     try:
         r = requests.get(url, headers=headers, timeout=10)
         r.raise_for_status()
@@ -172,7 +178,9 @@ def wd_feed(shared_state, start_time, request_from, mirror=None):
         releases = _parse_rows(soup, shared_state, wd, password, mirror)
     except Exception as e:
         info(f"Error loading {hostname.upper()} feed: {e}")
-        mark_hostname_issue(hostname, "feed", str(e) if "e" in dir() else "Error occurred")
+        mark_hostname_issue(
+            hostname, "feed", str(e) if "e" in dir() else "Error occurred"
+        )
         releases = []
     debug(f"Time taken: {time.time() - start_time:.2f}s ({hostname})")
 
@@ -181,14 +189,22 @@ def wd_feed(shared_state, start_time, request_from, mirror=None):
     return releases
 
 
-def wd_search(shared_state, start_time, request_from, search_string, mirror=None, season=None, episode=None):
+def wd_search(
+    shared_state,
+    start_time,
+    request_from,
+    search_string,
+    mirror=None,
+    season=None,
+    episode=None,
+):
     releases = []
     wd = shared_state.values["config"]("Hostnames").get(hostname.lower())
     password = wd
 
     imdb_id = shared_state.is_imdb_id(search_string)
     if imdb_id:
-        search_string = get_localized_title(shared_state, imdb_id, 'de')
+        search_string = get_localized_title(shared_state, imdb_id, "de")
         if not search_string:
             info(f"Could not extract title from IMDb-ID {imdb_id}")
             return releases
@@ -196,21 +212,28 @@ def wd_search(shared_state, start_time, request_from, search_string, mirror=None
 
     q = quote_plus(search_string)
     url = f"https://{wd}/search?q={q}"
-    headers = {'User-Agent': shared_state.values["user_agent"]}
+    headers = {"User-Agent": shared_state.values["user_agent"]}
 
     try:
         r = requests.get(url, headers=headers, timeout=10)
         r.raise_for_status()
         soup = BeautifulSoup(r.content, "html.parser")
         releases = _parse_rows(
-            soup, shared_state, wd, password, mirror,
+            soup,
+            shared_state,
+            wd,
+            password,
+            mirror,
             request_from=request_from,
             search_string=search_string,
-            season=season, episode=episode
+            season=season,
+            episode=episode,
         )
     except Exception as e:
         info(f"Error loading {hostname.upper()} search: {e}")
-        mark_hostname_issue(hostname, "search", str(e) if "e" in dir() else "Error occurred")
+        mark_hostname_issue(
+            hostname, "search", str(e) if "e" in dir() else "Error occurred"
+        )
         releases = []
     debug(f"Time taken: {time.time() - start_time:.2f}s ({hostname})")