quasarr 1.22.0__py3-none-any.whl → 1.24.0__py3-none-any.whl

quasarr/downloads/sources/sf.py

@@ -19,30 +19,62 @@ def is_last_section_integer(url):
     return None
 
 
-def get_sf_download_links(shared_state, url, mirror, title):  # signature must align with other download link functions!
+def resolve_sf_redirect(url, user_agent):
+    """Follow redirects and return final URL or None if 404."""
+    try:
+        response = requests.get(url, allow_redirects=True, timeout=10,
+                                headers={'User-Agent': user_agent})
+        if response.history:
+            for resp in response.history:
+                debug(f"Redirected from {resp.url} to {response.url}")
+            if "/404.html" in response.url:
+                info(f"SF link redirected to 404 page: {response.url}")
+                return None
+            return response.url
+        else:
+            info(f"SF blocked attempt to resolve {url}. Your IP may be banned. Try again later.")
+    except Exception as e:
+        info(f"Error fetching redirected URL for {url}: {e}")
+    return None
+
+
+def get_sf_download_links(shared_state, url, mirror, title, password):
+    """
+    KEEP THE SIGNATURE EVEN IF SOME PARAMETERS ARE UNUSED!
+
+    SF source handler - resolves redirects and returns filecrypt links.
+    """
+
+    sf = shared_state.values["config"]("Hostnames").get("sf")
+    user_agent = shared_state.values["user_agent"]
+
+    # Handle external redirect URLs
+    if url.startswith(f"https://{sf}/external"):
+        resolved_url = resolve_sf_redirect(url, user_agent)
+        if not resolved_url:
+            return {"links": [], "imdb_id": None}
+        return {"links": [[resolved_url, "filecrypt"]], "imdb_id": None}
+
+    # Handle series page URLs - need to find the right release
     release_pattern = re.compile(
         r'''
-          ^                                   # start of string
-          (?P<name>.+?)\.                     # show name (dots in name) up to the dot before “S”
-          S(?P<season>\d+)                    # “S” + season number
-          (?:E\d+(?:-E\d+)?)?                 # optional “E##” or “E##-E##”
-          \.                                  # literal dot
-          .*?\.                               # anything (e.g. language/codec) up to next dot
-          (?P<resolution>\d+p)                # resolution “720p”, “1080p”, etc.
-          \..+?                               # dot + more junk (e.g. “.WEB.h264”)
-          -(?P<group>\w+)                     # dash + release group at end
-          $                                   # end of string
+          ^                                   
+          (?P<name>.+?)\.                     
+          S(?P<season>\d+)                    
+          (?:E\d+(?:-E\d+)?)?                 
+          \.                                  
+          .*?\.                               
+          (?P<resolution>\d+p)                
+          \..+?                               
+          -(?P<group>\w+)                     
+          $                                   
         ''',
         re.IGNORECASE | re.VERBOSE
     )
 
     release_match = release_pattern.match(title)
-
     if not release_match:
-        return {
-            "real_url": None,
-            "imdb_id": None,
-        }
+        return {"links": [], "imdb_id": None}
 
     release_parts = release_match.groupdict()
 
@@ -51,15 +83,11 @@ def get_sf_download_links(shared_state, url, mirror, title):  # signature must a
         if not season:
             season = "ALL"
 
-        sf = shared_state.values["config"]("Hostnames").get("sf")
-        headers = {
-            'User-Agent': shared_state.values["user_agent"],
-        }
-
+        headers = {'User-Agent': user_agent}
         series_page = requests.get(url, headers=headers, timeout=10).text
-
         soup = BeautifulSoup(series_page, "html.parser")
-        # extract IMDb id if present
+
+        # Extract IMDb id if present
         imdb_id = None
         a_imdb = soup.find("a", href=re.compile(r"imdb\.com/title/tt\d+"))
         if a_imdb:
@@ -82,7 +110,6 @@ def get_sf_download_links(shared_state, url, mirror, title):  # signature must a
             data = response.json()["html"]
 
         content = BeautifulSoup(data, "html.parser")
-
         items = content.find_all("h3")
 
         for item in items:
@@ -101,15 +128,13 @@ def get_sf_download_links(shared_state, url, mirror, title):  # signature must a
 
                 result_parts = result_match.groupdict()
 
-                # Normalize all relevant fields for case-insensitive comparison
                 name_match = release_parts['name'].lower() == result_parts['name'].lower()
-                season_match = release_parts['season'] == result_parts['season']  # Numbers are case-insensitive
+                season_match = release_parts['season'] == result_parts['season']
                 resolution_match = release_parts['resolution'].lower() == result_parts['resolution'].lower()
 
-                # Handle multiple groups and case-insensitive matching
                 result_groups = {g.lower() for g in result_parts['group'].split('/')}
                 release_groups = {g.lower() for g in release_parts['group'].split('/')}
