replace click exit with system exit

Author: adhami3310

reflex/custom_components/custom_components.py

@@ -377,13 +377,13 @@ def _get_default_library_name_parts() -> list[str]:
             console.error(
                 f"Based on current directory name {current_dir_name}, the library name is {constants.Reflex.MODULE_NAME}. This package already exists. Please use --library-name to specify a different name."
             )
-            raise click.exceptions.Exit(code=1)
+            raise SystemExit(1)
     if not parts:
         # The folder likely has a name not suitable for python paths.
         console.error(
             f"Could not find a valid library name based on the current directory: got {current_dir_name}."
         )
-        raise click.exceptions.Exit(code=1)
+        raise SystemExit(1)
     return parts
 
 
@@ -419,7 +419,7 @@ def _validate_library_name(library_name: str | None) -> NameVariants:
         console.error(
             f"Please use only alphanumeric characters or dashes: got {library_name}"
         )
-        raise click.exceptions.Exit(code=1)
+        raise SystemExit(1)
 
     # If not specified, use the current directory name to form the module name.
     name_parts = (
@@ -519,7 +519,7 @@ def init(
 
     if CustomComponents.PYPROJECT_TOML.exists():
         console.error(f"A {CustomComponents.PYPROJECT_TOML} already exists. Aborting.")
-        click.exceptions.Exit(code=1)
+        SystemExit(1)
 
     # Show system info.
     exec.output_system_info()
@@ -544,7 +544,7 @@ def init(
         if _pip_install_on_demand(package_name=".", install_args=["-e"]):
             console.info(f"Package {package_name} installed!")
         else:
-            raise click.exceptions.Exit(code=1)
+            raise SystemExit(1)
 
     console.print("[bold]Custom component initialized successfully!")
     console.rule("[bold]Project Summary")
@@ -637,7 +637,7 @@ def _run_build():
     if _run_commands_in_subprocess(cmds):
         console.info("Custom component built successfully!")
     else:
-        raise click.exceptions.Exit(code=1)
+        raise SystemExit(1)
 
 
 @custom_components_cli.command(name="build")
@@ -664,7 +664,7 @@ def _collect_details_for_gallery():
         console.error(
             "Unable to authenticate with Reflex backend services. Make sure you are logged in."
         )
-        raise click.exceptions.Exit(code=1)
+        raise SystemExit(1)
 
     console.rule("[bold]Custom Component Information")
     params = {}
@@ -694,11 +694,11 @@ def _collect_details_for_gallery():
             console.error(
                 f"{package_name} is owned by another user. Unable to update the information for it."
             )
-            raise click.exceptions.Exit(code=1)
+            raise SystemExit(1)
         response.raise_for_status()
     except httpx.HTTPError as he:
         console.error(f"Unable to complete request due to {he}.")
-        raise click.exceptions.Exit(code=1) from he
+        raise SystemExit(1) from he
 
     files = []
     if (image_file_and_extension := _get_file_from_prompt_in_loop()) is not None:
@@ -733,7 +733,7 @@ def _collect_details_for_gallery():
 
     except httpx.HTTPError as he:
         console.error(f"Unable to complete request due to {he}.")
-        raise click.exceptions.Exit(code=1) from he
+        raise SystemExit(1) from he
 
     console.info("Custom component information successfully shared!")
 
@@ -769,7 +769,7 @@ def _get_file_from_prompt_in_loop() -> tuple[bytes, str] | None:
             image_file = image_file_path.read_bytes()
         except OSError as ose:
             console.error(f"Unable to read the {file_extension} file due to {ose}")
-            raise click.exceptions.Exit(code=1) from ose
+            raise SystemExit(1) from ose
         else:
             return image_file, file_extension
 
@@ -795,4 +795,4 @@ def install():
     if _pip_install_on_demand(package_name=".", install_args=["-e"]):
         console.info("Package installed successfully!")
     else:
-        raise click.exceptions.Exit(code=1)
+        raise SystemExit(1)

reflex/reflex.py

@@ -333,7 +333,7 @@ def run(
     """Run the app in the current directory."""
     if frontend_only and backend_only:
         console.error("Cannot use both --frontend-only and --backend-only options.")
-        raise click.exceptions.Exit(1)
+        raise SystemExit(1)
 
     config = get_config()
 

reflex/utils/frontend_skeleton.py

@@ -73,7 +73,7 @@ def initialize_requirements_txt() -> bool:
             continue
         except Exception as e:
             console.error(f"Failed to read {requirements_file_path}.")
-            raise click.exceptions.Exit(1) from e
+            raise SystemExit(1) from e
     else:
         return True
 

reflex/utils/js_runtimes.py

@@ -206,7 +206,7 @@ def download_and_run(url: str, *args, show_status: bool = False, **env):
         console.error(
             f"Failed to download bun install script. You can install or update bun manually from https://bun.com \n{e}"
         )
-        raise click.exceptions.Exit(1) from None
+        raise SystemExit(1) from None
 
     # Save the script to a temporary file.
     with tempfile.NamedTemporaryFile() as tempfile_file:
@@ -237,7 +237,7 @@ def install_bun():
         console.error(
             "REFLEX_USE_NPM is set, but Node.js is not installed. Please install Node.js to use npm."
         )
-        raise click.exceptions.Exit(1)
+        raise SystemExit(1)
 
     bun_path = path_ops.get_bun_path()
 
@@ -304,7 +304,7 @@ def validate_bun(bun_path: Path | None = None):
             console.error(
                 "Failed to obtain bun version. Make sure the specified bun path in your config is correct."
             )
-            raise click.exceptions.Exit(1)
+            raise SystemExit(1)
         if bun_version < version.parse(constants.Bun.MIN_VERSION):
             console.warn(
                 f"Reflex requires bun version {constants.Bun.MIN_VERSION} or higher to run, but the detected version is "
@@ -326,14 +326,14 @@ def validate_frontend_dependencies(init: bool = True):
         try:
             get_js_package_executor(raise_on_none=True)
         except FileNotFoundError as e:
-            raise click.exceptions.Exit(1) from e
+            raise SystemExit(1) from e
 
     if prefer_npm_over_bun() and not check_node_version():
         node_version = get_node_version()
         console.error(
             f"Reflex requires node version {constants.Node.MIN_VERSION} or higher to run, but the detected version is {node_version}",
         )
-        raise click.exceptions.Exit(1)
+        raise SystemExit(1)
 
 
 def remove_existing_bun_installation():

reflex/utils/prerequisites.py

@@ -452,14 +452,14 @@ def validate_app_name(app_name: str | None = None) -> str:
         console.error(
             f"The app directory cannot be named [bold]{constants.Reflex.MODULE_NAME}[/bold]."
         )
-        raise click.exceptions.Exit(1)
+        raise SystemExit(1)
 
     # Make sure the app name is standard for a python package name.
     if not re.match(r"^[a-zA-Z][a-zA-Z0-9_]*$", app_name):
         console.error(
             "The app directory name must start with a letter and can contain letters, numbers, and underscores."
         )
-        raise click.exceptions.Exit(1)
+        raise SystemExit(1)
 
     return app_name
 
@@ -505,7 +505,7 @@ def assert_in_reflex_dir():
         console.error(
             f"[cyan]{constants.Config.FILE}[/cyan] not found. Move to the root folder of your project, or run [bold]{constants.Reflex.MODULE_NAME} init[/bold] to start a new project."
         )
-        raise click.exceptions.Exit(1)
+        raise SystemExit(1)
 
 
 def needs_reinit() -> bool:

reflex/utils/processes.py

@@ -50,7 +50,7 @@ def get_num_workers() -> int:
         redis_client.ping()
     except RedisError as re:
         console.error(f"Unable to connect to Redis: {re}")
-        raise click.exceptions.Exit(1) from re
+        raise SystemExit(1) from re
     return (os.cpu_count() or 1) * 2 + 1
 
 
@@ -146,7 +146,7 @@ def handle_port(service_name: str, port: int, auto_increment: bool) -> int:
             f"Unable to bind to any port for {service_name}. "
             "Please check your network configuration."
         )
-        raise click.exceptions.Exit(1)
+        raise SystemExit(1)
 
     console.debug(
         f"Checking if {service_name.capitalize()} port: {port} is in use for families: {families}."
@@ -172,7 +172,7 @@ def handle_port(service_name: str, port: int, auto_increment: bool) -> int:
     else:
         console.error(f"{service_name.capitalize()} port: {port} is already in use.")
 
-    raise click.exceptions.Exit(1)
+    raise SystemExit(1)
 
 
 @overload
@@ -217,7 +217,7 @@ def new_process(
     non_empty_args = list(filter(None, args)) if isinstance(args, list) else [args]
     if isinstance(args, list) and len(non_empty_args) != len(args):
         console.error(f"Invalid command: {args}")
-        raise click.exceptions.Exit(1)
+        raise SystemExit(1)
 
     path_env: str = os.environ.get("PATH", "")
 
@@ -376,15 +376,15 @@ def stream_logs(
                 "NPM_CONFIG_REGISTRY environment variable. If TLS is the issue, and you know what "
                 "you are doing, you can disable it by setting the SSL_NO_VERIFY environment variable."
             )
-            raise click.exceptions.Exit(1)
+            raise SystemExit(1)
         for set_of_logs in (*prior_logs, tuple(logs)):
             for line in set_of_logs:
                 console.error(line, end="")
             console.error("\n\n")
         if analytics_enabled:
             telemetry.send("error", context=message)
         console.error("Run with [bold]--loglevel debug [/bold] for the full log.")
-        raise click.exceptions.Exit(1)
+        raise SystemExit(1)
 
 
 def show_logs(message: str, process: subprocess.Popen):

reflex/utils/rename.py

@@ -66,15 +66,15 @@ def rename_app(new_app_name: str, loglevel: constants.LogLevel):
         console.error(
             "No rxconfig.py found. Make sure you are in the root directory of your app."
         )
-        raise click.exceptions.Exit(1)
+        raise SystemExit(1)
 
     sys.path.insert(0, str(Path.cwd()))
 
     config = get_config()
     module_path = get_module_path(config.module)
     if module_path is None:
         console.error(f"Could not find module {config.module}.")
-        raise click.exceptions.Exit(1)
+        raise SystemExit(1)
 
     console.info(f"Renaming app directory to {new_app_name}.")
     process_directory(

reflex/utils/templates.py

@@ -61,15 +61,15 @@ def initialize_app_directory(
             console.error(
                 f"Only {template_name=} should be provided, got {template_code_dir_name=}, {template_dir=}."
             )
-            raise click.exceptions.Exit(1)
+            raise SystemExit(1)
         template_code_dir_name = constants.Templates.Dirs.CODE
         template_dir = Path(constants.Templates.Dirs.BASE, "apps", template_name)
     else:
         if template_code_dir_name is None or template_dir is None:
             console.error(
                 f"For `{template_name}` template, `template_code_dir_name` and `template_dir` should both be provided."
             )
-            raise click.exceptions.Exit(1)
+            raise SystemExit(1)
 
     console.debug(f"Using {template_name=} {template_dir=} {template_code_dir_name=}.")
 
@@ -127,7 +127,7 @@ def create_config_init_app_from_remote_template(app_name: str, template_url: str
         temp_dir = tempfile.mkdtemp()
     except OSError as ose:
         console.error(f"Failed to create temp directory for download: {ose}")
-        raise click.exceptions.Exit(1) from ose
+        raise SystemExit(1) from ose
 
     # Use httpx GET with redirects to download the zip file.
     zip_file_path: Path = Path(temp_dir) / "template.zip"
@@ -138,32 +138,32 @@ def create_config_init_app_from_remote_template(app_name: str, template_url: str
         response.raise_for_status()
     except httpx.HTTPError as he:
         console.error(f"Failed to download the template: {he}")
-        raise click.exceptions.Exit(1) from he
+        raise SystemExit(1) from he
     try:
         zip_file_path.write_bytes(response.content)
         console.debug(f"Downloaded the zip to {zip_file_path}")
     except OSError as ose:
         console.error(f"Unable to write the downloaded zip to disk {ose}")
-        raise click.exceptions.Exit(1) from ose
+        raise SystemExit(1) from ose
 
     # Create a temp directory for the zip extraction.
     try:
         unzip_dir = Path(tempfile.mkdtemp())
     except OSError as ose:
         console.error(f"Failed to create temp directory for extracting zip: {ose}")
-        raise click.exceptions.Exit(1) from ose
+        raise SystemExit(1) from ose
 
     try:
         zipfile.ZipFile(zip_file_path).extractall(path=unzip_dir)
         # The zip file downloaded from github looks like:
         # repo-name-branch/**/*, so we need to remove the top level directory.
     except Exception as uze:
         console.error(f"Failed to unzip the template: {uze}")
-        raise click.exceptions.Exit(1) from uze
+        raise SystemExit(1) from uze
 
     if len(subdirs := list(unzip_dir.iterdir())) != 1:
         console.error(f"Expected one directory in the zip, found {subdirs}")
-        raise click.exceptions.Exit(1)
+        raise SystemExit(1)
 
     template_dir = unzip_dir / subdirs[0]
     console.debug(f"Template folder is located at {template_dir}")
@@ -215,7 +215,7 @@ def validate_and_create_app_using_remote_template(
             console.print(
                 f"Please use `reflex login` to access the '{template}' template."
             )
-            raise click.exceptions.Exit(3)
+            raise SystemExit(3)
 
         template_url = templates[template].code_url
     else:
@@ -226,7 +226,7 @@ def validate_and_create_app_using_remote_template(
             template_url = f"https://github.com/{path}/archive/main.zip"
         else:
             console.error(f"Template `{template}` not found or invalid.")
-            raise click.exceptions.Exit(1)
+            raise SystemExit(1)
 
     if template_url is None:
         return
@@ -345,17 +345,17 @@ def prompt_for_template_options(templates: list[Template]) -> str:
 
     if not template:
         console.error("No template selected.")
-        raise click.exceptions.Exit(1)
+        raise SystemExit(1)
 
     try:
         template_index = int(template)
     except ValueError:
         console.error("Invalid template selected.")
-        raise click.exceptions.Exit(1) from None
+        raise SystemExit(1) from None
 
     if template_index < 0 or template_index >= len(templates):
         console.error("Invalid template selected.")
-        raise click.exceptions.Exit(1)
+        raise SystemExit(1)
 
     # Return the template.
     return templates[template_index].name
@@ -393,11 +393,11 @@ def initialize_app(app_name: str, template: str | None = None) -> str | None:
 
         if template == constants.Templates.CHOOSE_TEMPLATES:
             redir.reflex_templates()
-            raise click.exceptions.Exit(0)
+            raise SystemExit(0)
 
     if template == constants.Templates.AI:
         redir.reflex_build_redirect()
-        raise click.exceptions.Exit(0)
+        raise SystemExit(0)
 
     # If the blank template is selected, create a blank app.
     if template == constants.Templates.DEFAULT: