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

quasarr/search/sources/dd.py

@@ -7,10 +7,13 @@ import time
 from base64 import urlsafe_b64encode
 from datetime import datetime, timezone
 
-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.sessions.dd import create_and_persist_session, retrieve_and_validate_session
+from quasarr.providers.log import debug, info
+from quasarr.providers.sessions.dd import (
+    create_and_persist_session,
+    retrieve_and_validate_session,
+)
 
 hostname = "dd"
 supported_mirrors = ["ironfiles", "rapidgator", "filefactory"]
@@ -18,7 +21,7 @@ supported_mirrors = ["ironfiles", "rapidgator", "filefactory"]
 
 def convert_to_rss_date(unix_timestamp):
     parsed_date = datetime.fromtimestamp(unix_timestamp, tz=timezone.utc)
-    rss_date = parsed_date.strftime('%a, %d %b %Y %H:%M:%S %z')
+    rss_date = parsed_date.strftime("%a, %d %b %Y %H:%M:%S %z")
 
     return rss_date
 
@@ -31,13 +34,23 @@ def dd_feed(*args, **kwargs):
     return dd_search(*args, **kwargs)
 
 
-def dd_search(shared_state, start_time, request_from, search_string="", mirror=None, season=None, episode=None):
+def dd_search(
+    shared_state,
+    start_time,
+    request_from,
+    search_string="",
+    mirror=None,
+    season=None,
+    episode=None,
+):
     releases = []
     dd = shared_state.values["config"]("Hostnames").get(hostname.lower())
     password = dd
 
     if not "arr" 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
 
     try:
@@ -51,13 +64,15 @@ def dd_search(shared_state, start_time, request_from, search_string="", mirror=N
         return releases
 
     if mirror and mirror not in supported_mirrors:
-        debug(f'Mirror "{mirror}" not supported by "{hostname.upper()}". Supported mirrors: {supported_mirrors}.'
-              ' Skipping search!')
+        debug(
+            f'Mirror "{mirror}" not supported by "{hostname.upper()}". Supported mirrors: {supported_mirrors}.'
+            " Skipping search!"
+        )
         return releases
 
     imdb_id = shared_state.is_imdb_id(search_string)
     if imdb_id:
-        search_string = get_localized_title(shared_state, imdb_id, 'en')
+        search_string = get_localized_title(shared_state, imdb_id, "en")
         if not search_string:
             info(f"Could not extract title from IMDb-ID {imdb_id}")
             return releases
@@ -79,17 +94,17 @@ def dd_search(shared_state, start_time, request_from, search_string="", mirror=N
         "web-1080p-x265",
         "web-2160p-x265-hdr",
         "movie-1080p-x265",
-        "movie-2160p-webdl-x265-hdr"
+        "movie-2160p-webdl-x265-hdr",
     ]
 
     headers = {
-        'User-Agent': shared_state.values["user_agent"],
+        "User-Agent": shared_state.values["user_agent"],
     }
 
     try:
         release_list = []
         for page in range(0, 100, 20):
-            url = f'https://{dd}/index/search/keyword/{search_string}/qualities/{','.join(qualities)}/from/{page}/search'
+            url = f"https://{dd}/index/search/keyword/{search_string}/qualities/{','.join(qualities)}/from/{page}/search"
 
             r = dd_session.get(url, headers=headers, timeout=timeout)
             r.raise_for_status()
@@ -101,17 +116,16 @@ def dd_search(shared_state, start_time, request_from, search_string="", mirror=N
             try:
                 if release.get("fake"):
                     debug(
-                        f"Release {release.get('release')} marked as fake. Invalidating {hostname.upper()} session...")
+                        f"Release {release.get('release')} marked as fake. Invalidating {hostname.upper()} session..."
+                    )
                     create_and_persist_session(shared_state)
                     return []
                 else:
                     title = release.get("release")
 
-                    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
 
                     imdb_id = release.get("imdbid", None)
@@ -121,31 +135,39 @@ def dd_search(shared_state, start_time, request_from, search_string="", mirror=N
                     mb = shared_state.convert_to_mb(size_item) * 1024 * 1024
                     published = convert_to_rss_date(release.get("when"))
                     payload = urlsafe_b64encode(
-                        f"{title}|{source}|{mirror}|{mb}|{password}|{imdb_id}|{hostname}".encode("utf-8")).decode(
-                        "utf-8")
+                        f"{title}|{source}|{mirror}|{mb}|{password}|{imdb_id}|{hostname}".encode(
+                            "utf-8"
+                        )
+                    ).decode("utf-8")
                     link = f"{shared_state.values['internal_address']}/download/?payload={payload}"
 
-                    releases.append({
-                        "details": {
-                            "title": title,
-                            "hostname": hostname.lower(),
-                            "imdb_id": imdb_id,
-                            "link": link,
-                            "mirror": mirror,
-                            "size": mb,
-                            "date": published,
-                            "source": source
-                        },
-                        "type": "protected"
-                    })
+                    releases.append(
+                        {
+                            "details": {
+                                "title": title,
+                                "hostname": hostname.lower(),
+                                "imdb_id": imdb_id,
+                                "link": link,
+                                "mirror": mirror,
+                                "size": mb,
+                                "date": published,
+                                "source": source,
+                            },
+                            "type": "protected",
+                        }
+                    )
             except Exception as e:
                 info(f"Error parsing {hostname.upper()} feed: {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"
+                )
                 continue
 
     except Exception as e:
         info(f"Error loading {hostname.upper()} {search_type}: {e}")
-        mark_hostname_issue(hostname, search_type, str(e) if "e" in dir() else "Error occurred")
+        mark_hostname_issue(
+            hostname, search_type, str(e) if "e" in dir() else "Error occurred"
+        )
 
     elapsed_time = time.time() - start_time
     debug(f"Time taken: {elapsed_time:.2f}s ({hostname})")

quasarr/search/sources/dj.py

@@ -11,18 +11,18 @@ from datetime import datetime, timedelta
 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 = "dj"
 
 
 def convert_to_rss_date(date_str):
     try:
-        return datetime.fromisoformat(
-            date_str.replace("Z", "+00:00")
-        ).strftime("%a, %d %b %Y %H:%M:%S +0000")
+        return datetime.fromisoformat(date_str.replace("Z", "+00:00")).strftime(
+            "%a, %d %b %Y %H:%M:%S +0000"
+        )
     except Exception:
         return ""
 
@@ -31,7 +31,9 @@ def dj_feed(shared_state, start_time, request_from, mirror=None):
     releases = []
 
     if "sonarr" not 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
 
     sj_host = shared_state.values["config"]("Hostnames").get(hostname)
@@ -46,7 +48,9 @@ def dj_feed(shared_state, start_time, request_from, mirror=None):
         data = json.loads(r.content)
     except Exception as e:
         info(f"{hostname.upper()}: feed load error: {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
 
     for release in data:
@@ -71,24 +75,30 @@ def dj_feed(shared_state, start_time, request_from, mirror=None):
             imdb_id = None
 
             payload = urlsafe_b64encode(
-                f"{title}|{series_url}|{mirror}|{mb}|{password}|{imdb_id}|{hostname}".encode("utf-8")
+                f"{title}|{series_url}|{mirror}|{mb}|{password}|{imdb_id}|{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": series_url
-                },
-                "type": "protected"
-            })
+            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": series_url,
+                    },
+                    "type": "protected",
+                }
+            )
 
         except Exception as e:
             debug(f"{hostname.upper()}: feed parse error: {e}")