-                group_match = not result_groups.isdisjoint(release_groups)  # Checks if any group matches
+                group_match = not result_groups.isdisjoint(release_groups)
 
                 if name_match and season_match and resolution_match and group_match:
                     info(f'Release "{name}" found on SF at: {url}')
@@ -125,35 +150,16 @@ def get_sf_download_links(shared_state, url, mirror, title):  # signature must a
                     else:
                         release_url = next(iter(mirrors["season"].values()))
 
-                    real_url = resolve_sf_redirect(release_url, shared_state.values["user_agent"])
-                    return {
-                        "real_url": real_url,
-                        "imdb_id": imdb_id,
-                    }
+                    real_url = resolve_sf_redirect(release_url, user_agent)
+                    if real_url:
+                        # Use the mirror name if we have it, otherwise use "filecrypt"
+                        mirror_name = mirror if mirror else "filecrypt"
+                        return {"links": [[real_url, mirror_name]], "imdb_id": imdb_id}
+                    else:
+                        return {"links": [], "imdb_id": imdb_id}
             except:
                 continue
     except:
         pass
 
-    return {
-        "real_url": None,
-        "imdb_id": None,
-    }
-
-
-def resolve_sf_redirect(url, user_agent):
-    try:
-        response = requests.get(url, allow_redirects=True, timeout=10,
-                                headers={'User-Agent': user_agent})
-        if response.history:
-            for resp in response.history:
-                debug(f"Redirected from {resp.url} to {response.url}")
-            if "/404.html" in response.url:
-                info(f"SF link redirected to 404 page: {response.url}")
-                return None
-            return response.url
-        else:
-            info(f"SF blocked attempt to resolve {url}. Your IP may be banned. Try again later.")
-    except Exception as e:
-        info(f"Error fetching redirected URL for {url}: {e}")
-    return None
+    return {"links": [], "imdb_id": None}

quasarr/downloads/sources/sj.py

@@ -3,5 +3,14 @@
 # Project by https://github.com/rix1337
 
 
-def get_sj_download_links(shared_state, url, mirror, title):  # signature must align with other download link functions!
-    return [url]
+def get_sj_download_links(shared_state, url, mirror, title, password):
+    """
+    KEEP THE SIGNATURE EVEN IF SOME PARAMETERS ARE UNUSED!
+
+    SJ source handler - the site itself acts as a protected crypter.
+    Returns the URL for CAPTCHA solving via userscript.
+    """
+
+    return {
+        "links": [[url, "junkies"]]
+    }

quasarr/downloads/sources/sl.py

@@ -10,10 +10,23 @@ from bs4 import BeautifulSoup
 
 from quasarr.providers.log import info, debug
 
-supported_mirrors = ["nitroflare", "ddownload"]  # ignoring captcha-protected multiup/mirrorace for now
+supported_mirrors = ["nitroflare", "ddownload"]
 
 
-def get_sl_download_links(shared_state, url, mirror, title): # signature must align with other download link functions!
+def derive_mirror_from_host(host):
+    """Get mirror name from hostname."""
+    for m in supported_mirrors:
+        if host.startswith(m + "."):
+            return m
+    return host.split('.')[0] if host else "unknown"
+
+
+def get_sl_download_links(shared_state, url, mirror, title, password):
+    """
+    KEEP THE SIGNATURE EVEN IF SOME PARAMETERS ARE UNUSED!
+
+    SL source handler - returns plain download links.
+    """
     headers = {"User-Agent": shared_state.values["user_agent"]}
     session = requests.Session()
 
@@ -24,9 +37,8 @@ def get_sl_download_links(shared_state, url, mirror, title): # signature must al
         entry = soup.find("div", class_="entry")
         if not entry:
             info(f"Could not find main content section for {title}")
-            return False
+            return {"links": [], "imdb_id": None}
 
-        # extract IMDb id if present
         imdb_id = None
         a_imdb = soup.find("a", href=re.compile(r"imdb\.com/title/tt\d+"))
         if a_imdb:
@@ -50,7 +62,7 @@ def get_sl_download_links(shared_state, url, mirror, title): # signature must al
 
     except Exception as e:
         info(f"SL site has been updated. Grabbing download links for {title} not possible! ({e})")
-        return False
+        return {"links": [], "imdb_id": None}
 
     filtered = []
     for a in anchors:
@@ -59,14 +71,14 @@ def get_sl_download_links(shared_state, url, mirror, title): # signature must al
             continue
 
         host = (urlparse(href).hostname or "").lower()
-        # require host to start with one of supported_mirrors + "."
         if not any(host.startswith(m + ".") for m in supported_mirrors):
             continue
 
         if not mirror or mirror in href:
-            filtered.append(href)
+            mirror_name = derive_mirror_from_host(host)
+            filtered.append([href, mirror_name])
 
