Skip to content
Snippets Groups Projects
Commit 60c89a07 authored by Saket Agrahari's avatar Saket Agrahari
Browse files

fixing error

parent 3334e2aa
No related branches found
No related tags found
1 merge request!421Draft: update to add output to file for import_edge_port
Pipeline #94044 failed
......@@ -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}")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment