added journal modifier
This commit is contained in:
@@ -0,0 +1,107 @@
|
||||
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()
|
||||
Reference in New Issue
Block a user