catocli 2.0.5__py3-none-any.whl → 2.1.0__py3-none-any.whl

build/lib/schema/importSchema.py

@@ -1,60 +0,0 @@
-#!/usr/bin/python
-import catolib
-import logging
-import json
-
-############ ENV Settings ############
-logging.basicConfig(filename="download-schema.log", filemode='w', format='%(name)s - %(levelname)s - %(message)s')
-options = catolib.initParser()
-
-def run():
-	# ########################## OPTION 1 ##############################
-	# ## Uncomment this secion to build cli as one continuous job 
-	# ##################################################################
-	# query = {
-	# 	'query':'query IntrospectionQuery { __schema { description } }',
-	# 	'operationName':'IntrospectionQuery'
-	# }
-	# success,introspection = catolib.send(options.api_key,query)
-	# catolib.parseSchema(introspection)
-	# catolib.writeCliDriver(catolib.catoApiSchema)
-	# catolib.writeOperationParsers(catolib.catoApiSchema)
-	# catolib.writeReadmes(catolib.catoApiSchema)
-	# ##################################################################
-
-	######################### OPTION 2 ##############################
-	## Uncomment to manually run each step manually writing files locally 
-	######################### Step 1 ################################ 
-	## Run intrpsection to pull down schema and write locally
-	################################################################# 
-	query = {
-		'query':'query IntrospectionQuery { __schema { description } }',
-		'operationName':'IntrospectionQuery'
-	}
-	success,resp = catolib.send(options.api_key,query)
-	catolib.writeFile("introspection.json",json.dumps(resp, indent=4, sort_keys=True))
-	# ################################################################# 
-	
-	######################### Step 2 ################################ 
-	## Load from introsspeection from local file, and 
-	## use catolib to parse the schema, write locally 
-	## to catoApiIntrospection.json and catoApiSchema.json
-	################################################################# 
-	introspection = catolib.loadJSON("introspection.json")
-	catolib.parseSchema(introspection)
-	catolib.writeFile("catoApiIntrospection.json",json.dumps(catolib.catoApiIntrospection, indent=4, sort_keys=True))
-	catolib.writeFile("catoApiSchema.json",json.dumps(catolib.catoApiSchema, indent=4, sort_keys=True))
-	#################################################################      
-
-	####################### Step 3 ################################ 
-	## Load catoApiSchema from local file, and write cliDriver
-	## operation parsers, and readmes locally
-	#############################################################
-	catolib.catoApiSchema = catolib.loadJSON("catoApiSchema.json")
-	catolib.writeCliDriver(catolib.catoApiSchema)
-	catolib.writeOperationParsers(catolib.catoApiSchema)
-	catolib.writeReadmes(catolib.catoApiSchema)
-	###############################################################
-
-if __name__ == '__main__':
-	run()

build/lib/schema/remove_policyid.py

