diff --git a/gso/cli/imports.py b/gso/cli/imports.py index 37888fd2a0203f548235be4a581627321a147961..84370b569db1dea831095372568267356a4a968a 100644 --- a/gso/cli/imports.py +++ b/gso/cli/imports.py @@ -540,24 +540,17 @@ def import_opengear(filepath: str = common_filepath_option) -> None: @app.command() -def import_edge_port(filepath: str, output_dir: str | None = None) -> None: +def import_edge_port( + filepath: str = typer.Option(default="data.json", help="Path to the file"), + output_dir: str | None = None, +) -> None: """Import Edge Port into GSO.""" successfully_imported_data = [] data = _read_data(Path(filepath)) - # Save successfully imported data to a file if output_dir is provided - if output_dir: - output_file = Path(output_dir) / "successful_imported_ep.json" - output_file.parent.mkdir(parents=True, exist_ok=True) # Ensure the directory exists - - # Load existing data if the file exists - if output_file.exists(): - with output_file.open("r", encoding="utf-8") as f: - existing_data = json.load(f) - else: - existing_data = {} + ep_map = {} for edge_port in data: - typer.echo(f"Importing Edge Port {edge_port["name"]} on {edge_port["node"]}. ") + typer.echo(f"Importing Edge Port {edge_port["name"]} on {edge_port["node"]}.") try: key = f"{edge_port["name"]}@{edge_port["node"]}" edge_port["node"] = _get_router_subscription_id(edge_port["node"]) @@ -565,7 +558,7 @@ def import_edge_port(filepath: str, output_dir: str | None = None) -> None: start_process("create_imported_edge_port", [initial_data.model_dump()]) successfully_imported_data.append(edge_port["name"]) typer.echo(f"Successfully imported Edge Port {edge_port["name"]} on {edge_port["node"]}.") - existing_data[key] = edge_port["node"] + ep_map[key] = edge_port["node"] except ValidationError as e: typer.echo(f"Validation error: {e}") @@ -584,9 +577,22 @@ def import_edge_port(filepath: str, output_dir: str | None = None) -> None: for item in successfully_imported_data: typer.echo(f"- {item}") + if output_dir: + output_file = Path(output_dir) / "successful_imported_ep.json" + output_file.parent.mkdir(parents=True, exist_ok=True) # Ensure the directory exists + + # Load existing data if the file exists + existing_data = {} + if output_file.exists(): + with output_file.open("r", encoding="utf-8") as f: + existing_data = json.load(f) + + # Merge existing data with ep_map + ep_map = {**existing_data, **ep_map} + # Write the updated map back to the file with output_file.open("w", encoding="utf-8") as f: - json.dump(existing_data, f, indent=4) + json.dump(ep_map, f, indent=4) typer.echo(f"Successfully imported edge ports saved to {output_file}")