quasarr 2.6.1__py3-none-any.whl → 2.7.0__py3-none-any.whl

quasarr/search/sources/he.py

@@ -13,7 +13,7 @@ from bs4 import BeautifulSoup
 
 from quasarr.providers.hostname_issues import clear_hostname_issue, mark_hostname_issue
 from quasarr.providers.imdb_metadata import get_localized_title, get_year
-from quasarr.providers.log import debug, info
+from quasarr.providers.log import debug, info, trace
 
 hostname = "he"
 supported_mirrors = ["rapidgator", "nitroflare"]
@@ -74,7 +74,7 @@ def he_search(
 
     if not "arr" in request_from.lower():
         debug(
-            f'Skipping {request_from} search on "{hostname.upper()}" (unsupported media type)!'
+            f'<d>Skipping {request_from} search on "{hostname.upper()}" (unsupported media type)!</d>'
         )
         return releases
 
@@ -84,7 +84,7 @@ def he_search(
         tag = "tv-shows"
 
     if mirror and mirror not in supported_mirrors:
-        debug(f'Mirror "{mirror}" not supported by {hostname}.')
+        debug(f'Mirror "{mirror}" not supported.')
         return releases
 
     source_search = ""
@@ -93,7 +93,7 @@ def he_search(
         if imdb_id:
             local_title = get_localized_title(shared_state, imdb_id, "en")
             if not local_title:
-                info(f"{hostname}: no title for IMDb {imdb_id}")
+                info(f"No title for IMDb {imdb_id}")
                 return releases
             if not season:
                 year = get_year(imdb_id)
@@ -130,7 +130,7 @@ def he_search(
         soup = BeautifulSoup(r.content, "html.parser")
         results = soup.find_all("div", class_="item")
     except Exception as e:
-        info(f"{hostname}: {search_type} load error: {e}")
+        info(f"{search_type} load error: {e}")
         mark_hostname_issue(
             hostname, search_type, str(e) if "e" in dir() else "Error occurred"
         )
@@ -196,22 +196,25 @@ def he_search(
                 mark_hostname_issue(
                     hostname, search_type, str(e) if "e" in dir() else "Error occurred"
                 )
+            release_imdb_id = None
             try:
                 imdb_link = soup.find(
                     "a", href=re.compile(r"imdb\.com/title/tt\d+", re.IGNORECASE)
                 )
                 if imdb_link:
                     release_imdb_id = re.search(r"tt\d+", imdb_link["href"]).group()
-                    if imdb_id and release_imdb_id != imdb_id:
-                        debug(
-                            f"{hostname}: IMDb ID mismatch: expected {imdb_id}, found {release_imdb_id}"
+                    if imdb_id and release_imdb_id and release_imdb_id != imdb_id:
+                        trace(
+                            f"IMDb ID mismatch: expected {imdb_id}, found {release_imdb_id}"
                         )
                         continue
                 else:
-                    debug(f"{hostname}: imdb link not found for title {title}")
+                    trace(f"imdb link not found for title {title}")
             except Exception:
-                debug(f"{hostname}: failed to determine imdb_id for title {title}")
-                continue
+                debug(f"failed to determine imdb_id for title {title}")
+
+            if release_imdb_id is None:
+                release_imdb_id = imdb_id
 
             password = None
             payload = urlsafe_b64encode(
@@ -239,12 +242,11 @@ def he_search(
                 }
             )
         except Exception as e:
-            debug(f"{hostname}: error parsing search result: {e}")
+            debug(f"error parsing search result: {e}")
             continue
 
     elapsed = time.time() - start_time
-    debug(f"Time taken: {elapsed:.2f}s ({hostname})")
-
+    debug(f"Time taken: {elapsed:.2f}s")
     if releases:
         clear_hostname_issue(hostname)
     return releases

quasarr/search/sources/hs.py

@@ -14,7 +14,7 @@ from bs4 import BeautifulSoup, XMLParsedAsHTMLWarning
 from quasarr.providers.hostname_issues import clear_hostname_issue, mark_hostname_issue
 
 warnings.filterwarnings("ignore", category=XMLParsedAsHTMLWarning)
-from quasarr.providers.log import debug, info
+from quasarr.providers.log import debug, warn
 
 hostname = "hs"
 supported_mirrors = ["rapidgator", "ddownload", "katfile"]
@@ -353,7 +353,7 @@ def hs_feed(shared_state, start_time, request_from, mirror=None):
     # HS only supports movies and series
     if "lazylibrarian" in request_from.lower():
         debug(
-            f'Skipping {request_from} feed on "{hostname.upper()}" (unsupported media type)!'
+            f'<d>Skipping {request_from} feed on "{hostname.upper()}" (unsupported media type)!</d>'
         )
         return releases
 
@@ -431,12 +431,12 @@ def hs_feed(shared_state, start_time, request_from, mirror=None):
                 continue
 
     except Exception as e:
-        info(f"Error loading {hostname.upper()} feed: {e}")
+        warn(f"Error loading {hostname.upper()} feed: {e}")
         mark_hostname_issue(hostname, "feed", str(e))
         return releases
 
     elapsed_time = time.time() - start_time
-    debug(f"Time taken: {elapsed_time:.2f}s ({hostname})")
+    debug(f"Time taken: {elapsed_time:.2f}s")
 
     if releases:
         clear_hostname_issue(hostname)
@@ -463,7 +463,7 @@ def hs_search(
     # HS only supports movies and series
     if "lazylibrarian" in request_from.lower():
         debug(
-            f'Skipping {request_from} search on "{hostname.upper()}" (unsupported media type)!'
+            f'<d>Skipping {request_from} search on "{hostname.upper()}" (unsupported media type)!</d>'
         )
         return releases
 
@@ -503,12 +503,12 @@ def hs_search(
         )
 
     except Exception as e:
-        info(f"Error loading {hostname.upper()} search: {e}")
+        warn(f"Error loading {hostname.upper()} search: {e}")
         mark_hostname_issue(hostname, "search", str(e))
         return releases
 
     elapsed_time = time.time() - start_time
-    debug(f"Time taken: {elapsed_time:.2f}s ({hostname})")
+    debug(f"Time taken: {elapsed_time:.2f}s")
 
     if releases:
         clear_hostname_issue(hostname)

quasarr/search/sources/mb.py

@@ -12,7 +12,7 @@ import requests
 from bs4 import BeautifulSoup
 
 from quasarr.providers.hostname_issues import clear_hostname_issue, mark_hostname_issue
-from quasarr.providers.log import debug, info
+from quasarr.providers.log import debug, warn
 
 hostname = "mb"
 supported_mirrors = ["rapidgator", "ddownload"]
@@ -169,7 +169,7 @@ def mb_feed(shared_state, start_time, request_from, mirror=None):
 
     if not "arr" in request_from.lower():
         debug(
-            f'Skipping {request_from} search on "{hostname.upper()}" (unsupported media type)!'
+            f'<d>Skipping {request_from} search on "{hostname.upper()}" (unsupported media type)!</d>'
         )
         return []
 
@@ -183,12 +183,12 @@ def mb_feed(shared_state, start_time, request_from, mirror=None):
         soup = BeautifulSoup(r.content, "html.parser")
         releases = _parse_posts(soup, shared_state, password, mirror_filter=mirror)
     except Exception as e:
-        info(f"Error loading {hostname.upper()} feed: {e}")
+        warn(f"Error loading {hostname.upper()} feed: {e}")
         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})")
+    debug(f"Time taken: {time.time() - start_time:.2f}s")
 
     if releases:
         clear_hostname_issue(hostname)
@@ -208,7 +208,7 @@ def mb_search(
 
     if not "arr" in request_from.lower():
         debug(
-            f'Skipping {request_from} search on "{hostname.upper()}" (unsupported media type)!'
+            f'<d>Skipping {request_from} search on "{hostname.upper()}" (unsupported media type)!</d>'
         )
         return []
 
@@ -236,12 +236,12 @@ def mb_search(
             episode=episode,
         )
     except Exception as e:
-        info(f"Error loading {hostname.upper()} search: {e}")
+        warn(f"Error loading {hostname.upper()} search: {e}")
         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})")
+    debug(f"Time taken: {time.time() - start_time:.2f}s")
 
     if releases:
         clear_hostname_issue(hostname)

quasarr/search/sources/nk.py

@@ -14,7 +14,7 @@ from bs4 import BeautifulSoup
 
 from quasarr.providers.hostname_issues import clear_hostname_issue, mark_hostname_issue
 from quasarr.providers.imdb_metadata import get_localized_title, get_year
-from quasarr.providers.log import debug, info
+from quasarr.providers.log import debug, info, trace
 
 hostname = "nk"
 supported_mirrors = ["rapidgator", "ddownload"]
@@ -74,12 +74,12 @@ def nk_search(
 
     if not "arr" in request_from.lower():
         debug(
-            f'Skipping {request_from} search on "{hostname.upper()}" (unsupported media type)!'
+            f'<d>Skipping {request_from} search on "{hostname.upper()}" (unsupported media type)!</d>'
         )
         return releases
 
     if mirror and mirror not in supported_mirrors:
-        debug(f'Mirror "{mirror}" not supported by {hostname}.')
+        debug(f'Mirror "{mirror}" not supported.')
         return releases
 
     source_search = ""
@@ -88,7 +88,7 @@ def nk_search(
         if imdb_id:
             local_title = get_localized_title(shared_state, imdb_id, "de")
             if not local_title:
-                info(f"{hostname}: no title for IMDb {imdb_id}")
+                info(f"No title for IMDb {imdb_id}")
                 return releases
             if not season:
                 year = get_year(imdb_id)
@@ -124,7 +124,7 @@ def nk_search(
         soup = BeautifulSoup(r.content, "html.parser")
         results = soup.find_all("div", class_="article-right")
     except Exception as e:
-        info(f"{hostname}: {search_type} load error: {e}")
+        info(f"{search_type} load error: {e}")
         mark_hostname_issue(
             hostname, search_type, str(e) if "e" in dir() else "Error occurred"
         )
@@ -135,23 +135,6 @@ def nk_search(
 
     for result in results:
         try:
-            imdb_a = result.select_one("a.imdb")
-            if imdb_a and imdb_a.get("href"):
-                try:
-                    release_imdb_id = re.search(r"tt\d+", imdb_a["href"]).group()
-                    if imdb_id:
-                        if release_imdb_id != imdb_id:
-                            debug(
-                                f"{hostname}: IMDb ID mismatch: expected {imdb_id}, found {release_imdb_id}"
-                            )
-                            continue
-                except Exception:
-                    debug(f"{hostname}: could not extract IMDb ID")
-                    continue
-            else:
-                debug(f"{hostname}: could not extract IMDb ID")
-                continue
-
             a = result.find("a", class_="release-details", href=True)
             if not a:
                 continue
@@ -162,6 +145,23 @@ def nk_search(
             else:
                 continue
 
+            imdb_a = result.select_one("a.imdb")
+            if imdb_a and imdb_a.get("href"):
+                try:
+                    release_imdb_id = re.search(r"tt\d+", imdb_a["href"]).group()
+                    if imdb_id and release_imdb_id and release_imdb_id != imdb_id:
+                        trace(
+                            f"IMDb ID mismatch: expected {imdb_id}, found {release_imdb_id}"
+                        )
+                        continue
+                except Exception:
+                    debug(f"failed to determine imdb_id for {title}")
+            else:
+                trace(f"imdb link not found for {title}")
+
+            if release_imdb_id is None:
+                release_imdb_id = imdb_id
+
             if not shared_state.is_valid_release(
                 title, request_from, search_string, season, episode
             ):
@@ -230,12 +230,11 @@ def nk_search(
             )
         except Exception as e:
             info(e)
-            debug(f"{hostname}: error parsing search result: {e}")
+            debug(f"error parsing search result: {e}")
             continue
 
     elapsed = time.time() - start_time
-    debug(f"Time taken: {elapsed:.2f}s ({hostname})")
-
+    debug(f"Time taken: {elapsed:.2f}s")
     if releases:
         clear_hostname_issue(hostname)
     return releases

quasarr/search/sources/nx.py

@@ -10,7 +10,7 @@ import requests
 
 from quasarr.providers.hostname_issues import clear_hostname_issue, mark_hostname_issue
 from quasarr.providers.imdb_metadata import get_localized_title, get_year
-from quasarr.providers.log import debug, info
+from quasarr.providers.log import debug, info, trace, warn
 
 hostname = "nx"
 supported_mirrors = ["filer"]
@@ -30,8 +30,8 @@ def nx_feed(shared_state, start_time, request_from, mirror=None):
 
     if mirror and mirror not in supported_mirrors:
         debug(
-            f'Mirror "{mirror}" not supported by "{hostname.upper()}". Supported mirrors: {supported_mirrors}.'
-            " Skipping search!"
+            f'Mirror "{mirror}" not supported by "{hostname.upper()}". Supported mirrors: {supported_mirrors}. '
+            "Skipping search!"
         )
         return releases
 
@@ -47,7 +47,7 @@ def nx_feed(shared_state, start_time, request_from, mirror=None):
         r.raise_for_status()
         feed = r.json()
     except Exception as e:
-        info(f"Error loading {hostname.upper()} feed: {e}")
+        warn(f"Error loading {hostname.upper()} feed: {e}")
         mark_hostname_issue(
             hostname, "feed", str(e) if "e" in dir() else "Error occurred"
         )
@@ -103,13 +103,13 @@ def nx_feed(shared_state, start_time, request_from, mirror=None):
                 )
 
         except Exception as e:
-            info(f"Error parsing {hostname.upper()} feed: {e}")
+            warn(f"Error parsing {hostname.upper()} feed: {e}")
             mark_hostname_issue(
                 hostname, "feed", str(e) if "e" in dir() else "Error occurred"
             )
 
     elapsed_time = time.time() - start_time
-    debug(f"Time taken: {elapsed_time:.2f}s ({hostname})")
+    debug(f"Time taken: {elapsed_time:.2f}s")
 
     if releases:
         clear_hostname_issue(hostname)
@@ -138,8 +138,8 @@ def nx_search(
 
     if mirror and mirror not in supported_mirrors:
         debug(
-            f'Mirror "{mirror}" not supported by "{hostname.upper()}". Supported mirrors: {supported_mirrors}.'
-            " Skipping search!"
+            f'Mirror "{mirror}" not supported by "{hostname.upper()}". Supported mirrors: {supported_mirrors}. '
+            "Skipping search!"
         )
         return releases
 
@@ -164,7 +164,7 @@ def nx_search(
         r.raise_for_status()
         feed = r.json()
     except Exception as e:
-        info(f"Error loading {hostname.upper()} search: {e}")
+        warn(f"Error loading {hostname.upper()} search: {e}")
         mark_hostname_issue(
             hostname, "search", str(e) if "e" in dir() else "Error occurred"
         )
@@ -187,12 +187,19 @@ def nx_search(
 
                     try:
                         source = f"https://{nx}/release/{item['slug']}"
-                        if not imdb_id:
-                            imdb_id = item.get("_media", {}).get("imdbid", None)
+                        release_imdb_id = item.get("_media", {}).get("imdbid", None)
+                        if imdb_id and release_imdb_id and release_imdb_id != imdb_id:
+                            trace(
+                                f"{hostname.upper()}: Skipping result '{title}' due to IMDb ID mismatch."
+                            )
+                            continue
+
+                        if release_imdb_id is None:
+                            release_imdb_id = imdb_id
 
                         mb = shared_state.convert_to_mb(item)
                         payload = urlsafe_b64encode(
-                            f"{title}|{source}|{mirror}|{mb}|{password}|{imdb_id}".encode(
+                            f"{title}|{source}|{mirror}|{mb}|{password}|{release_imdb_id}".encode(
                                 "utf-8"
                             )
                         ).decode("utf-8")
@@ -215,7 +222,7 @@ def nx_search(
                             "details": {
                                 "title": title,
                                 "hostname": hostname.lower(),
-                                "imdb_id": imdb_id,
+                                "imdb_id": release_imdb_id,
                                 "link": link,
                                 "mirror": mirror,
                                 "size": size,
@@ -227,13 +234,13 @@ def nx_search(
                     )
 
         except Exception as e:
-            info(f"Error parsing {hostname.upper()} search: {e}")
+            warn(f"Error parsing {hostname.upper()} search: {e}")
             mark_hostname_issue(
                 hostname, "search", str(e) if "e" in dir() else "Error occurred"
             )
 
     elapsed_time = time.time() - start_time
-    debug(f"Time taken: {elapsed_time:.2f}s ({hostname})")
+    debug(f"Time taken: {elapsed_time:.2f}s")
 
     if releases:
         clear_hostname_issue(hostname)

quasarr/search/sources/sf.py

@@ -12,7 +12,7 @@ import requests
 
 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 debug, info
+from quasarr.providers.log import debug, info, trace, warn
 
 hostname = "sf"
 supported_mirrors = ["1fichier", "ddownload", "katfile", "rapidgator", "turbobit"]
@@ -102,7 +102,7 @@ def sf_feed(shared_state, start_time, request_from, mirror=None):
 
     if not "sonarr" in request_from.lower():
         debug(
-            f'Skipping {request_from} search on "{hostname.upper()}" (unsupported media type)!'
+            f'<d>Skipping {request_from} search on "{hostname.upper()}" (unsupported media type)!</d>'
         )
         return releases
 
@@ -131,7 +131,7 @@ def sf_feed(shared_state, start_time, request_from, mirror=None):
             )
             r.raise_for_status()
         except Exception as e:
-            info(f"Error loading {hostname.upper()} feed: {e} for {formatted_date}")
+            warn(f"Error loading {hostname.upper()} feed: {e} for {formatted_date}")
             mark_hostname_issue(
                 hostname, "feed", str(e) if "e" in dir() else "Error occurred"
             )
@@ -194,7 +194,7 @@ def sf_feed(shared_state, start_time, request_from, mirror=None):
                 )
 
     elapsed_time = time.time() - start_time
-    debug(f"Time taken: {elapsed_time:.2f}s ({hostname})")
+    debug(f"Time taken: {elapsed_time:.2f}s")
 
     if releases:
         clear_hostname_issue(hostname)
@@ -234,7 +234,7 @@ def sf_search(
 
     if not "sonarr" in request_from.lower():
         debug(
-            f'Skipping {request_from} search on "{hostname.upper()}" (unsupported media type)!'
+            f'<d>Skipping {request_from} search on "{hostname.upper()}" (unsupported media type)!</d>'
         )
         return releases
 
@@ -255,7 +255,7 @@ def sf_search(
         r.raise_for_status()
         feed = r.json()
     except Exception as e:
-        info(f"Error loading {hostname.upper()} search: {e}")
+        warn(f"Error loading {hostname.upper()} search: {e}")
         mark_hostname_issue(
             hostname, "search", str(e) if "e" in dir() else "Error occurred"
         )
@@ -266,11 +266,11 @@ def sf_search(
         sanitized_search_string = shared_state.sanitize_string(search_string)
         sanitized_title = shared_state.sanitize_string(result.get("title", ""))
         if not re.search(rf"\b{re.escape(sanitized_search_string)}\b", sanitized_title):
-            debug(
+            trace(
                 f"Search string '{search_string}' doesn't match '{result.get('title')}'"
             )
             continue
-        debug(
+        trace(
             f"Matched search string '{search_string}' with result '{result.get('title')}'"
         )
 
@@ -334,6 +334,15 @@ def sf_search(
                 )
                 continue
 
+            if imdb_id_in_search and imdb_id and imdb_id != imdb_id_in_search:
+                trace(
+                    f"{hostname.upper()}: Skipping result '{result.get('title')}' due to IMDb ID mismatch."
+                )
+                continue
+
+            if imdb_id is None:
+                imdb_id = imdb_id_in_search
+
             # cache content and imdb_id
             entry["content"] = data_html
             entry["imdb_id"] = imdb_id
@@ -451,7 +460,7 @@ def sf_search(
                 debug(f"Error parsing item for '{search_string}': {e}")
 
     elapsed_time = time.time() - start_time
-    debug(f"Time taken: {elapsed_time:.2f}s ({hostname})")
+    debug(f"Time taken: {elapsed_time:.2f}s")
 
     if releases:
         clear_hostname_issue(hostname)

quasarr/search/sources/sj.py

@@ -13,7 +13,7 @@ from bs4 import BeautifulSoup
 
 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 debug, info
+from quasarr.providers.log import debug, info, trace
 
 hostname = "sj"
 
@@ -32,7 +32,7 @@ def sj_feed(shared_state, start_time, request_from, mirror=None):
 
     if "sonarr" not in request_from.lower():
         debug(
-            f'Skipping {request_from} search on "{hostname.upper()}" (unsupported media type)!'
+            f'<d>Skipping {request_from} search on "{hostname.upper()}" (unsupported media type)!</d>'
         )
         return releases
 
@@ -104,7 +104,7 @@ def sj_feed(shared_state, start_time, request_from, mirror=None):
             debug(f"{hostname.upper()}: feed parse error: {e}")
             continue
 
-    debug(f"Time taken: {time.time() - start_time:.2f}s ({hostname})")
+    debug(f"Time taken: {time.time() - start_time:.2f}s")
 
     if releases:
         clear_hostname_issue(hostname)
@@ -124,7 +124,7 @@ def sj_search(
 
     if "sonarr" not in request_from.lower():
         debug(
-            f'Skipping {request_from} search on "{hostname.upper()}" (unsupported media type)!'
+            f'<d>Skipping {request_from} search on "{hostname.upper()}" (unsupported media type)!</d>'
         )
         return releases
 
@@ -168,12 +168,12 @@ def sj_search(
             if not re.search(
                 rf"\b{re.escape(sanitized_search_string)}\b", sanitized_title
             ):
-                debug(
+                trace(
                     f"Search string '{localized_title}' doesn't match '{result_title}'"
                 )
                 continue
 
-            debug(
+            trace(
                 f"Matched search string '{localized_title}' with result '{result_title}'"
             )
 
@@ -240,7 +240,7 @@ def sj_search(
             debug(f"{hostname.upper()}: search parse error: {e}")
             continue
 
-    debug(f"Time taken: {time.time() - start_time:.2f}s ({hostname})")
+    debug(f"Time taken: {time.time() - start_time:.2f}s")
 
     if releases:
         clear_hostname_issue(hostname)