quasarr 1.26.7__py3-none-any.whl → 1.28.0__py3-none-any.whl

quasarr/downloads/sources/wd.py

@@ -10,6 +10,7 @@ from bs4 import BeautifulSoup
 
 from quasarr.providers.cloudflare import flaresolverr_get, is_cloudflare_challenge
 from quasarr.providers.log import info, debug
+from quasarr.providers.utils import is_flaresolverr_available
 
 
 def resolve_wd_redirect(url, user_agent):
@@ -47,8 +48,13 @@ def get_wd_download_links(shared_state, url, mirror, title, password):
     try:
         output = requests.get(url)
         if output.status_code == 403 or is_cloudflare_challenge(output.text):
-            info("WD is protected by Cloudflare. Using FlareSolverr to bypass protection.")
-            output = flaresolverr_get(shared_state, url)
+            if is_flaresolverr_available(shared_state):
+                info("WD is protected by Cloudflare. Using FlareSolverr to bypass protection.")
+                output = flaresolverr_get(shared_state, url)
+            else:
+                info("WD is protected by Cloudflare but FlareSolverr is not configured. "
+                     "Please configure FlareSolverr in the web UI to access this site.")
+                return {"links": [], "imdb_id": None}
 
         soup = BeautifulSoup(output.text, "html.parser")
 
@@ -76,6 +82,10 @@ def get_wd_download_links(shared_state, url, mirror, title, password):
         link_tags = body.find_all(
             "a", href=True, class_=lambda c: c and "background-" in c
         )
+    except RuntimeError as e:
+        # Catch FlareSolverr not configured error
+        info(f"WD access failed: {e}")
+        return {"links": [], "imdb_id": None}
     except Exception:
         info(f"WD site has been updated. Grabbing download links for {title} not possible!")
         return {"links": [], "imdb_id": None}

quasarr/providers/cloudflare.py

@@ -5,6 +5,8 @@
 import requests
 from bs4 import BeautifulSoup
 
+from quasarr.providers.utils import is_flaresolverr_available
+
 
 def is_cloudflare_challenge(html: str) -> bool:
     soup = BeautifulSoup(html, "html.parser")
@@ -39,9 +41,14 @@ def update_session_via_flaresolverr(info,
                                     sess,
                                     target_url: str,
                                     timeout: int = 60):
+    # Check if FlareSolverr is available
+    if not is_flaresolverr_available(shared_state):
+        info("FlareSolverr is not configured. Cannot bypass Cloudflare protection.")
+        return False
+
     flaresolverr_url = shared_state.values["config"]('FlareSolverr').get('url')
     if not flaresolverr_url:
-        info("Cannot proceed without FlareSolverr. Please set it up to try again!")
+        info("Cannot proceed without FlareSolverr. Please configure it in the web UI!")
         return False
 
     fs_payload = {
@@ -104,6 +111,12 @@ def ensure_session_cf_bypassed(info, shared_state, session, url, headers):
 
     # If page is protected, try FlareSolverr
     if resp.status_code == 403 or is_cloudflare_challenge(resp.text):
+        # Check if FlareSolverr is available before attempting bypass
+        if not is_flaresolverr_available(shared_state):
+            info("Cloudflare protection detected but FlareSolverr is not configured. "
+                 "Please configure FlareSolverr in the web UI to access this site.")
+            return None, None, None
+
         info("Encountered Cloudflare protection. Solving challenge with FlareSolverr...")
         flaresolverr_result = update_session_via_flaresolverr(info, shared_state, session, url)
         if not flaresolverr_result:
@@ -156,7 +169,13 @@ def flaresolverr_get(shared_state, url, timeout=60):
     """
     Core function for performing a GET request via FlareSolverr only.
     Used internally by FlareSolverrSession.get()
+
+    Returns None if FlareSolverr is not available.
     """
+    # Check if FlareSolverr is available
+    if not is_flaresolverr_available(shared_state):
+        raise RuntimeError("FlareSolverr is not configured. Please configure it in the web UI.")
+
     flaresolverr_url = shared_state.values["config"]('FlareSolverr').get('url')
     if not flaresolverr_url:
         raise RuntimeError("FlareSolverr URL not configured in shared_state.")

quasarr/providers/sessions/al.py

@@ -13,6 +13,18 @@ from bs4 import BeautifulSoup
 from requests.exceptions import Timeout, RequestException
 
 from quasarr.providers.log import info, debug
+from quasarr.providers.utils import is_site_usable, is_flaresolverr_available
+
+
+class SkippedSiteError(Exception):
+    """Raised when a site is skipped due to missing credentials or login being skipped."""
+    pass
+
+
+class FlareSolverrNotAvailableError(Exception):
+    """Raised when FlareSolverr is required but not available."""
+    pass
+
 
 hostname = "al"
 
@@ -20,6 +32,12 @@ SESSION_MAX_AGE_SECONDS = 24 * 60 * 60  # 24 hours
 
 
 def create_and_persist_session(shared_state):
+    # AL requires FlareSolverr - check availability first
+    if not is_flaresolverr_available(shared_state):
+        info(f'"{hostname.upper()}" requires FlareSolverr which is not configured. '
+             f'Please configure FlareSolverr in the web UI to use this site.')
+        return None
+
     cfg = shared_state.values["config"]("Hostnames")
     host = cfg.get(hostname)
     credentials_cfg = shared_state.values["config"](hostname.upper())
@@ -106,6 +124,14 @@ def create_and_persist_session(shared_state):
 
 
 def retrieve_and_validate_session(shared_state):
+    if not is_site_usable(shared_state, hostname):
+        return None
+
+    # AL requires FlareSolverr - check availability
+    if not is_flaresolverr_available(shared_state):
+        info(f'"{hostname.upper()}" requires FlareSolverr which is not configured')
+        return None
+
     db = shared_state.values["database"]("sessions")
     stored = db.retrieve(hostname)
     if not stored:
@@ -213,9 +239,32 @@ def fetch_via_flaresolverr(shared_state,
     – post_data: dict of form‐fields if method=="POST"
     – timeout: seconds (FlareSolverr's internal maxTimeout = timeout*1000 ms)
     """
+    # Check if FlareSolverr is available
+    if not is_flaresolverr_available(shared_state):
+        info(f'"{hostname.upper()}" requires FlareSolverr which is not configured. '
+             f'Please configure FlareSolverr in the web UI.')
+        return {
+            "status_code": None,
+            "headers": {},
+            "json": None,
+            "text": "",
+            "cookies": [],
+            "error": "FlareSolverr is not configured"
+        }
+
     flaresolverr_url = shared_state.values["config"]('FlareSolverr').get('url')
 
     sess = retrieve_and_validate_session(shared_state)
+    if not sess:
+        debug(f"Skipping {hostname}: site not usable (login skipped or no credentials)")
+        return {
+            "status_code": None,
+            "headers": {},
+            "json": None,
+            "text": "",
+            "cookies": [],
+            "error": f"Site '{hostname}' is not usable (login skipped or no credentials)"
+        }
 
     cmd = "request.get" if method.upper() == "GET" else "request.post"
     fs_payload = {
@@ -301,6 +350,8 @@ def fetch_via_requests_session(shared_state, method: str, target_url: str, post_
     – timeout: seconds
     """
     sess = retrieve_and_validate_session(shared_state)
+    if not sess:
+        raise SkippedSiteError(f"{hostname}: site not usable (login skipped or no credentials)")
 
     # Execute request
     if method.upper() == "GET":

quasarr/providers/sessions/dd.py

@@ -7,7 +7,10 @@ import pickle
 
 import requests
 
-from quasarr.providers.log import info
+from quasarr.providers.log import info, debug
+from quasarr.providers.utils import is_site_usable
+
+hostname = "dd"
 
 
 def create_and_persist_session(shared_state):
@@ -62,6 +65,10 @@ def create_and_persist_session(shared_state):
 
 
 def retrieve_and_validate_session(shared_state):
+    if not is_site_usable(shared_state, hostname):
+        debug(f"Skipping {hostname}: site not usable (login skipped or no credentials)")
+        return None
+
     session_string = shared_state.values["database"]("sessions").retrieve("dd")
     if not session_string:
         dd_session = create_and_persist_session(shared_state)

quasarr/providers/sessions/dl.py

@@ -9,6 +9,13 @@ import requests
 from bs4 import BeautifulSoup
 
 from quasarr.providers.log import info, debug
+from quasarr.providers.utils import is_site_usable
+
+
+class SkippedSiteError(Exception):
+    """Raised when a site is skipped due to missing credentials or login being skipped."""
+    pass
+
 
 hostname = "dl"
 
@@ -16,17 +23,17 @@ hostname = "dl"
 def create_and_persist_session(shared_state):
     """
     Create and persist a session using user and password.
-    
+
     Args:
         shared_state: Shared state object
-    
+
     Returns:
         requests.Session or None
     """
     cfg = shared_state.values["config"]("Hostnames")
     host = cfg.get(hostname)
     credentials_cfg = shared_state.values["config"](hostname.upper())
-    
+
     user = credentials_cfg.get("user")
     password = credentials_cfg.get("password")
 
@@ -35,30 +42,30 @@ def create_and_persist_session(shared_state):
         return None
 
     sess = requests.Session()
-    
+
     # Set user agent
     ua = shared_state.values["user_agent"]
     sess.headers.update({'User-Agent': ua})
-    
+
     try:
         # Step 1: Get login page to retrieve CSRF token
         login_page_url = f'https://www.{host}/login/'
         login_page = sess.get(login_page_url, timeout=30)
-        
+
         if login_page.status_code != 200:
             info(f'Failed to load login page for: "{hostname}" - Status {login_page.status_code}')
             return None
-        
+
         # Extract CSRF token from login form
         soup = BeautifulSoup(login_page.text, 'html.parser')
         csrf_input = soup.find('input', {'name': '_xfToken'})
-        
+
         if not csrf_input or not csrf_input.get('value'):
             info(f'Could not find CSRF token on login page for: "{hostname}"')
             return None
-        
+
         csrf_token = csrf_input['value']
-        
+
         # Step 2: Submit login form
         login_data = {
             'login': user,
@@ -67,18 +74,18 @@ def create_and_persist_session(shared_state):
             'remember': '1',
             '_xfRedirect': f'https://www.{host}/'
         }
-        
+
         login_url = f'https://www.{host}/login/login'
         login_response = sess.post(login_url, data=login_data, timeout=30)
-        
+
         # Step 3: Verify login success
         # Check if we're logged in by accessing the main page
         verify_response = sess.get(f'https://www.{host}/', timeout=30)
-        
+
         if 'data-logged-in="true"' not in verify_response.text:
             info(f'Login verification failed for: "{hostname}" - invalid credentials or login failed')
             return None
-        
+
         info(f'Session successfully created for: "{hostname}" using user/password')
     except Exception as e:
         info(f'Failed to create session for: "{hostname}" - {e}')
@@ -88,20 +95,23 @@ def create_and_persist_session(shared_state):
     blob = pickle.dumps(sess)
     token = base64.b64encode(blob).decode("utf-8")
     shared_state.values["database"]("sessions").update_store(hostname, token)
-    
+
     return sess
 
 
 def retrieve_and_validate_session(shared_state):
     """
     Retrieve session from database or create a new one.
-    
+
     Args:
         shared_state: Shared state object
-    
+
     Returns:
         requests.Session or None
     """
+    if not is_site_usable(shared_state, hostname):
+        return None
+
     db = shared_state.values["database"]("sessions")
     token = db.retrieve(hostname)
     if not token:
@@ -122,7 +132,7 @@ def retrieve_and_validate_session(shared_state):
 def invalidate_session(shared_state):
     """
     Invalidate the current session.
-    
+
     Args:
         shared_state: Shared state object
     """
@@ -134,7 +144,7 @@ def invalidate_session(shared_state):
 def _persist_session_to_db(shared_state, sess):
     """
     Serialize & store the given requests.Session into the database under `hostname`.
-    
+
     Args:
         shared_state: Shared state object
         sess: requests.Session to persist
@@ -144,10 +154,11 @@ def _persist_session_to_db(shared_state, sess):
     shared_state.values["database"]("sessions").update_store(hostname, token)
 
 
-def fetch_via_requests_session(shared_state, method: str, target_url: str, post_data: dict = None, get_params: dict = None, timeout: int = 30):
+def fetch_via_requests_session(shared_state, method: str, target_url: str, post_data: dict = None,
+                               get_params: dict = None, timeout: int = 30):
     """
     Execute request using the session.
-    
+
     Args:
         shared_state: Shared state object
         method: "GET" or "POST"
@@ -155,13 +166,13 @@ def fetch_via_requests_session(shared_state, method: str, target_url: str, post_
         post_data: POST data (for POST requests)
         get_params: URL parameters (for GET requests)
         timeout: Request timeout in seconds
-    
+
     Returns:
         Response object
     """
     sess = retrieve_and_validate_session(shared_state)
     if not sess:
-        raise Exception(f"Could not retrieve valid session for {hostname}")
+        raise SkippedSiteError(f"{hostname}: site not usable (login skipped or no credentials)")
 
     # Execute request
     if method.upper() == "GET":

quasarr/providers/sessions/nx.py

@@ -7,7 +7,10 @@ import pickle
 
 import requests
 
-from quasarr.providers.log import info
+from quasarr.providers.log import info, debug
+from quasarr.providers.utils import is_site_usable
+
+hostname = "nx"
 
 
 def create_and_persist_session(shared_state):
@@ -60,6 +63,10 @@ def create_and_persist_session(shared_state):
 
 
 def retrieve_and_validate_session(shared_state):
+    if not is_site_usable(shared_state, hostname):
+        debug(f"Skipping {hostname}: site not usable (login skipped or no credentials)")
+        return None
+
     session_string = shared_state.values["database"]("sessions").retrieve("nx")
     if not session_string:
         nx_session = create_and_persist_session(shared_state)

quasarr/providers/utils.py

@@ -0,0 +1,190 @@
+# -*- coding: utf-8 -*-
+# Quasarr
+# Project by https://github.com/rix1337
+
+import re
+import socket
+import sys
+from urllib.parse import urlparse
+
+import requests
+
+# Fallback user agent when FlareSolverr is not available
+FALLBACK_USER_AGENT = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/143.0.0.0 Safari/537.36"
+
+
+class Unbuffered(object):
+    def __init__(self, stream):
+        self.stream = stream
+
+    def write(self, data):
+        self.stream.write(data)
+        self.stream.flush()
+
+    def writelines(self, datas):
+        self.stream.writelines(datas)
+        self.stream.flush()
+
+    def __getattr__(self, attr):
+        return getattr(self.stream, attr)
+
+
+def is_valid_url(url):
+    """Validate if a URL is properly formatted."""
+    if "/raw/eX4Mpl3" in url:
+        print("Example URL detected. Please provide a valid URL found on pastebin or any other public site!")
+        return False
+
+    parsed = urlparse(url)
+    return parsed.scheme in ("http", "https") and bool(parsed.netloc)
+
+
+def extract_allowed_keys(config, section):
+    """
+    Extracts allowed keys from the specified section in the configuration.
+
+    :param config: The configuration dictionary.
+    :param section: The section from which to extract keys.
+    :return: A list of allowed keys.
+    """
+    if section not in config:
+        raise ValueError(f"Section '{section}' not found in configuration.")
+    return [key for key, *_ in config[section]]
+
+
+def extract_kv_pairs(input_text, allowed_keys):
+    """
+    Extracts key-value pairs from the given text where keys match allowed_keys.
+
+    :param input_text: The input text containing key-value pairs.
+    :param allowed_keys: A list of allowed two-letter shorthand keys.
+    :return: A dictionary of extracted key-value pairs.
+    """
+    kv_pattern = re.compile(rf"^({'|'.join(map(re.escape, allowed_keys))})\s*=\s*(.*)$")
+    kv_pairs = {}
+
+    for line in input_text.splitlines():
+        match = kv_pattern.match(line.strip())
+        if match:
+            key, value = match.groups()
+            kv_pairs[key] = value
+        elif "[Hostnames]" in line:
+            pass
+        else:
+            print(f"Skipping line because it does not contain any supported hostname: {line}")
+
+    return kv_pairs
+
+
+def check_ip():
+    s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+    try:
+        s.connect(('10.255.255.255', 0))
+        ip = s.getsockname()[0]
+    except:
+        ip = '127.0.0.1'
+    finally:
+        s.close()
+    return ip
+
+
+def check_flaresolverr(shared_state, flaresolverr_url):
+    # Ensure it ends with /v<digit+>
+    if not re.search(r"/v\d+$", flaresolverr_url):
+        print(f"FlareSolverr URL does not end with /v#: {flaresolverr_url}")
+        return False
+
+    # Try sending a simple test request
+    headers = {"Content-Type": "application/json"}
+    data = {
+        "cmd": "request.get",
+        "url": "http://www.google.com/",
+        "maxTimeout": 10000
+    }
+
+    try:
+        response = requests.post(flaresolverr_url, headers=headers, json=data, timeout=10)
+        response.raise_for_status()
+        json_data = response.json()
+
+        # Check if the structure looks like a valid FlareSolverr response
+        if "status" in json_data and json_data["status"] == "ok":
+            solution = json_data["solution"]
+            solution_ua = solution.get("userAgent", None)
+            if solution_ua:
+                shared_state.update("user_agent", solution_ua)
+            return True
+        else:
+            print(f"Unexpected FlareSolverr response: {json_data}")
+            return False
+
+    except Exception as e:
+        print(f"Failed to connect to FlareSolverr: {e}")
+        return False
+
+
+def validate_address(address, name):
+    if not address.startswith("http"):
+        sys.exit(f"Error: {name} '{address}' is invalid. It must start with 'http'.")
+
+    colon_count = address.count(":")
+    if colon_count < 1 or colon_count > 2:
+        sys.exit(
+            f"Error: {name} '{address}' is invalid. It must contain 1 or 2 colons, but it has {colon_count}.")
+
+
+def is_flaresolverr_available(shared_state):
+    """
+    Check if FlareSolverr is configured and available.
+
+    Returns:
+        bool: True if FlareSolverr URL is set and not skipped, False otherwise
+    """
+    # Check if FlareSolverr was skipped
+    if shared_state.values["database"]("skip_flaresolverr").retrieve("skipped"):
+        return False
+
+    # Check if FlareSolverr URL is configured
+    flaresolverr_url = shared_state.values["config"]('FlareSolverr').get('url')
+    if not flaresolverr_url:
+        return False
+
+    return True
+
+
+def is_site_usable(shared_state, shorthand):
+    """
+    Check if a site is fully configured and usable.
+
+    For sites that don't require login, just checks if hostname is set.
+    For login-required sites (al, dd, dl, nx), also checks that login wasn't skipped
+    and that credentials exist.
+
+    Args:
+        shared_state: Shared state object
+        shorthand: Site shorthand (e.g., 'al', 'dd', etc.)
+
+    Returns:
+        bool: True if site is usable, False otherwise
+    """
+    shorthand = shorthand.lower()
+
+    # Check if hostname is set
+    hostname = shared_state.values["config"]('Hostnames').get(shorthand)
+    if not hostname:
+        return False
+
+    login_required_sites = ['al', 'dd', 'dl', 'nx']
+    if shorthand not in login_required_sites:
+        return True  # No login needed, hostname is enough
+
+    # Check if login was skipped
+    if shared_state.values["database"]("skip_login").retrieve(shorthand):
+        return False  # Hostname set but login was skipped
+
+    # Check for credentials
+    config = shared_state.values["config"](shorthand.upper())
+    user = config.get('user')
+    password = config.get('password')
+
+    return bool(user and password)

quasarr/providers/version.py

@@ -8,7 +8,7 @@ import requests
 
 
 def get_version():
-    return "1.26.7"
+    return "1.28.0"
 
 
 def get_latest_version():

quasarr/search/sources/al.py

@@ -2,10 +2,8 @@
 # Quasarr
 # Project by https://github.com/rix1337
 
-import re
 import time
 from base64 import urlsafe_b64encode
-from datetime import datetime, timedelta
 from html import unescape
 from urllib.parse import urljoin, quote_plus
 

quasarr/storage/config.py

@@ -25,6 +25,9 @@ class Config(object):
             ("password", "secret", ""),
             ("device", "str", ""),
         ],
+        'Settings': [
+            ("hostnames_url", "secret", ""),
+        ],
         'Hostnames': [
             ("al", "secret", ""),
             ("by", "secret", ""),