@@ -101,11 +111,21 @@ def dj_feed(shared_state, start_time, request_from, mirror=None):
     return releases
 
 
-def dj_search(shared_state, start_time, request_from, search_string, mirror=None, season=None, episode=None):
+def dj_search(
+    shared_state,
+    start_time,
+    request_from,
+    search_string,
+    mirror=None,
+    season=None,
+    episode=None,
+):
     releases = []
 
     if "sonarr" not 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
 
     sj_host = shared_state.values["config"]("Hostnames").get(hostname)
@@ -130,10 +150,12 @@ def dj_search(shared_state, start_time, request_from, search_string, mirror=None
         results = soup.find_all("a", href=re.compile(r"^/serie/"))
     except Exception as e:
         info(f"{hostname.upper()}: search load error: {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"
+        )
         return releases
 
-    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")
     sanitized_search_string = shared_state.sanitize_string(localized_title)
 
     for result in results:
@@ -143,8 +165,7 @@ def dj_search(shared_state, start_time, request_from, search_string, mirror=None
             sanitized_title = shared_state.sanitize_string(result_title)
 
             if not re.search(
-                    rf"\b{re.escape(sanitized_search_string)}\b",
-                    sanitized_title
+                rf"\b{re.escape(sanitized_search_string)}\b", sanitized_title
             ):
                 debug(
                     f"Search string '{localized_title}' doesn't match '{result_title}'"
@@ -177,11 +198,7 @@ def dj_search(shared_state, start_time, request_from, search_string, mirror=None
                         continue
 
                     if not shared_state.is_valid_release(
-                            title,
-                            request_from,
-                            search_string,
-                            season,
-                            episode
+                        title, request_from, search_string, season, episode
                     ):
                         continue
 
@@ -194,24 +211,28 @@ def dj_search(shared_state, start_time, request_from, search_string, mirror=None
                     size = 0
 
                     payload = urlsafe_b64encode(
-                        f"{title}|{series_url}|{mirror}|{mb}|{password}|{imdb_id}|{hostname}".encode("utf-8")
+                        f"{title}|{series_url}|{mirror}|{mb}|{password}|{imdb_id}|{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": series_url
-                        },
-                        "type": "protected"
-                    })
+                    releases.append(
+                        {
+                            "details": {
+                                "title": title,
+                                "hostname": hostname,
+                                "imdb_id": imdb_id,
+                                "link": link,
+                                "mirror": mirror,
+                                "size": size,
+                                "date": published,
+                                "source": series_url,
+                            },
+                            "type": "protected",
+                        }
+                    )
 
         except Exception as e:
             debug(f"{hostname.upper()}: search parse error: {e}")