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

quasarr/__init__.py

@@ -15,13 +15,26 @@ import requests
 import quasarr.providers.web_server
 from quasarr.api import get_api
 from quasarr.providers import shared_state, version
-from quasarr.providers.log import info, debug
+from quasarr.providers.log import debug, info
 from quasarr.providers.notifications import send_discord_message
-from quasarr.providers.utils import extract_allowed_keys, extract_kv_pairs, is_valid_url, check_ip, check_flaresolverr, \
-    validate_address, Unbuffered, FALLBACK_USER_AGENT
+from quasarr.providers.utils import (
+    FALLBACK_USER_AGENT,
+    Unbuffered,
+    check_flaresolverr,
+    check_ip,
+    extract_allowed_keys,
+    extract_kv_pairs,
+    is_valid_url,
+    validate_address,
+)
 from quasarr.storage.config import Config, get_clean_hostnames
-from quasarr.storage.setup import path_config, hostnames_config, hostname_credentials_config, flaresolverr_config, \
-    jdownloader_config
+from quasarr.storage.setup import (
+    flaresolverr_config,
+    hostname_credentials_config,
+    hostnames_config,
+    jdownloader_config,
+    path_config,
+)
 from quasarr.storage.sqlite_database import DataBase
 
 
@@ -33,10 +46,17 @@ def run():
 
         parser = argparse.ArgumentParser()
         parser.add_argument("--port", help="Desired Port, defaults to 8080")
-        parser.add_argument("--internal_address", help="Must be provided when running in Docker")
-        parser.add_argument("--external_address", help="External address for CAPTCHA notifications")
+        parser.add_argument(
+            "--internal_address", help="Must be provided when running in Docker"
+        )
+        parser.add_argument(
+            "--external_address", help="External address for CAPTCHA notifications"
+        )
         parser.add_argument("--discord", help="Discord Webhook URL")
-        parser.add_argument("--hostnames", help="Public HTTP(s) Link that contains hostnames definition.")
+        parser.add_argument(
+            "--hostnames",
+            help="Public HTTP(s) Link that contains hostnames definition.",
+        )
         arguments = parser.parse_args()
 
         sys.stdout = Unbuffered(sys.stdout)
