flwr 1.25.0__py3-none-any.whl → 1.26.0__py3-none-any.whl

flwr/cli/login/login.py

@@ -15,20 +15,14 @@
 """Flower command line interface `login` command."""
 
 
-from pathlib import Path
-from typing import Annotated
+from typing import Annotated, cast
 
+import click
 import typer
 
 from flwr.cli.auth_plugin import LoginError, NoOpCliAuthPlugin
-from flwr.cli.config_utils import (
-    exit_if_no_address,
-    get_insecure_flag,
-    load_and_validate,
-    process_loaded_project_config,
-    validate_federation_in_project_config,
-)
 from flwr.cli.constant import FEDERATION_CONFIG_HELP_MESSAGE
+from flwr.cli.utils import init_channel_from_connection
 from flwr.common.typing import AccountAuthLoginDetails
 from flwr.proto.control_pb2 import (  # pylint: disable=E0611
     GetLoginDetailsRequest,
@@ -36,68 +30,45 @@ from flwr.proto.control_pb2 import (  # pylint: disable=E0611
 )
 from flwr.proto.control_pb2_grpc import ControlStub
 
-from ..utils import (
-    account_auth_enabled,
-    flwr_cli_grpc_exc_handler,
-    init_channel,
-    load_cli_auth_plugin,
-)
+from ..config_migration import migrate, warn_if_federation_config_overrides
+from ..flower_config import read_superlink_connection
+from ..utils import flwr_cli_grpc_exc_handler, load_cli_auth_plugin_from_connection
 
 
-def login(  # pylint: disable=R0914
-    app: Annotated[
-        Path,
-        typer.Argument(help="Path of the Flower App to run."),
-    ] = Path("."),
-    federation: Annotated[
+def login(
+    ctx: typer.Context,
+    superlink: Annotated[
         str | None,
-        typer.Argument(help="Name of the federation to login into."),
+        typer.Argument(help="Name of the SuperLink connection."),
     ] = None,
     federation_config_overrides: Annotated[
         list[str] | None,
         typer.Option(
             "--federation-config",
             help=FEDERATION_CONFIG_HELP_MESSAGE,
+            hidden=True,
         ),
     ] = None,
 ) -> None:
     """Login to Flower SuperLink."""
-    typer.secho("Loading project configuration... ", fg=typer.colors.BLUE)
+    # Warn `--federation-config` is ignored
+    warn_if_federation_config_overrides(federation_config_overrides)
 
-    pyproject_path = app / "pyproject.toml" if app else None
-    config, errors, warnings = load_and_validate(pyproject_path, check_module=False)
+    # Migrate legacy usage if any
+    migrate(superlink, args=ctx.args)
 
-    config = process_loaded_project_config(config, errors, warnings)
-    federation, federation_config = validate_federation_in_project_config(
-        federation, config, federation_config_overrides
-    )
-    exit_if_no_address(federation_config, "login")
+    # Read superlink connection configuration
+    superlink_connection = read_superlink_connection(superlink)
+    superlink = superlink_connection.name
 
-    # Check if `enable-account-auth` is set to `true`
-
-    if not account_auth_enabled(federation_config):
-        typer.secho(
-            "❌ Account authentication is not enabled for the federation "
-            f"'{federation}'. To enable it, set `enable-account-auth = true` "
-            "in the federation configuration.",
-            fg=typer.colors.RED,
-            bold=True,
-            err=True,
-        )
-        raise typer.Exit(code=1)
     # Check if insecure flag is set to `True`
-    insecure = get_insecure_flag(federation_config)
-    if insecure:
-        typer.secho(
-            "❌ `flwr login` requires TLS to be enabled. `insecure` must NOT be set to "
-            "`true` in the federation configuration.",
-            fg=typer.colors.RED,
-            bold=True,
-            err=True,
+    if superlink_connection.insecure:
+        raise click.ClickException(
+            "`flwr login` requires TLS to be enabled. `insecure` must NOT be set to "
+            "`true` in the federation configuration."
         )
-        raise typer.Exit(code=1)
 
-    channel = init_channel(app, federation_config, NoOpCliAuthPlugin(Path()))
+    channel = init_channel_from_connection(superlink_connection, NoOpCliAuthPlugin())
     stub = ControlStub(channel)
 
     login_request = GetLoginDetailsRequest()
@@ -105,8 +76,9 @@ def login(  # pylint: disable=R0914
         login_response: GetLoginDetailsResponse = stub.GetLoginDetails(login_request)
 
     # Get the auth plugin
-    authn_type = login_response.authn_type
-    auth_plugin = load_cli_auth_plugin(app, federation, federation_config, authn_type)
+    authn_plugin = load_cli_auth_plugin_from_connection(
+        cast(str, superlink_connection.address), login_response.authn_type
+    )
 
     # Login
     details = AccountAuthLoginDetails(
@@ -118,20 +90,14 @@ def login(  # pylint: disable=R0914
     )
     try:
         with flwr_cli_grpc_exc_handler():
-            credentials = auth_plugin.login(details, stub)
+            credentials = authn_plugin.login(details, stub)
         typer.secho(
             "✅ Login successful.",
             fg=typer.colors.GREEN,
             bold=False,
         )
     except LoginError as e:
-        typer.secho(
-            f"❌ Login failed: {e.message}",
-            fg=typer.colors.RED,
-            bold=True,
-            err=True,
-        )
-        raise typer.Exit(code=1) from None
+        raise click.ClickException(f"Login failed: {e.message}") from None
 
     # Store the tokens
-    auth_plugin.store_tokens(credentials)
+    authn_plugin.store_tokens(credentials)

flwr/cli/ls.py

@@ -15,25 +15,18 @@
 """Flower command line interface `ls` command."""
 
 
-import io
 import json
-from pathlib import Path
-from typing import Annotated, cast
+from typing import Annotated
 
 import typer
 from rich.console import Console
 from rich.table import Table
 from rich.text import Text
 
-from flwr.cli.config_utils import (
-    exit_if_no_address,
-    load_and_validate,
-    process_loaded_project_config,
-    validate_federation_in_project_config,
-)
+from flwr.cli.config_migration import migrate, warn_if_federation_config_overrides
 from flwr.cli.constant import FEDERATION_CONFIG_HELP_MESSAGE
-from flwr.common.constant import FAB_CONFIG_FILE, CliOutputFormat, Status, SubStatus
-from flwr.common.logger import print_json_error, redirect_output, restore_output
+from flwr.cli.flower_config import read_superlink_connection
+from flwr.common.constant import CliOutputFormat, Status, SubStatus
 from flwr.common.serde import run_from_proto
 from flwr.proto.control_pb2 import (  # pylint: disable=E0611
     ListRunsRequest,
@@ -43,24 +36,26 @@ from flwr.proto.control_pb2_grpc import ControlStub
 from flwr.supercore.utils import humanize_bytes, humanize_duration
 
 from .run_utils import RunRow, format_runs
-from .utils import flwr_cli_grpc_exc_handler, init_channel, load_cli_auth_plugin
+from .utils import (
+    cli_output_handler,
+    flwr_cli_grpc_exc_handler,
+    init_channel_from_connection,
+    print_json_to_stdout,
+)
 
 
 def ls(  # pylint: disable=too-many-locals, too-many-branches, R0913, R0917
     ctx: typer.Context,
-    app: Annotated[
-        Path,
-        typer.Argument(help="Path of the Flower project"),
-    ] = Path("."),
-    federation: Annotated[
+    superlink: Annotated[
         str | None,
-        typer.Argument(help="Name of the federation"),
+        typer.Argument(help="Name of the SuperLink connection."),
     ] = None,
     federation_config_overrides: Annotated[
         list[str] | None,
         typer.Option(
             "--federation-config",
             help=FEDERATION_CONFIG_HELP_MESSAGE,
+            hidden=True,
         ),
     ] = None,
     runs: Annotated[
@@ -86,7 +81,7 @@ def ls(  # pylint: disable=too-many-locals, too-many-branches, R0913, R0917
         ),
     ] = CliOutputFormat.DEFAULT,
 ) -> None:
-    """List the details of one provided run ID or all runs in a Flower federation.
+    """List the details of one provided run ID or all runs (alias: ls).
 
     The following details are displayed:
 
@@ -99,32 +94,23 @@ def ls(  # pylint: disable=too-many-locals, too-many-branches, R0913, R0917
 
     All timestamps follow ISO 8601, UTC and are formatted as ``YYYY-MM-DD HH:MM:SSZ``.
     """
-    # Resolve command used (list or ls)
-    command_name = cast(str, ctx.command.name) if ctx.command else "list"
-
-    suppress_output = output_format == CliOutputFormat.JSON
-    captured_output = io.StringIO()
-    try:
-        if suppress_output:
-            redirect_output(captured_output)
-        # Load and validate federation config
-        typer.secho("Loading project configuration... ", fg=typer.colors.BLUE)
-
-        pyproject_path = app / FAB_CONFIG_FILE if app else None
-        config, errors, warnings = load_and_validate(pyproject_path, check_module=False)
-        config = process_loaded_project_config(config, errors, warnings)
-        federation, federation_config = validate_federation_in_project_config(
-            federation, config, federation_config_overrides
-        )
-        exit_if_no_address(federation_config, command_name)
+    with cli_output_handler(output_format=output_format) as is_json:
+        # Warn `--federation-config` is ignored
+        warn_if_federation_config_overrides(federation_config_overrides)
+
+        # Migrate legacy usage if any
+        migrate(superlink, args=ctx.args)
+
+        # Read superlink connection configuration
+        superlink_connection = read_superlink_connection(superlink)
         channel = None
+
         try:
             if runs and run_id is not None:
                 raise ValueError(
                     "The options '--runs' and '--run-id' are mutually exclusive."
                 )
-            auth_plugin = load_cli_auth_plugin(app, federation, federation_config)
-            channel = init_channel(app, federation_config, auth_plugin)
+            channel = init_channel_from_connection(superlink_connection)
             stub = ControlStub(channel)
 
             # Display information about a specific run ID
@@ -135,9 +121,9 @@ def ls(  # pylint: disable=too-many-locals, too-many-branches, R0913, R0917
             else:
                 typer.echo("📄 Listing all runs...")
                 formatted_runs = _list_runs(stub)
-            restore_output()
-            if output_format == CliOutputFormat.JSON:
-                Console().print_json(_to_json(formatted_runs))
+
+            if is_json:
+                print_json_to_stdout(_to_json(formatted_runs))
             else:
                 if run_id is not None:
                     Console().print(_to_detail_table(formatted_runs[0]))
@@ -146,22 +132,6 @@ def ls(  # pylint: disable=too-many-locals, too-many-branches, R0913, R0917
         finally:
             if channel:
                 channel.close()
-    except (typer.Exit, Exception) as err:  # pylint: disable=broad-except
-        if suppress_output:
-            restore_output()
-            e_message = captured_output.getvalue()
-            print_json_error(e_message, err)
-        else:
-            typer.secho(
-                f"{err}",
-                fg=typer.colors.RED,
-                bold=True,
-                err=True,
-            )
-    finally:
-        if suppress_output:
-            restore_output()
-        captured_output.close()
 
 
 def _get_status_style(status_text: str) -> str:

flwr/cli/new/new.py

@@ -20,6 +20,7 @@ import zipfile
 from pathlib import Path
 from typing import Annotated, cast
 
+import click
 import requests
 import typer
 
@@ -52,16 +53,12 @@ def new(
 ) -> None:
     """Create new Flower App."""
     if framework is not None or username is not None:
-        typer.secho(
-            "❌ The --framework and --username options are deprecated and will be "
+        raise click.ClickException(
+            "The --framework and --username options are deprecated and will be "
             "removed in future versions of Flower. Please provide an app specifier "
             "after `flwr new` instead, e.g., '@account_name/app_name' or "
-            "'@account_name/app_name==x.y.z'.",
-            fg=typer.colors.RED,
-            bold=True,
-            err=True,
+            "'@account_name/app_name==x.y.z'."
         )
-        raise typer.Exit(code=1)
 
     if app_spec is None:
         # Fetch recommended apps
@@ -141,12 +138,7 @@ def fetch_recommended_apps() -> list[dict[str, str]]:
         return cast(list[dict[str, str]], apps)
 
     except requests.RequestException as e:
-        typer.secho(
-            f"❌ Failed to fetch recommended apps: {e}",
-            fg=typer.colors.RED,
-            err=True,
-        )
-        raise typer.Exit(code=1) from e
+        raise click.ClickException(f"Failed to fetch recommended apps: {e}") from e
 
 
 # Security: prevent zip-slip
@@ -190,22 +182,12 @@ def _download_zip_to_memory(presigned_url: str) -> io.BytesIO:
         r = requests.get(presigned_url, timeout=60)
         r.raise_for_status()
     except requests.RequestException as e:
-        typer.secho(
-            f"ZIP download failed: {e}",
-            fg=typer.colors.RED,
-            err=True,
-        )
-        raise typer.Exit(code=1) from e
+        raise click.ClickException(f"ZIP download failed: {e}") from e
 
     buf = io.BytesIO(r.content)
     # Validate it's a zip
     if not zipfile.is_zipfile(buf):
-        typer.secho(
-            "Downloaded file is not a valid ZIP",
-            fg=typer.colors.RED,
-            err=True,
-        )
-        raise typer.Exit(code=1)
+        raise click.ClickException("Downloaded file is not a valid ZIP")
     buf.seek(0)
     return buf
 
@@ -216,8 +198,7 @@ def download_remote_app_via_api(app_spec: str) -> None:
     try:
         app_id, app_version = parse_app_spec(app_spec)
     except ValueError as e:
-        typer.secho(f"❌ {e}", fg=typer.colors.RED, err=True)
-        raise typer.Exit(code=1) from e
+        raise click.ClickException(str(e)) from e
 
     app_name = app_id.split("/")[1]
 
@@ -242,8 +223,7 @@ def download_remote_app_via_api(app_spec: str) -> None:
     try:
         presigned_url, _ = request_download_link(app_id, app_version, url, "zip_url")
     except ValueError as e:
-        typer.secho(f"❌ {e}", fg=typer.colors.RED, err=True)
-        raise typer.Exit(code=1) from e
+        raise click.ClickException(str(e)) from e
 
     typer.secho(
         "🔽 Downloading ZIP into memory...",

flwr/cli/pull.py

@@ -15,49 +15,39 @@
 """Flower command line interface `pull` command."""
 
 
-from pathlib import Path
 from typing import Annotated
 
+import click
 import typer
 
-from flwr.cli.config_utils import (
-    exit_if_no_address,
-    load_and_validate,
-    process_loaded_project_config,
-    validate_federation_in_project_config,
-)
+from flwr.cli.config_migration import migrate, warn_if_federation_config_overrides
 from flwr.cli.constant import FEDERATION_CONFIG_HELP_MESSAGE
-from flwr.common.constant import FAB_CONFIG_FILE
+from flwr.cli.flower_config import read_superlink_connection
 from flwr.proto.control_pb2 import (  # pylint: disable=E0611
     PullArtifactsRequest,
     PullArtifactsResponse,
 )
 from flwr.proto.control_pb2_grpc import ControlStub
 
-from .utils import flwr_cli_grpc_exc_handler, init_channel, load_cli_auth_plugin
+from .utils import flwr_cli_grpc_exc_handler, init_channel_from_connection
 
 
 def pull(  # pylint: disable=R0914
+    ctx: typer.Context,
     run_id: Annotated[
         int,
-        typer.Option(
-            "--run-id",
-            help="Run ID to pull artifacts from.",
-        ),
+        typer.Argument(help="Run ID to pull artifacts from."),
     ],
-    app: Annotated[
-        Path,
-        typer.Argument(help="Path of the Flower App to run."),
-    ] = Path("."),
-    federation: Annotated[
+    superlink: Annotated[
         str | None,
-        typer.Argument(help="Name of the federation."),
+        typer.Argument(help="Name of the SuperLink connection."),
     ] = None,
     federation_config_overrides: Annotated[
         list[str] | None,
         typer.Option(
             "--federation-config",
             help=FEDERATION_CONFIG_HELP_MESSAGE,
+            hidden=True,
         ),
     ] = None,
 ) -> None:
@@ -66,20 +56,17 @@ def pull(  # pylint: disable=R0914
     Retrieve a download URL for artifacts generated during a completed Flower run. The
     artifacts can then be downloaded from the provided URL.
     """
-    typer.secho("Loading project configuration... ", fg=typer.colors.BLUE)
+    # Warn `--federation-config` is ignored
+    warn_if_federation_config_overrides(federation_config_overrides)
 
-    pyproject_path = app / FAB_CONFIG_FILE if app else None
-    config, errors, warnings = load_and_validate(pyproject_path, check_module=False)
-    config = process_loaded_project_config(config, errors, warnings)
-    federation, federation_config = validate_federation_in_project_config(
-        federation, config, federation_config_overrides
-    )
-    exit_if_no_address(federation_config, "pull")
+    # Migrate legacy usage if any
+    migrate(superlink, args=ctx.args)
+
+    # Read superlink connection configuration
+    superlink_connection = read_superlink_connection(superlink)
     channel = None
     try:
-
-        auth_plugin = load_cli_auth_plugin(app, federation, federation_config)
-        channel = init_channel(app, federation_config, auth_plugin)
+        channel = init_channel_from_connection(superlink_connection)
         stub = ControlStub(channel)
         with flwr_cli_grpc_exc_handler():
             res: PullArtifactsResponse = stub.PullArtifacts(
@@ -87,14 +74,9 @@ def pull(  # pylint: disable=R0914
             )
 
         if not res.url:
-            typer.secho(
-                f"❌ A download URL for artifacts from run {run_id} couldn't be "
-                "obtained.",
-                fg=typer.colors.RED,
-                bold=True,
-                err=True,
+            raise click.ClickException(
+                f"A download URL for artifacts from run {run_id} couldn't be obtained."
             )
-            raise typer.Exit(code=1)
 
         typer.secho(
             f"✅ Artifacts for run {run_id} can be downloaded from: {res.url}",