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

quasarr/search/sources/wx.py

@@ -10,14 +10,15 @@ from base64 import urlsafe_b64encode
 from datetime import datetime
 
 import requests
-from bs4 import BeautifulSoup
-from bs4 import XMLParsedAsHTMLWarning
+from bs4 import BeautifulSoup, XMLParsedAsHTMLWarning
 
-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
 
-warnings.filterwarnings("ignore", category=XMLParsedAsHTMLWarning)  # we dont want to use lxml
+warnings.filterwarnings(
+    "ignore", category=XMLParsedAsHTMLWarning
+)  # we dont want to use lxml
 
 hostname = "wx"
 supported_mirrors = []
@@ -31,23 +32,25 @@ def wx_feed(shared_state, start_time, request_from, mirror=None):
     host = shared_state.values["config"]("Hostnames").get(hostname)
 
     if "lazylibrarian" in request_from.lower():
-        debug(f'Skipping {request_from} search on "{hostname.upper()}" (unsupported media type)!')
+        debug(
+            f'Skipping {request_from} search on "{hostname.upper()}" (unsupported media type)!'
+        )
         return releases
 
-    rss_url = f'https://{host}/rss'
+    rss_url = f"https://{host}/rss"
     headers = {
-        'User-Agent': shared_state.values["user_agent"],
+        "User-Agent": shared_state.values["user_agent"],
     }
 
     try:
         r = requests.get(rss_url, headers=headers, timeout=10)
         r.raise_for_status()
 
-        soup = BeautifulSoup(r.content, 'html.parser')
-        items = soup.find_all('entry')
+        soup = BeautifulSoup(r.content, "html.parser")
+        items = soup.find_all("entry")
 
         if not items:
-            items = soup.find_all('item')
+            items = soup.find_all("item")
 
         if not items:
             info(f"{hostname.upper()}: No entries found in RSS feed")
@@ -57,7 +60,7 @@ def wx_feed(shared_state, start_time, request_from, mirror=None):
 
         for item in items:
             try:
-                title_tag = item.find('title')
+                title_tag = item.find("title")
                 if not title_tag:
                     continue
 
@@ -66,14 +69,14 @@ def wx_feed(shared_state, start_time, request_from, mirror=None):
                     continue
 
                 title = html.unescape(title)
-                title = title.replace(']]>', '').replace('<![CDATA[', '')
-                title = title.replace(' ', '.')
+                title = title.replace("]]>", "").replace("<![CDATA[", "")
+                title = title.replace(" ", ".")
 
-                link_tag = item.find('link', rel='alternate')
-                if link_tag and link_tag.has_attr('href'):
-                    source = link_tag['href']
+                link_tag = item.find("link", rel="alternate")
+                if link_tag and link_tag.has_attr("href"):
+                    source = link_tag["href"]
                 else:
-                    link_tag = item.find('link')
+                    link_tag = item.find("link")
                     if not link_tag:
                         continue
                     source = link_tag.get_text(strip=True)
@@ -81,7 +84,7 @@ def wx_feed(shared_state, start_time, request_from, mirror=None):
                 if not source:
                     continue
 
-                pub_date = item.find('updated') or item.find('pubDate')
+                pub_date = item.find("updated") or item.find("pubDate")
                 if pub_date:
                     published = pub_date.get_text(strip=True)
                 else:
@@ -94,23 +97,27 @@ def wx_feed(shared_state, start_time, request_from, mirror=None):
                 password = host.upper()
 
                 payload = urlsafe_b64encode(
-                    f"{title}|{source}|{mirror}|{mb}|{password}|{imdb_id or ''}|{hostname}".encode("utf-8")
+                    f"{title}|{source}|{mirror}|{mb}|{password}|{imdb_id or ''}|{hostname}".encode(
+                        "utf-8"
+                    )
                 ).decode("utf-8")
                 link = f"{shared_state.values['internal_address']}/download/?payload={payload}"
 