@@ -1,89 +0,0 @@
-#!/usr/bin/env python3
-
-import json
-import glob
-import os
-import sys
-
-def remove_policy_id_from_object(obj, path=""):
-    """
-    Recursively remove policyId fields from a JSON object.
-    """
-    if isinstance(obj, dict):
-        # Remove policyId field if it exists
-        if "policyId" in obj:
-            print(f"Removing policyId field at path: {path}")
-            del obj["policyId"]
-        
-        # Recursively process all other fields
-        for key, value in obj.items():
-            new_path = f"{path}.{key}" if path else key
-            remove_policy_id_from_object(value, new_path)
-    
-    elif isinstance(obj, list):
-        # Process each item in the list
-        for i, item in enumerate(obj):
-            new_path = f"{path}[{i}]" if path else f"[{i}]"
-            remove_policy_id_from_object(item, new_path)
-
-def process_file(file_path):
-    """
-    Process a single JSON file to remove policyId fields.
-    """
-    print(f"Processing file: {file_path}")
-    
-    try:
-        # Read the JSON file
-        with open(file_path, 'r') as f:
-            data = json.load(f)
-        
-        # Remove policyId fields
-        remove_policy_id_from_object(data, os.path.basename(file_path))
-        
-        # Write the modified JSON back to the file
-        with open(file_path, 'w') as f:
-            json.dump(data, f, indent=4, sort_keys=True)
-        
-        print(f"Successfully processed: {file_path}")
-        
-    except Exception as e:
-        print(f"Error processing {file_path}: {e}")
-        return False
-    
-    return True
-
-def main():
-    # Get the directory where this script is located
-    script_dir = os.path.dirname(os.path.abspath(__file__))
-    base_dir = os.path.dirname(script_dir)  # Go up one level to cato-cli directory
-    
-    # Find all query.policy*.json files in the models directory
-    models_dir = os.path.join(base_dir, "models")
-    pattern = os.path.join(models_dir, "query.policy*.json")
-    
-    files = glob.glob(pattern)
-    
-    if not files:
-        print("No query.policy*.json files found!")
-        return 1
-    
-    print(f"Found {len(files)} files to process:")
-    for file in files:
-        print(f"  - {os.path.basename(file)}")
-    
-    # Process each file
-    success_count = 0
-    for file_path in files:
-        if process_file(file_path):
-            success_count += 1
-    
-    print(f"\nProcessing complete!")
-    print(f"Successfully processed: {success_count}/{len(files)} files")
-    
-    if success_count != len(files):
-        return 1
-    
-    return 0
-
-if __name__ == "__main__":
-    sys.exit(main())

build/lib/schema/remove_policyid_mutations.py

@@ -1,89 +0,0 @@
-#!/usr/bin/env python3
-
-import json
-import os
-import glob
-
-def remove_policy_id_recursive(obj, path=""):
-    """
-    Recursively remove all 'policyId' keys from a JSON object.
-    Returns the number of removals made.
-    """
-    removals = 0
-    
-    if isinstance(obj, dict):
-        # Create a list of keys to remove to avoid modifying dict during iteration
-        keys_to_remove = []
-        for key in obj.keys():
-            if key == "policyId":
-                keys_to_remove.append(key)
-                current_path = f"{path}.{key}" if path else key
-                print(f"  Removing policyId at: {current_path}")
-                removals += 1
-        
-        # Remove the policyId keys
-        for key in keys_to_remove:
-            del obj[key]
-        
-        # Recursively process remaining values
-        for key, value in obj.items():
-            current_path = f"{path}.{key}" if path else key
-            removals += remove_policy_id_recursive(value, current_path)
-    
-    elif isinstance(obj, list):
-        for i, item in enumerate(obj):
-            current_path = f"{path}[{i}]" if path else f"[{i}]"
-            removals += remove_policy_id_recursive(item, current_path)
-    
-    return removals
-
-def process_mutation_policy_files():
-    """
-    Find and process all mutation.policy*.json files in the models directory.
-    """
-    models_dir = "/Users/briananderson/Documents/Dev/github.com/catonetworks/cato-cli/models"
-    pattern = os.path.join(models_dir, "mutation.policy*.json")
-    
-    files = glob.glob(pattern)
-    
-    if not files:
-        print("No mutation.policy*.json files found!")
-        return
-    
-    print(f"Found {len(files)} mutation.policy*.json files to process:")
-    
-    total_removals = 0
-    processed_files = 0
-    
-    for file_path in sorted(files):
-        filename = os.path.basename(file_path)
-        print(f"\nProcessing: {filename}")
-        
-        try:
-            # Read the JSON file
-            with open(file_path, 'r', encoding='utf-8') as f:
-                data = json.load(f)
-            
-            # Remove policyId fields
-            removals = remove_policy_id_recursive(data, filename)
-            
-            if removals > 0:
-                # Write back the modified JSON
-                with open(file_path, 'w', encoding='utf-8') as f:
-                    json.dump(data, f, indent=4, ensure_ascii=False)
-                
-                print(f"  Removed {removals} policyId field(s)")
-                total_removals += removals
-                processed_files += 1
-            else:
-                print("  No policyId fields found")
-                
-        except Exception as e:
-            print(f"  Error processing {filename}: {e}")
-    
-    print(f"\n=== Summary ===")
-    print(f"Files processed: {processed_files}")
-    print(f"Total policyId fields removed: {total_removals}")
-
-if __name__ == "__main__":
-    process_mutation_policy_files()