-    # regex‐fallback if still empty
+    # regex fallback if still empty
     if not filtered:
         text = "".join(str(x) for x in anchors)
         urls = re.findall(r"https?://[^\s<>'\"]+", text)
@@ -82,7 +94,8 @@ def get_sl_download_links(shared_state, url, mirror, title): # signature must al
                 continue
 
             if not mirror or mirror in u:
-                filtered.append(u)
+                mirror_name = derive_mirror_from_host(host)
+                filtered.append([u, mirror_name])
 
     return {
         "links": filtered,

quasarr/downloads/sources/wd.py

@@ -34,7 +34,13 @@ def resolve_wd_redirect(url, user_agent):
     return None
 
 
-def get_wd_download_links(shared_state, url, mirror, title):  # signature must align with other download link functions!
+def get_wd_download_links(shared_state, url, mirror, title, password):
+    """
+    KEEP THE SIGNATURE EVEN IF SOME PARAMETERS ARE UNUSED!
+
+    WD source handler - resolves redirects and returns protected download links.
+    """
+
     wd = shared_state.values["config"]("Hostnames").get("wd")
     user_agent = shared_state.values["user_agent"]
 
@@ -63,7 +69,7 @@ def get_wd_download_links(shared_state, url, mirror, title):  # signature must a
         )
         if not header:
             info(f"WD Downloads section not found. Grabbing download links for {title} not possible!")
-            return False
+            return {"links": [], "imdb_id": None}
 
         card = header.find_parent("div", class_="card")
         body = card.find("div", class_="card-body")
@@ -72,7 +78,7 @@ def get_wd_download_links(shared_state, url, mirror, title):  # signature must a
         )
     except Exception:
         info(f"WD site has been updated. Grabbing download links for {title} not possible!")
-        return False
+        return {"links": [], "imdb_id": None}
 
     results = []
     try:

quasarr/downloads/sources/wx.py

@@ -11,12 +11,11 @@ from quasarr.providers.log import info, debug
 hostname = "wx"
 
 
-def get_wx_download_links(shared_state, url, mirror, title):
+def get_wx_download_links(shared_state, url, mirror, title, password):
     """
-    Get download links from API based on title and mirror.
+    KEEP THE SIGNATURE EVEN IF SOME PARAMETERS ARE UNUSED!
 
-    Returns:
-        list of [url, hoster] pairs where hoster is the actual mirror (e.g., 'ddownload.com', 'rapidgator.net')
+    WX source handler - Grabs download links from API based on title and mirror.
     """
     host = shared_state.values["config"]("Hostnames").get(hostname)
 
@@ -33,13 +32,13 @@ def get_wx_download_links(shared_state, url, mirror, title):
 
         if response.status_code != 200:
             info(f"{hostname.upper()}: Failed to load page: {url} (Status: {response.status_code})")
-            return []
+            return {"links": []}
 
         # Extract slug from URL
         slug_match = re.search(r'/detail/([^/]+)', url)
         if not slug_match:
             info(f"{hostname.upper()}: Could not extract slug from URL: {url}")
-            return []
+            return {"links": []}
 
         api_url = f'https://api.{host}/start/d/{slug_match.group(1)}'
 
@@ -54,14 +53,14 @@ def get_wx_download_links(shared_state, url, mirror, title):
 
         if api_response.status_code != 200:
             info(f"{hostname.upper()}: Failed to load API: {api_url} (Status: {api_response.status_code})")
-            return []
+            return {"links": []}
 
         data = api_response.json()
 
         # Navigate to releases in the API response
         if 'item' not in data or 'releases' not in data['item']:
             info(f"{hostname.upper()}: No releases found in API response")
-            return []
+            return {"links": []}
 
         releases = data['item']['releases']
 
@@ -74,14 +73,14 @@ def get_wx_download_links(shared_state, url, mirror, title):
 
         if not matching_release:
             info(f"{hostname.upper()}: No release found matching title: {title}")
-            return []
+            return {"links": []}
 
         # Extract crypted_links based on mirror
         crypted_links = matching_release.get('crypted_links', {})
 
         if not crypted_links:
             info(f"{hostname.upper()}: No crypted_links found for: {title}")
-            return []
+            return {"links": []}
 
         links = []
 
@@ -117,11 +116,11 @@ def get_wx_download_links(shared_state, url, mirror, title):
 
         if not links:
             info(f"{hostname.upper()}: No supported crypted links found for: {title}")
-            return []
+            return {"links": []}
 
         debug(f"{hostname.upper()}: Found {len(links)} crypted link(s) for: {title}")
-        return links
+        return {"links": links}
 
     except Exception as e:
         info(f"{hostname.upper()}: Error extracting download links from {url}: {e}")
-        return []
+        return {"links": []}