-                releases.append({
-                    "details": {
-                        "title": title,
-                        "hostname": hostname,
-                        "imdb_id": imdb_id,
-                        "link": link,
-                        "mirror": mirror,
-                        "size": size,
-                        "date": published,
-                        "source": source
-                    },
-                    "type": "protected"
-                })
+                releases.append(
+                    {
+                        "details": {
+                            "title": title,
+                            "hostname": hostname,
+                            "imdb_id": imdb_id,
+                            "link": link,
+                            "mirror": mirror,
+                            "size": size,
+                            "date": published,
+                            "source": source,
+                        },
+                        "type": "protected",
+                    }
+                )
 
             except Exception as e:
                 debug(f"{hostname.upper()}: error parsing RSS entry: {e}")
@@ -118,7 +125,9 @@ def wx_feed(shared_state, start_time, request_from, mirror=None):
 
     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"
+        )
         return releases
 
     elapsed_time = time.time() - start_time
@@ -129,7 +138,15 @@ def wx_feed(shared_state, start_time, request_from, mirror=None):
     return releases
 
 
-def wx_search(shared_state, start_time, request_from, search_string, mirror=None, season=None, episode=None):
+def wx_search(
+    shared_state,
+    start_time,
+    request_from,
+    search_string,
+    mirror=None,
+    season=None,
+    episode=None,
+):
     """
     Search using internal API.
     Deduplicates results by fulltitle - each unique release appears only once.
@@ -138,48 +155,52 @@ def wx_search(shared_state, start_time, request_from, search_string, mirror=None
     host = shared_state.values["config"]("Hostnames").get(hostname)
 
     if "lazylibrarian" in request_from.lower():
-        debug(f'Skipping {request_from} search on "{hostname.upper()}" (unsupported media type)!')
+        debug(
+            f'Skipping {request_from} search on "{hostname.upper()}" (unsupported media type)!'
+        )
         return releases
 
     imdb_id = shared_state.is_imdb_id(search_string)
     if imdb_id:
         debug(f"{hostname.upper()}: Received IMDb ID: {imdb_id}")
-        title = get_localized_title(shared_state, imdb_id, 'de')
+        title = get_localized_title(shared_state, imdb_id, "de")
         if not title:
             debug(f"{hostname.upper()}: no title for IMDb {imdb_id}")
             return releases
-        debug(f"{hostname.upper()}: Translated IMDb {imdb_id} to German title: '{title}'")
+        debug(
+            f"{hostname.upper()}: Translated IMDb {imdb_id} to German title: '{title}'"
+        )
         search_string = html.unescape(title)
     else:
         debug(f"{hostname.upper()}: Using search string directly: '{search_string}'")
 
-    api_url = f'https://api.{host}/start/search'
+    api_url = f"https://api.{host}/start/search"
 
     headers = {
-        'User-Agent': shared_state.values["user_agent"],
-        'Accept': 'application/json, text/plain, */*',
-        'Referer': f'https://{host}/search'
+        "User-Agent": shared_state.values["user_agent"],
+        "Accept": "application/json, text/plain, */*",
+        "Referer": f"https://{host}/search",
     }
 
     params = {
-        '__LOAD_P': '',
-        'per_page': 50,
-        'q': search_string,
-        'selectedTypes': '',
-        'selectedGenres': '',
-        'types': 'movie,series,anime',
-        'genres': '',
-        'years': '',
-        'ratings': '',
-        'page': 1,
-        'sortBy': 'latest',
-        'sortOrder': 'desc'
+        "__LOAD_P": "",
+        "per_page": 50,
+        "q": search_string,
+        "selectedTypes": "",
+        "selectedGenres": "",
+        "types": "movie,series,anime",
+        "genres": "",
+        "years": "",
+        "ratings": "",
+        "page": 1,
+        "sortBy": "latest",
+        "sortOrder": "desc",
     }
 
     if "sonarr" in request_from.lower():
-        params['types'] = 'series,anime'
+        params["types"] = "series,anime"
     elif "radarr" in request_from.lower():
-        params['types'] = 'movie'
+        params["types"] = "movie"
 
     debug(f"{hostname.upper()}: Searching: '{search_string}'")
 
@@ -189,12 +210,12 @@ def wx_search(shared_state, start_time, request_from, search_string, mirror=None
 
         data = r.json()
 
-        if 'items' in data and 'data' in data['items']:
-            items = data['items']['data']
-        elif 'data' in data:
-            items = data['data']
-        elif 'results' in data:
-            items = data['results']
+        if "items" in data and "data" in data["items"]:
+            items = data["items"]["data"]
+        elif "data" in data:
+            items = data["data"]
+        elif "results" in data:
+            items = data["results"]
         else:
             items = data if isinstance(data, list) else []
 
@@ -205,124 +226,164 @@ def wx_search(shared_state, start_time, request_from, search_string, mirror=None
 
         for item in items:
             try:
-                uid = item.get('uid')
+                uid = item.get("uid")
                 if not uid:
                     debug(f"{hostname.upper()}: Item has no UID, skipping")
                     continue
 
                 debug(f"{hostname.upper()}: Fetching details for UID: {uid}")
 
-                detail_url = f'https://api.{host}/start/d/{uid}'
+                detail_url = f"https://api.{host}/start/d/{uid}"
                 detail_r = requests.get(detail_url, headers=headers, timeout=10)
                 detail_r.raise_for_status()
 
                 detail_data = detail_r.json()
 
-                if 'item' in detail_data:
-                    detail_item = detail_data['item']
+                if "item" in detail_data:
+                    detail_item = detail_data["item"]
                 else:
                     detail_item = detail_data
 
                 item_imdb_id = imdb_id
                 if not item_imdb_id:
-                    item_imdb_id = detail_item.get('imdb_id') or detail_item.get('imdbid')
-                    if not item_imdb_id and 'options' in detail_item:
-                        item_imdb_id = detail_item['options'].get('imdb_id')
+                    item_imdb_id = detail_item.get("imdb_id") or detail_item.get(
+                        "imdbid"
+                    )
+                    if not item_imdb_id and "options" in detail_item:
+                        item_imdb_id = detail_item["options"].get("imdb_id")
 
                 source = f"https://{host}/detail/{uid}"
 
-                main_title = detail_item.get('fulltitle') or detail_item.get('title') or detail_item.get('name')
+                main_title = (
+                    detail_item.get("fulltitle")
+                    or detail_item.get("title")
+                    or detail_item.get("name")
+                )
                 if main_title:
                     title = html.unescape(main_title)
-                    title = title.replace(' ', '.')
+                    title = title.replace(" ", ".")
 
-                    if shared_state.is_valid_release(title, request_from, search_string, season, episode):
+                    if shared_state.is_valid_release(
+                        title, request_from, search_string, season, episode
+                    ):
                         # Skip if we've already seen this exact title
                         if title in seen_titles:
-                            debug(f"{hostname.upper()}: Skipping duplicate main title: {title}")
+                            debug(
+                                f"{hostname.upper()}: Skipping duplicate main title: {title}"
+                            )
                         else:
                             seen_titles.add(title)
-                            published = detail_item.get('updated_at') or detail_item.get('created_at')
+                            published = detail_item.get(
+                                "updated_at"
+                            ) or detail_item.get("created_at")
                             if not published:
-                                published = datetime.now().strftime("%a, %d %b %Y %H:%M:%S +0000")
+                                published = datetime.now().strftime(
+                                    "%a, %d %b %Y %H:%M:%S +0000"
+                                )
                             password = f"www.{host}"
 
                             payload = urlsafe_b64encode(
                                 f"{title}|{source}|{mirror}|0|{password}|{item_imdb_id or ''}|{hostname}".encode(
-                                    "utf-8")
+                                    "utf-8"
+                                )
                             ).decode("utf-8")
                             link = f"{shared_state.values['internal_address']}/download/?payload={payload}"
 
-                            releases.append({
-                                "details": {
-                                    "title": title,
-                                    "hostname": hostname,
-                                    "imdb_id": item_imdb_id,
-                                    "link": link,
-                                    "mirror": mirror,
-                                    "size": 0,
-                                    "date": published,
-                                    "source": source
-                                },
-                                "type": "protected"
-                            })
-
-                if 'releases' in detail_item and isinstance(detail_item['releases'], list):
-                    debug(f"{hostname.upper()}: Found {len(detail_item['releases'])} releases for {uid}")
-
-                    for release in detail_item['releases']:
+                            releases.append(
+                                {
+                                    "details": {
+                                        "title": title,
+                                        "hostname": hostname,
+                                        "imdb_id": item_imdb_id,
+                                        "link": link,
+                                        "mirror": mirror,
+                                        "size": 0,
+                                        "date": published,
+                                        "source": source,
+                                    },
+                                    "type": "protected",
+                                }
+                            )
+
+                if "releases" in detail_item and isinstance(
+                    detail_item["releases"], list
+                ):
+                    debug(
+                        f"{hostname.upper()}: Found {len(detail_item['releases'])} releases for {uid}"
+                    )
+
+                    for release in detail_item["releases"]:
                         try:
-                            release_title = release.get('fulltitle')
+                            release_title = release.get("fulltitle")
                             if not release_title:
                                 continue
 
                             release_title = html.unescape(release_title)
-                            release_title = release_title.replace(' ', '.')
-
-                            if not shared_state.is_valid_release(release_title, request_from, search_string, season,
-                                                                 episode):
-                                debug(f"{hostname.upper()}: ✗ Release filtered out: {release_title}")
+                            release_title = release_title.replace(" ", ".")
+
+                            if not shared_state.is_valid_release(
+                                release_title,
+                                request_from,
+                                search_string,
+                                season,
+                                episode,
+                            ):
+                                debug(
+                                    f"{hostname.upper()}: ✗ Release filtered out: {release_title}"
+                                )
                                 continue
 
                             # Skip if we've already seen this exact title (deduplication)
                             if release_title in seen_titles:
-                                debug(f"{hostname.upper()}: Skipping duplicate release: {release_title}")
+                                debug(
+                                    f"{hostname.upper()}: Skipping duplicate release: {release_title}"
+                                )
                                 continue
 
                             seen_titles.add(release_title)
 
-                            release_uid = release.get('uid')
+                            release_uid = release.get("uid")
                             if release_uid:
-                                release_source = f"https://{host}/detail/{uid}?release={release_uid}"
+                                release_source = (
+                                    f"https://{host}/detail/{uid}?release={release_uid}"
+                                )
                             else:
                                 release_source = source
 
-                            release_published = release.get('updated_at') or release.get(
-                                'created_at') or detail_item.get('updated_at')
+                            release_published = (
+                                release.get("updated_at")
+                                or release.get("created_at")
+                                or detail_item.get("updated_at")
+                            )
                             if not release_published:
-                                release_published = datetime.now().strftime("%a, %d %b %Y %H:%M:%S +0000")
-                            release_size = release.get('size', 0)
+                                release_published = datetime.now().strftime(
+                                    "%a, %d %b %Y %H:%M:%S +0000"
+                                )
+                            release_size = release.get("size", 0)
                             password = f"www.{host}"
 
                             payload = urlsafe_b64encode(
                                 f"{release_title}|{release_source}|{mirror}|{release_size}|{password}|{item_imdb_id or ''}|{hostname}".encode(
-                                    "utf-8")
+                                    "utf-8"
+                                )
                             ).decode("utf-8")
                             link = f"{shared_state.values['internal_address']}/download/?payload={payload}"
 
-                            releases.append({
-                                "details": {
-                                    "title": release_title,
-                                    "hostname": hostname,
-                                    "imdb_id": item_imdb_id,
-                                    "link": link,
-                                    "mirror": mirror,
-                                    "size": release_size,
-                                    "date": release_published,
-                                    "source": release_source
-                                },
-                                "type": "protected"
-                            })
+                            releases.append(
+                                {
+                                    "details": {
+                                        "title": release_title,
+                                        "hostname": hostname,
+                                        "imdb_id": item_imdb_id,
+                                        "link": link,
+                                        "mirror": mirror,
+                                        "size": release_size,
+                                        "date": release_published,
+                                        "source": release_source,
+                                    },
+                                    "type": "protected",
+                                }
+                            )
 
                         except Exception as e:
                             debug(f"{hostname.upper()}: Error parsing release: {e}")
@@ -335,11 +396,15 @@ def wx_search(shared_state, start_time, request_from, search_string, mirror=None
                 debug(f"{hostname.upper()}: {traceback.format_exc()}")
                 continue
 
-        debug(f"{hostname.upper()}: Returning {len(releases)} total releases (deduplicated)")
+        debug(
+            f"{hostname.upper()}: Returning {len(releases)} total releases (deduplicated)"
+        )
 
     except Exception as e:
         info(f"Error in {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"
+        )
 
         debug(f"{hostname.upper()}: {traceback.format_exc()}")
         return releases