build/lib/scripts/catolib.py

@@ -1,62 +0,0 @@
-#!/usr/bin/env python3
-import json
-import os
-import sys
-import subprocess
-from typing import Any, Dict, List, Union
-
-def should_strip_id(obj: Dict[str, Any]) -> bool:
-    if not isinstance(obj, dict):
-        return False
-    
-    keys = set(obj.keys())
-    return keys == {'name', 'id'}
-
-def strip_ids_recursive(obj: Union[Dict, List, Any]) -> Union[Dict, List, Any, None]:
-    if isinstance(obj, dict):
-        new_obj = {}
-        for key, value in obj.items():
-            ## Ignoring sections to retain id and name pair
-            if key == 'sections':
-                new_obj[key] = value
-            else:
-                processed_value = strip_ids_recursive(value)
-                # Skip keys where the value is an empty list
-                if processed_value != []:
-                    new_obj[key] = processed_value
-            # Check if this object should have its id stripped
-            if should_strip_id(new_obj):
-                return {'name': new_obj['name']}
-        return new_obj
-
-    elif isinstance(obj, list):
-        # Recursively process all items in the list
-        processed_list = [strip_ids_recursive(item) for item in obj]
-        # Remove None values from the list (if any were removed)
-        processed_list = [item for item in processed_list if item is not None]
-        return processed_list
-    
-    else:
-        # For primitive types (str, int, bool, None), return as-is
-        return obj
-
-## General purpose functions
-def exec_cli(command):
-    result = None
-    try:
-        response = subprocess.run(command, shell=True, text=True, capture_output=True)
-        if response.returncode != 0:
-            print(f"Command failed with return code {response.returncode}")
-            print(f"stderr: {response.stderr}")
-            return None
-        result = json.loads(response.stdout)
-    except json.JSONDecodeError as e:
-        print(f"Failed to parse JSON response: {e}")
-        print(f"stdout: {response.stdout}")
-        print(f"stderr: {response.stderr}")
-        return None
-    except Exception as e:
-        print(f"Failed to execute command: {e}")
-        return None
-    return result
-

build/lib/scripts/export_if_rules_to_json.py

