Kosta Mushkin
107 lines
4.9 KiB
Python
107 lines
4.9 KiB
Python
import json
|
|
import argparse
|
|
import sys
|
|
from typing import Dict, Any, List, Tuple
|
|
|
|
def modify_vpg_settings(json_data: Dict[str, Any], modifications: List[Tuple[str, str, str]]) -> None:
|
|
"""
|
|
Modify VPG settings based on the list of modifications
|
|
Each modification is a tuple of (field_path, new_value, level)
|
|
"""
|
|
for vpg in json_data['ExportedVpgSettingsApi']:
|
|
# Check and modify VPG name
|
|
if 'Basic' in vpg and 'Name' in vpg['Basic']:
|
|
name = vpg['Basic']['Name']
|
|
if not name.startswith('dr-'):
|
|
print(f"Warning: VPG name '{name}' does not start with 'dr-'")
|
|
else:
|
|
vpg['Basic']['Name'] = 'adr-' + name[3:] # Replace 'dr-' with 'adr-'
|
|
|
|
# Handle VPG level modifications
|
|
for field_path, new_value, level in modifications:
|
|
if level == 'vpg':
|
|
if field_path == 'Basic.RecoverySiteIdentifier':
|
|
vpg['Basic']['RecoverySiteIdentifier'] = new_value
|
|
elif field_path == 'Recovery.DefaultHostIdentifier':
|
|
vpg['Recovery']['DefaultHostIdentifier'] = new_value
|
|
elif field_path == 'Networks.Failover.PublicCloud.SubnetIdentifier':
|
|
vpg['Networks']['Failover']['PublicCloud']['SubnetIdentifier'] = new_value
|
|
elif field_path == 'Networks.FailoverTest.PublicCloud.SubnetIdentifier':
|
|
vpg['Networks']['FailoverTest']['PublicCloud']['SubnetIdentifier'] = new_value
|
|
elif field_path == 'Recovery.DefaultDatastoreIdentifier':
|
|
vpg['Recovery']['DefaultDatastoreIdentifier'] = new_value
|
|
|
|
# Handle VM level modifications
|
|
if 'Vms' in vpg:
|
|
for vm in vpg['Vms']:
|
|
for field_path, new_value, level in modifications:
|
|
if level == 'vm':
|
|
if field_path == 'Vms[].Recovery.PublicCloud.Failover.VirtualNetworkIdentifier':
|
|
vm['Recovery']['PublicCloud']['Failover']['VirtualNetworkIdentifier'] = new_value
|
|
elif field_path == 'Vms[].Recovery.PublicCloud.FailoverTest.VirtualNetworkIdentifier':
|
|
vm['Recovery']['PublicCloud']['FailoverTest']['VirtualNetworkIdentifier'] = new_value
|
|
|
|
# Update NIC level SubnetIdentifier
|
|
if 'Nics' in vm:
|
|
for nic in vm['Nics']:
|
|
for field_path, new_value, level in modifications:
|
|
if level == 'vpg':
|
|
if field_path == 'Networks.Failover.PublicCloud.SubnetIdentifier':
|
|
nic['Failover']['PublicCloud']['SubnetIdentifier'] = new_value
|
|
elif field_path == 'Networks.FailoverTest.PublicCloud.SubnetIdentifier':
|
|
nic['FailoverTest']['PublicCloud']['SubnetIdentifier'] = new_value
|
|
|
|
# Remove Preseed field from volumes if it exists
|
|
if 'Volumes' in vm:
|
|
for volume in vm['Volumes']:
|
|
if 'Preseed' in volume:
|
|
del volume['Preseed']
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description='Modify VPG settings in JSON file')
|
|
parser.add_argument('input_file', help='Input JSON file path')
|
|
parser.add_argument('output_file', help='Output JSON file path')
|
|
parser.add_argument('--modifications', nargs='+', required=True,
|
|
help='List of modifications in format: "field_path:new_value:level"')
|
|
|
|
args = parser.parse_args()
|
|
|
|
# Parse modifications
|
|
modifications = []
|
|
for mod in args.modifications:
|
|
try:
|
|
field_path, new_value, level = mod.split(':')
|
|
if level not in ['vpg', 'vm']:
|
|
raise ValueError(f"Invalid level: {level}")
|
|
modifications.append((field_path, new_value, level))
|
|
except ValueError as e:
|
|
print(f"Error parsing modification '{mod}': {str(e)}")
|
|
sys.exit(1)
|
|
|
|
try:
|
|
# Read input JSON file
|
|
with open(args.input_file, 'r') as f:
|
|
json_data = json.load(f)
|
|
|
|
# Modify the JSON data
|
|
modify_vpg_settings(json_data, modifications)
|
|
|
|
# Write output JSON file
|
|
with open(args.output_file, 'w') as f:
|
|
json.dump(json_data, f, indent=2)
|
|
|
|
print(f"Successfully modified {len(modifications)} fields in {args.input_file}")
|
|
print(f"Output written to {args.output_file}")
|
|
|
|
except FileNotFoundError:
|
|
print(f"Error: File {args.input_file} not found")
|
|
sys.exit(1)
|
|
except json.JSONDecodeError:
|
|
print(f"Error: {args.input_file} is not a valid JSON file")
|
|
sys.exit(1)
|
|
except Exception as e:
|
|
print(f"Error: {str(e)}")
|
|
sys.exit(1)
|
|
|
|
if __name__ == "__main__":
|
|
main() |