@@ -47,25 +67,31 @@ def run():
 └────────────────────────────────────┘""")
 
         print("\n===== Recommended Services =====")
-        print('For convenient universal premium downloads use: "https://linksnappy.com/?ref=397097"')
         print(
-            'Sponsors get automated CAPTCHA solutions: "https://github.com/rix1337/Quasarr?tab=readme-ov-file#sponsorshelper"')
+            'For convenient universal premium downloads use: "https://linksnappy.com/?ref=397097"'
+        )
+        print(
+            'Sponsors get automated CAPTCHA solutions: "https://github.com/rix1337/Quasarr?tab=readme-ov-file#sponsorshelper"'
+        )
 
         print("\n===== Startup Info =====")
-        port = int('8080')
+        port = int("8080")
         config_path = ""
-        if os.environ.get('DOCKER'):
+        if os.environ.get("DOCKER"):
             config_path = "/config"
             if not arguments.internal_address:
                 print(
-                    "You must set the INTERNAL_ADDRESS variable to a locally reachable URL, e.g. http://192.168.1.1:8080")
-                print("The local URL will be used by Radarr/Sonarr to connect to Quasarr")
+                    "You must set the INTERNAL_ADDRESS variable to a locally reachable URL, e.g. http://192.168.1.1:8080"
+                )
+                print(
+                    "The local URL will be used by Radarr/Sonarr to connect to Quasarr"
+                )
                 print("Stopping Quasarr...")
                 sys.exit(1)
         else:
             if arguments.port:
                 port = int(arguments.port)
-            internal_address = f'http://{check_ip()}:{port}'
+            internal_address = f"http://{check_ip()}:{port}"
 
         if arguments.internal_address:
             internal_address = arguments.internal_address
@@ -92,14 +118,13 @@ def run():
             temp_file = tempfile.TemporaryFile(dir=config_path)
             temp_file.close()
         except Exception as e:
-            print(f'Could not access "{config_path}": {e}"'
-                  f'Stopping Quasarr...')
+            print(f'Could not access "{config_path}": {e}"Stopping Quasarr...')
             sys.exit(1)
 
         shared_state.set_files(config_path)
         shared_state.update("config", Config)
         shared_state.update("database", DataBase)
-        supported_hostnames = extract_allowed_keys(Config._DEFAULT_CONFIG, 'Hostnames')
+        supported_hostnames = extract_allowed_keys(Config._DEFAULT_CONFIG, "Hostnames")
         shared_state.update("sites", [key.upper() for key in supported_hostnames])
         # Set fallback user agent immediately so it's available while background check runs
         shared_state.update("user_agent", FALLBACK_USER_AGENT)
@@ -117,31 +142,46 @@ def run():
                     print(f"Extracting hostnames from {hostnames_link}...")
                     allowed_keys = supported_hostnames
                     max_keys = len(allowed_keys)
-                    shorthand_list = ', '.join(
-                        [f'"{key}"' for key in allowed_keys[:-1]]) + ' and ' + f'"{allowed_keys[-1]}"'
-                    print(f'There are up to {max_keys} hostnames currently supported: {shorthand_list}')
+                    shorthand_list = (
+                        ", ".join([f'"{key}"' for key in allowed_keys[:-1]])
+                        + " and "
+                        + f'"{allowed_keys[-1]}"'
+                    )
+                    print(
+                        f"There are up to {max_keys} hostnames currently supported: {shorthand_list}"
+                    )
                     data = requests.get(hostnames_link).text
                     results = extract_kv_pairs(data, allowed_keys)
 
                     extracted_hostnames = 0
 
                     if results:
-                        hostnames = Config('Hostnames')
+                        hostnames = Config("Hostnames")
                         for shorthand, hostname in results.items():
-                            domain_check = shared_state.extract_valid_hostname(hostname, shorthand)
-                            valid_domain = domain_check.get('domain', None)
+                            domain_check = shared_state.extract_valid_hostname(
+                                hostname, shorthand
+                            )
+                            valid_domain = domain_check.get("domain", None)
                             if valid_domain:
                                 hostnames.save(shorthand, hostname)
                                 extracted_hostnames += 1
-                                print(f'Hostname for "{shorthand}" successfully set to "{hostname}"')
+                                print(
+                                    f'Hostname for "{shorthand}" successfully set to "{hostname}"'
+                                )
                             else:
-                                print(f'Skipping invalid hostname for "{shorthand}" ("{hostname}")')
+                                print(
+                                    f'Skipping invalid hostname for "{shorthand}" ("{hostname}")'
+                                )
                         if extracted_hostnames == max_keys:
-                            print(f'All {max_keys} hostnames successfully extracted!')
-                            print('You can now remove the hostnames link from the command line / environment variable.')
+                            print(f"All {max_keys} hostnames successfully extracted!")
+                            print(
+                                "You can now remove the hostnames link from the command line / environment variable."
+                            )
                     else:
-                        print(f'No Hostnames found at "{hostnames_link}". '
-                              'Ensure to pass a plain hostnames list, not html or json!')
+                        print(
+                            f'No Hostnames found at "{hostnames_link}". '
+                            "Ensure to pass a plain hostnames list, not html or json!"
+                        )
                 else:
                     print(f'Invalid hostnames URL: "{hostnames_link}"')
         except Exception as e:
@@ -151,42 +191,48 @@ def run():
         if not hostnames:
             hostnames_config(shared_state)
             hostnames = get_clean_hostnames(shared_state)
-        print(f"You have [{len(hostnames)} of {len(Config._DEFAULT_CONFIG['Hostnames'])}] supported hostnames set up")
+        print(
+            f"You have [{len(hostnames)} of {len(Config._DEFAULT_CONFIG['Hostnames'])}] supported hostnames set up"
+        )
         print(f"For efficiency it is recommended to set up as few hostnames as needed.")
 
         # Check credentials for login-required hostnames
         skip_login_db = DataBase("skip_login")
-        login_required_sites = ['al', 'dd', 'dl', 'nx']
+        login_required_sites = ["al", "dd", "dl", "nx"]
 
         quasarr.providers.web_server.temp_server_success = False
 
         for site in login_required_sites:
-            hostname = Config('Hostnames').get(site)
+            hostname = Config("Hostnames").get(site)
             if hostname:
                 site_config = Config(site.upper())
-                user = site_config.get('user')
-                password = site_config.get('password')
+                user = site_config.get("user")
+                password = site_config.get("password")
                 if not user or not password:
                     skip_val = skip_login_db.retrieve(site)
                     if skip_val and str(skip_val).lower() == "true":
                         info(f'"{site.upper()}" login skipped by user preference')
                     else:
-                        info(f'"{site.upper()}" credentials missing. Launching setup...')
+                        info(
+                            f'"{site.upper()}" credentials missing. Launching setup...'
+                        )
                         quasarr.providers.web_server.temp_server_success = False
-                        hostname_credentials_config(shared_state, site.upper(), hostname)
+                        hostname_credentials_config(
+                            shared_state, site.upper(), hostname
+                        )
 
         # Check FlareSolverr configuration
         skip_flaresolverr_db = DataBase("skip_flaresolverr")
         flaresolverr_skipped = skip_flaresolverr_db.retrieve("skipped")
-        flaresolverr_url = Config('FlareSolverr').get('url')
+        flaresolverr_url = Config("FlareSolverr").get("url")
 
         if not flaresolverr_url and not flaresolverr_skipped:
             flaresolverr_config(shared_state)
 
-        config = Config('JDownloader')
-        user = config.get('user')
-        password = config.get('password')
-        device = config.get('device')
+        config = Config("JDownloader")
+        user = config.get("user")
+        password = config.get("password")
+        device = config.get("device")
 
         if not user or not password or not device:
             jdownloader_config(shared_state)
@@ -194,7 +240,7 @@ def run():
         print("\n===== Notifications =====")
         discord_url = ""
         if arguments.discord:
-            discord_webhook_pattern = r'^https://discord\.com/api/webhooks/\d+/[\w-]+$'
+            discord_webhook_pattern = r"^https://discord\.com/api/webhooks/\d+/[\w-]+$"
             if re.match(discord_webhook_pattern, arguments.discord):
                 shared_state.update("webhook", arguments.discord)
                 print(f"Using Discord Webhook URL for notifications.")
@@ -206,45 +252,49 @@ def run():
         shared_state.update("discord", discord_url)
 
         print("\n===== API Information =====")
-        api_key = Config('API').get('key')
+        api_key = Config("API").get("key")
         if not api_key:
             api_key = shared_state.generate_api_key()
 
-        print('Setup instructions: "https://github.com/rix1337/Quasarr?tab=readme-ov-file#instructions"')
-        print(f'URL: "{shared_state.values['internal_address']}"')
+        print(
+            'Setup instructions: "https://github.com/rix1337/Quasarr?tab=readme-ov-file#instructions"'
+        )
+        print(f'URL: "{shared_state.values["internal_address"]}"')
         print(f'API Key: "{api_key}" (without quotes)')
 
         if external_address != internal_address:
             print(f'External URL: "{shared_state.values["external_address"]}"')
 
         print("\n===== Quasarr Info Log =====")
-        if os.getenv('DEBUG'):
+        if os.getenv("DEBUG"):
             print("=====    / Debug Log   =====")
 
         protected = shared_state.get_db("protected").retrieve_all_titles()
         if protected:
             package_count = len(protected)
-            info(f'CAPTCHA-Solution required for {package_count} package{'s' if package_count > 1 else ''} at: '
-                 f'"{shared_state.values["external_address"]}/captcha"!')
+            info(
+                f"CAPTCHA-Solution required for {package_count} package{'s' if package_count > 1 else ''} at: "
+                f'"{shared_state.values["external_address"]}/captcha"!'
+            )
 
         flaresolverr = multiprocessing.Process(
             target=flaresolverr_checker,
             args=(shared_state_dict, shared_state_lock),
-            daemon=True
+            daemon=True,
         )
         flaresolverr.start()
 
         jdownloader = multiprocessing.Process(
             target=jdownloader_connection,
             args=(shared_state_dict, shared_state_lock),
-            daemon=True
+            daemon=True,
         )
         jdownloader.start()
 
         updater = multiprocessing.Process(
             target=update_checker,
             args=(shared_state_dict, shared_state_lock),
-            daemon=True
+            daemon=True,
         )
         updater.start()
 
@@ -262,27 +312,31 @@ def flaresolverr_checker(shared_state_dict, shared_state_lock):
         skip_flaresolverr_db = DataBase("skip_flaresolverr")
         flaresolverr_skipped = skip_flaresolverr_db.retrieve("skipped")
 
-        flaresolverr_url = Config('FlareSolverr').get('url')
+        flaresolverr_url = Config("FlareSolverr").get("url")
 
         # If FlareSolverr is not configured and not skipped, it means it's the first run
         # and the user needs to be prompted via the WebUI.
         # This background process should NOT block or prompt the user.
         # It should only check and log the status.
         if not flaresolverr_url and not flaresolverr_skipped:
-            info('FlareSolverr URL not configured. Please configure it via the WebUI.')
-            info('Some sites (AL) will not work without FlareSolverr.')
+            info("FlareSolverr URL not configured. Please configure it via the WebUI.")
+            info("Some sites (AL) will not work without FlareSolverr.")
             return  # Exit the checker, it will be re-checked if user configures it later
 
         if flaresolverr_skipped:
-            info('FlareSolverr setup skipped by user preference')
-            info('Some sites (AL) will not work without FlareSolverr. Configure it later in the web UI.')
+            info("FlareSolverr setup skipped by user preference")
+            info(
+                "Some sites (AL) will not work without FlareSolverr. Configure it later in the web UI."
+            )
         elif flaresolverr_url:
             info(f'Checking FlareSolverr at URL: "{flaresolverr_url}"')
             flaresolverr_check = check_flaresolverr(shared_state, flaresolverr_url)
             if flaresolverr_check:
-                info(f'FlareSolverr connection successful. Using User-Agent: "{shared_state.values["user_agent"]}"')
+                info(
+                    f'FlareSolverr connection successful. Using User-Agent: "{shared_state.values["user_agent"]}"'
+                )
             else:
-                info('FlareSolverr check failed - using fallback user agent')
+                info("FlareSolverr check failed - using fallback user agent")
                 # Fallback user agent is already set in main process, but we log it
                 info(f'User Agent (fallback): "{FALLBACK_USER_AGENT}"')
 
@@ -309,16 +363,18 @@ def update_checker(shared_state_dict, shared_state_lock):
                 info(f'Please manually check: "{link}" for more information!')
                 update_available = None
 
-            if update_available and shared_state.values["last_checked_version"] != update_available:
+            if (
+                update_available
+                and shared_state.values["last_checked_version"] != update_available
+            ):
                 shared_state.update("last_checked_version", update_available)
                 info(message)
                 info(f"Please update to {update_available} as soon as possible!")
                 info(f'Release notes at: "{link}"')
-                update_available = {
-                    "version": update_available,
-                    "link": link
-                }
-                send_discord_message(shared_state, message, "quasarr_update", details=update_available)
+                update_available = {"version": update_available, "link": link}
+                send_discord_message(
+                    shared_state, message, "quasarr_update", details=update_available
+                )
 
             # wait one hour before next check
             time.sleep(60 * 60)
@@ -332,22 +388,30 @@ def jdownloader_connection(shared_state_dict, shared_state_lock):
 
         shared_state.set_device_from_config()
 
-        connection_established = shared_state.get_device() and shared_state.get_device().name
+        connection_established = (
+            shared_state.get_device() and shared_state.get_device().name
+        )
         if not connection_established:
             i = 0
             while i < 10:
                 i += 1
-                info(f'Connection {i} to JDownloader failed. Device name: "{shared_state.values["device"]}"')
+                info(
+                    f'Connection {i} to JDownloader failed. Device name: "{shared_state.values["device"]}"'
+                )
                 time.sleep(60)
                 shared_state.set_device_from_config()
-                connection_established = shared_state.get_device() and shared_state.get_device().name
+                connection_established = (
+                    shared_state.get_device() and shared_state.get_device().name
+                )
                 if connection_established:
                     break
 
         try:
-            info(f'Connection to JDownloader successful. Device name: "{shared_state.get_device().name}"')
+            info(
+                f'Connection to JDownloader successful. Device name: "{shared_state.get_device().name}"'
+            )
         except Exception as e:
-            info(f'Error connecting to JDownloader: {e}! Stopping Quasarr!')
+            info(f"Error connecting to JDownloader: {e}! Stopping Quasarr!")
             sys.exit(1)
 
         try:

quasarr/api/__init__.py

@@ -12,9 +12,13 @@ from quasarr.api.packages import setup_packages_routes
 from quasarr.api.sponsors_helper import setup_sponsors_helper_routes
 from quasarr.api.statistics import setup_statistics
 from quasarr.providers import shared_state
-from quasarr.providers.auth import add_auth_routes, add_auth_hook, show_logout_link
+from quasarr.providers.auth import add_auth_hook, add_auth_routes, show_logout_link
 from quasarr.providers.hostname_issues import get_all_hostname_issues
-from quasarr.providers.html_templates import render_button, render_centered_html, render_success
+from quasarr.providers.html_templates import (
+    render_button,
+    render_centered_html,
+    render_success,
+)
 from quasarr.providers.web_server import Server
 from quasarr.storage.config import Config
 from quasarr.storage.sqlite_database import DataBase
@@ -27,10 +31,11 @@ def get_api(shared_state_dict, shared_state_lock):
 
     # Auth: routes must come first, then hook
     add_auth_routes(app)
-    add_auth_hook(app,
-                  whitelist_prefixes=['/api', '/api/', '/sponsors_helper/', '/download/'],
-                  whitelist_suffixes=['.user.js']
-                  )
+    add_auth_hook(
+        app,
+        whitelist_prefixes=["/api", "/api/", "/sponsors_helper/", "/download/"],
+        whitelist_suffixes=[".user.js"],
+    )
 
     setup_arr_routes(app)
     setup_captcha_routes(app)
@@ -39,10 +44,10 @@ def get_api(shared_state_dict, shared_state_lock):
     setup_sponsors_helper_routes(app)
     setup_packages_routes(app)
 
-    @app.get('/')
+    @app.get("/")
     def index():
         protected = shared_state.get_db("protected").retrieve_all_titles()
-        api_key = Config('API').get('key')
+        api_key = Config("API").get("key")
 
         # Get quick status summary
         try:
@@ -52,10 +57,10 @@ def get_api(shared_state_dict, shared_state_lock):
             jd_connected = False
 
         # Calculate hostname status
-        hostnames_config = Config('Hostnames')
+        hostnames_config = Config("Hostnames")
         skip_login_db = DataBase("skip_login")
         hostname_issues = get_all_hostname_issues()
-        login_required_sites = ['al', 'dd', 'dl', 'nx']
+        login_required_sites = ["al", "dd", "dl", "nx"]
 
         working_count = 0
         total_count = 0
@@ -81,26 +86,30 @@ def get_api(shared_state_dict, shared_state_lock):
 
         # Determine status
         if total_count == 0:
-            hostname_status_class = 'error'
-            hostname_status_emoji = '⚫️'
-            hostname_status_text = 'No hostnames configured'
+            hostname_status_class = "error"
+            hostname_status_emoji = "⚫️"
+            hostname_status_text = "No hostnames configured"
         elif working_count == 0:
-            hostname_status_class = 'error'
-            hostname_status_emoji = '🔴'
-            hostname_status_text = f'0/{total_count} hostnames operational'
+            hostname_status_class = "error"
+            hostname_status_emoji = "🔴"
+            hostname_status_text = f"0/{total_count} hostnames operational"
         elif working_count < total_count:
-            hostname_status_class = 'warning'
-            hostname_status_emoji = '🟡'
-            hostname_status_text = f'{working_count}/{total_count} hostnames operational'
+            hostname_status_class = "warning"
+            hostname_status_emoji = "🟡"
+            hostname_status_text = (
+                f"{working_count}/{total_count} hostnames operational"
+            )
         else:
-            hostname_status_class = 'success'
-            hostname_status_emoji = '🟢'
-            hostname_status_text = f'{working_count}/{total_count} hostnames operational'
+            hostname_status_class = "success"
+            hostname_status_emoji = "🟢"
+            hostname_status_text = (
+                f"{working_count}/{total_count} hostnames operational"
+            )
 
         # CAPTCHA banner
         captcha_hint = ""
         if protected:
-            plural = 's' if len(protected) > 1 else ''
+            plural = "s" if len(protected) > 1 else ""
             captcha_hint = f"""
             <div class="alert alert-warning">
                 <span class="alert-icon">🔒</span>
@@ -109,7 +118,7 @@ def get_api(shared_state_dict, shared_state_lock):
                     {"" if shared_state.values.get("helper_active") else '<br><a href="https://github.com/rix1337/Quasarr?tab=readme-ov-file#sponsorshelper" target="_blank">Sponsors get automated CAPTCHA solutions!</a>'}
                 </div>
                 <div class="alert-action">
-                    {render_button(f"Solve CAPTCHA{plural}", 'primary', {'onclick': "location.href='/captcha'"})}
+                    {render_button(f"Solve CAPTCHA{plural}", "primary", {"onclick": "location.href='/captcha'"})}
                 </div>
             </div>
             """
@@ -117,8 +126,8 @@ def get_api(shared_state_dict, shared_state_lock):
         # Status bars
         status_bars = f"""
             <div class="status-bar">
-                <span class="status-pill {'success' if jd_connected else 'error'}">
-                    {'✅' if jd_connected else '❌'} JDownloader {'connected' if jd_connected else 'disconnected'}
+                <span class="status-pill {"success" if jd_connected else "error"}">
+                    {"✅" if jd_connected else "❌"} JDownloader {"connected" if jd_connected else "disconnected"}
                 </span>
                 <span class="status-pill {hostname_status_class}">
                     {hostname_status_emoji} {hostname_status_text}
@@ -160,7 +169,7 @@ def get_api(shared_state_dict, shared_state_lock):
                     <div class="input-group">
                         <label>URL</label>
                         <div class="input-row">
-                            <input id="urlInput" type="text" readonly value="{shared_state.values['internal_address']}" />
+                            <input id="urlInput" type="text" readonly value="{shared_state.values["internal_address"]}" />
                             <button id="copyUrl" type="button">Copy</button>
                         </div>
                     </div>
@@ -471,12 +480,12 @@ def get_api(shared_state_dict, shared_state_lock):
         </script>
         """
         # Add logout link for form auth
-        logout_html = '<a href="/logout">Logout</a>' if show_logout_link() else ''
+        logout_html = '<a href="/logout">Logout</a>' if show_logout_link() else ""
         return render_centered_html(info, footer_content=logout_html)
 
-    @app.get('/regenerate-api-key')
+    @app.get("/regenerate-api-key")
     def regenerate_api_key():
         shared_state.generate_api_key()
-        return render_success(f'API Key replaced!', 5)
+        return render_success(f"API Key replaced!", 5)
 
-    Server(app, listen='0.0.0.0', port=shared_state.values["port"]).serve_forever()
+    Server(app, listen="0.0.0.0", port=shared_state.values["port"]).serve_forever()