@@ -1,188 +0,0 @@
-import sys
-import csv
-import subprocess
-import json
-import os
-import argparse
-sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..'))
-from graphql_client.api.call_api import ApiClient, CallApi
-from graphql_client import Configuration
-from graphql_client.api_client import ApiException
-import catocli
-
-# Configuration variables
-DESTINATION_DIR = "config_data"
-
-def strip_ids_recursive(data):
-    """Recursively strip id attributes from data structure"""
-    if isinstance(data, dict):
-        return {k: strip_ids_recursive(v) for k, v in data.items() if k != 'id'}
-    elif isinstance(data, list):
-        return [strip_ids_recursive(item) for item in data]
-    else:
-        return data
-
-def main():
-    # Parse command line arguments
-    parser = argparse.ArgumentParser(description='Export IFW rules to JSON')
-    parser.add_argument('-accountID', help='Account ID to export rules from', required=True)
-    parser.add_argument('--output-file-path', help='Full path including filename and extension for output file. If not specified, uses default: config_data/all_ifw_rules_and_sections_{account_id}.json')
-    parser.add_argument('-v', '--verbose', action='store_true', help='Verbose output')
-    args = parser.parse_args()
-    
-    ACCOUNT_ID = args.accountID
-    
-    # Setup API client configuration
-    if "CATO_TOKEN" not in os.environ:
-        print("Missing authentication, please set the CATO_TOKEN environment variable with your api key.")
-        sys.exit(1)
-    
-    configuration = Configuration()
-    configuration.verify_ssl = False
-    configuration.api_key["x-api-key"] = os.getenv("CATO_TOKEN")
-    configuration.host = "{}".format(catocli.__cato_host__)
-    configuration.accountID = ACCOUNT_ID
-    
-    # Set up output file path
-    if args.output_file_path:
-        # Use output file path if provided
-        output_file = args.output_file_path
-        destination_dir = os.path.dirname(output_file)
-        if args.verbose:
-            print(f"Using output file path: {output_file}")
-    else:
-        # Use default path and filename
-        destination_dir = 'config_data'
-        json_output_file = f"all_ifw_rules_and_sections_{ACCOUNT_ID}.json"
-        output_file = os.path.join(destination_dir, json_output_file)
-        if args.verbose:
-            print(f"Using default path: {output_file}")
-    
-    # Create destination directory if it doesn't exist
-    if destination_dir and not os.path.exists(destination_dir):
-        if args.verbose:
-            print(f"Creating directory: {destination_dir}")
-        os.makedirs(destination_dir)
-    policyQuery = {
-        "query": "query policy ( $accountId:ID! ) { policy ( accountId:$accountId ) { internetFirewall { policy { enabled rules { audit { updatedTime updatedBy } rule { id name description index section { id name } enabled source { ip host { id name } site { id name } subnet ipRange { from to } globalIpRange { id name } networkInterface { id name } siteNetworkSubnet { id name } floatingSubnet { id name } user { id name } usersGroup { id name } group { id name } systemGroup { id name } } connectionOrigin country { id name } device { id name } deviceOS deviceAttributes { category type model manufacturer os osVersion } destination { application { id name } customApp { id name } appCategory { id name } customCategory { id name } sanctionedAppsCategory { id name } country { id name } domain fqdn ip subnet ipRange { from to } globalIpRange { id name } remoteAsn containers { fqdnContainer { id name } ipAddressRangeContainer { id name } } } service { standard { id name } custom { port portRange { from to } protocol } } action tracking { event { enabled } alert { enabled frequency subscriptionGroup { id name } webhook { id name } mailingList { id name } } } schedule { activeOn customTimeframePolicySchedule: customTimeframe { from to } customRecurringPolicySchedule: customRecurring { from to days } } exceptions { name source { ip host { id name } site { id name } subnet ipRange { from to } globalIpRange { id name } networkInterface { id name } siteNetworkSubnet { id name } floatingSubnet { id name } user { id name } usersGroup { id name } group { id name } systemGroup { id name } } deviceOS country { id name } device { id name } deviceAttributes { category type model manufacturer os osVersion } destination { application { id name } customApp { id name } appCategory { id name } customCategory { id name } sanctionedAppsCategory { id name } country { id name } domain fqdn ip subnet ipRange { from to } globalIpRange { id name } remoteAsn containers { fqdnContainer { id name } ipAddressRangeContainer { id name } } } service { standard { id name } custom { port portRangeCustomService: portRange { from to } protocol } } connectionOrigin } } properties } sections { audit { updatedTime updatedBy } section { id name } properties } audit { publishedTime publishedBy } revision { id name description changes createdTime updatedTime } } } } }",
-        "variables": {
-            "accountId": ACCOUNT_ID
-        },
-        "operationName": "policy"
-    }
-    print(f"Retrieving all IFW rules and sections for account {ACCOUNT_ID}...")
-    
-    # Create API client instance
-    instance = CallApi(ApiClient(configuration))
-    
-    # Create params object for the API call
-    params = {
-        'v': False,  # verbose mode
-        'f': 'json',  # format
-        'p': False,  # pretty print
-        't': False   # test mode
-    }
-    
-    try:
-        # Call the API directly
-        response = instance.call_api(policyQuery, params)
-        allIfwRules = response[0] if response else {}
-        
-        if not allIfwRules or 'data' not in allIfwRules:
-            print("ERROR: Failed to retrieve data from API")
-            sys.exit(1)
-            
-    except ApiException as e:
-        print(f"ERROR: API call failed - {e}")
-        sys.exit(1)
-    except Exception as e:
-        print(f"ERROR: Unexpected error - {e}")
-        sys.exit(1)
-    
-    # First, preserve section IDs before stripping them
-    section_id_map = {}
-    sections_with_ids = allIfwRules['data']['policy']['internetFirewall']['policy']['sections']
-    for section_data in sections_with_ids:
-        section_id_map[section_data['section']['name']] = section_data['section']['id']
-    
-    # Get the first section ID for section_to_start_after_id
-    section_to_start_after_id = None
-    if len(sections_with_ids) > 0:
-        section_to_start_after_id = sections_with_ids[0]['section']['id']
-        print(f"Excluding first section with system rule: {sections_with_ids[0]['section']['name']}")
-    
-    ## Processing data to strip id attributes
-    processed_data = strip_ids_recursive(allIfwRules)
-    
-    ## Filter out rules with properties[0]=="SYSTEM"
-    filtered_rules = []
-    for rule_data in processed_data['data']['policy']['internetFirewall']['policy']['rules']:
-        rule_properties = rule_data.get('properties', [])
-        # Skip rules where the first property is "SYSTEM"
-        if rule_properties and rule_properties[0] == "SYSTEM":
-            print(f"Excluding SYSTEM rule: {rule_data['rule']['name']}")
-        else:
-            filtered_rules.append(rule_data)
-    processed_data['data']['policy']['internetFirewall']['policy']['rules'] = filtered_rules
-    # rules = filtered_rules
-    
-    # Add index_in_section to each rule
-    # Group rules by section and add index_in_section
-    section_counters = {}
-    for rule_data in processed_data['data']['policy']['internetFirewall']['policy']['rules']:
-        section_name = rule_data['rule']['section']['name']
-        if section_name not in section_counters:
-            section_counters[section_name] = 0
-        section_counters[section_name] += 1
-        rule_data['rule']['index_in_section'] = section_counters[section_name]
-    
-    # Create rules_in_sections array
-    rules_in_sections = []
-    for rule_data in processed_data['data']['policy']['internetFirewall']['policy']['rules']:
-        rule_info = rule_data['rule']
-        rules_in_sections.append({
-            "index_in_section": rule_info['index_in_section'],
-            "section_name": rule_info['section']['name'],
-            "rule_name": rule_info['name']
-        })
-        rule_info.pop("index_in_section", None)
-        rule_info.pop("index", None)
-        rule_info["enabled"] = True
-
-    # Add rules_in_sections to the policy structure
-    processed_data['data']['policy']['internetFirewall']['policy']['rules_in_sections'] = rules_in_sections
-    
-    # Reformat sections array to have index, section_id and section_name structure
-    # Exclude the first section from export
-    processed_sections = [] 
-    # Add first section containing reserved SYSTEM rules as section_to_start_after_id
-    for index, section_data in enumerate(processed_data['data']['policy']['internetFirewall']['policy']['sections']):
-        if index == 0:
-            # Skip the first section which contains reserved SYSTEM rules
-            continue
-        else:
-            processed_sections.append({
-                "section_index": index,
-                "section_name": section_data['section']['name']
-            })
-    
-    # Add preserved section IDs and section_to_start_after_id
-    processed_data['data']['policy']['internetFirewall']['policy']['section_ids'] = section_id_map
-    if section_to_start_after_id:
-        processed_data['data']['policy']['internetFirewall']['policy']['section_to_start_after_id'] = section_to_start_after_id
-
-    if len(sections_with_ids) > 0:
-        print(f"Excluded first section: {sections_with_ids[0]['section']['name']} containing reserved SYSTEM rules")
-    
-    # Replace the original sections array with the reformatted one
-    processed_data['data']['policy']['internetFirewall']['policy']['sections'] = processed_sections
-        
-    # Write the processed data to the new file
-    with open(output_file, 'w', encoding='utf-8') as f:
-        json.dump(processed_data, f, indent=4, ensure_ascii=False)    
-    print(f"Successfully created {output_file}")
-    
-
-if __name__ == "__main__":
-    main()
-

build/lib/scripts/export_wf_rules_to_json.py

@@ -1,111 +0,0 @@
-import sys
-import csv
-import subprocess
-import json
-import os
-import argparse
-import catolib
-
-# Configuration variables
-DESTINATION_DIR = "config_data"
-
-def main():
-    # Parse command line arguments
-    parser = argparse.ArgumentParser(description='Export WAN rules to JSON')
-    parser.add_argument('account_id', help='Account ID to export rules from')
-    args = parser.parse_args()
-    
-    ACCOUNT_ID = args.account_id
-    JSON_OUTPUT_FILE = f"all_wf_rules_and_sections_{ACCOUNT_ID}.json"
-    # Create destination directory if it doesn't exist
-    if not os.path.exists(DESTINATION_DIR):
-        print(f"Creating directory: {DESTINATION_DIR}")
-        os.makedirs(DESTINATION_DIR)
-    
-    # Set full file paths
-    output_file = os.path.join(DESTINATION_DIR, JSON_OUTPUT_FILE)
-    policyQuery = {
-        "query": "query policy ( $accountId:ID! ) { policy ( accountId:$accountId ) { wanFirewall { policy { enabled rules { audit { updatedTime updatedBy } rule { id name description index section { id name } enabled source { host { id name } site { id name } subnet ip ipRange { from to } globalIpRange { id name } networkInterface { id name } siteNetworkSubnet { id name } floatingSubnet { id name } user { id name } usersGroup { id name } group { id name } systemGroup { id name } } connectionOrigin country { id name } device { id name } deviceOS deviceAttributes { category type model manufacturer os osVersion } destination { host { id name } site { id name } subnet ip ipRange { from to } globalIpRange { id name } networkInterface { id name } siteNetworkSubnet { id name } floatingSubnet { id name } user { id name } usersGroup { id name } group { id name } systemGroup { id name } } application { application { id name } appCategory { id name } customApp { id name } customCategory { id name } sanctionedAppsCategory { id name } domain fqdn ip subnet ipRange { from to } globalIpRange { id name } } service { standard { id name } custom { port portRange { from to } protocol } } action tracking { event { enabled } alert { enabled frequency subscriptionGroup { id name } webhook { id name } mailingList { id name } } } schedule { activeOn customTimeframePolicySchedule: customTimeframe { from to } customRecurringPolicySchedule: customRecurring { from to days } } direction exceptions { name source { host { id name } site { id name } subnet ip ipRange { from to } globalIpRange { id name } networkInterface { id name } siteNetworkSubnet { id name } floatingSubnet { id name } user { id name } usersGroup { id name } group { id name } systemGroup { id name } } deviceOS destination { host { id name } site { id name } subnet ip ipRange { from to } globalIpRange { id name } networkInterface { id name } siteNetworkSubnet { id name } floatingSubnet { id name } user { id name } usersGroup { id name } group { id name } systemGroup { id name } } country { id name } device { id name } deviceAttributes { category type model manufacturer os osVersion } application { application { id name } appCategory { id name } customApp { id name } customCategory { id name } sanctionedAppsCategory { id name } domain fqdn ip subnet ipRange { from to } globalIpRange { id name } } service { standard { id name } custom { port portRangeCustomService: portRange { from to } protocol } } connectionOrigin direction } } properties } sections { audit { updatedTime updatedBy } section { id name } properties } audit { publishedTime publishedBy } revision { id name description changes createdTime updatedTime } } } } }",
-        "variables": {
-            "accountId": ACCOUNT_ID
-        },
-        "operationName": "policy"
-    }
-    catocliCommand = "catocli raw '"+json.dumps(policyQuery)+"'"
-    print(f"Retrieving all WAN rules and sections for account {ACCOUNT_ID}...")
-    allWanRules = catolib.exec_cli(catocliCommand)
-    
-    ## Processing data to strip id attributes
-    processed_data = catolib.strip_ids_recursive(allWanRules)
-
-    ## Preserving section IDs index by section name
-    section_id_map = {}
-    if 'data' in allWanRules and 'policy' in allWanRules['data'] and 'wanFirewall' in allWanRules['data']['policy']:
-        for section_data in allWanRules['data']['policy']['wanFirewall']['policy']['sections']:
-            section_id_map[section_data['section']['name']] = section_data['section']['id']
-    processed_data['data']['policy']['wanFirewall']['policy']['section_ids'] = section_id_map
-    
-    ## Filter out rules with properties[0]=="SYSTEM"
-    filtered_rules = []
-    for rule_data in processed_data['data']['policy']['wanFirewall']['policy']['rules']:
-        rule_properties = rule_data.get('properties', [])
-        # Skip rules where the first property is "SYSTEM"
-        if rule_properties and rule_properties[0] == "SYSTEM":
-            print(f"Excluding SYSTEM rule: {rule_data['rule']['name']}")
-        else:
-            filtered_rules.append(rule_data)
-    processed_data['data']['policy']['wanFirewall']['policy']['rules'] = filtered_rules
-    # rules = filtered_rules
-    
-    # Add index_in_section to each rule
-    # Group rules by section and add index_in_section
-    section_counters = {}
-    for rule_data in processed_data['data']['policy']['wanFirewall']['policy']['rules']:
-        section_name = rule_data['rule']['section']['name']
-        if section_name not in section_counters:
-            section_counters[section_name] = 0
-        section_counters[section_name] += 1
-        rule_data['rule']['index_in_section'] = section_counters[section_name]
-    
-    # Create rules_in_sections array
-    rules_in_sections = []
-    for rule_data in processed_data['data']['policy']['wanFirewall']['policy']['rules']:
-        rule_info = rule_data['rule']
-        rules_in_sections.append({
-            "index_in_section": rule_info['index_in_section'],
-            "section_name": rule_info['section']['name'],
-            "rule_name": rule_info['name']
-        })
-        rule_info.pop("index_in_section", None)
-        rule_info.pop("index", None)
-        rule_info["enabled"] = True
-
-    # Add rules_in_sections to the policy structure
-    processed_data['data']['policy']['wanFirewall']['policy']['rules_in_sections'] = rules_in_sections
-    
-    # Reformat sections array to have index, section_id and section_name structure
-    # Exclude the first section from export, excluding first section
-    
-    processed_sections = []
-    # Add first section containing reserved SYSTEM rules as section_to_start_after_id
-    for index, section_data in enumerate(processed_data['data']['policy']['wanFirewall']['policy']['sections']):
-        processed_sections.append({
-            "section_index": index,
-            "section_name": section_data['section']['name']
-        })
-
-    if len(processed_data['data']['policy']['wanFirewall']['policy']['sections']) > 0:
-        print(f"Excluded first section: {processed_data['data']['policy']['wanFirewall']['policy']['sections'][0]['section']['name']} containing reserved SYSTEM rules")
-    
-    # Replace the original sections array with the reformatted one
-    processed_data['data']['policy']['wanFirewall']['policy']['sections'] = processed_sections
-        
-    # Write the processed data to the new file
-    with open(output_file, 'w', encoding='utf-8') as f:
-        json.dump(processed_data, f, indent=4, ensure_ascii=False)    
-    print(f"Successfully created {output_file}")
-    
-
-if __name__ == "__main__":
-    main()
-