catocli 2.1.6__tar.gz → 2.1.8__tar.gz

{catocli-2.1.6/catocli.egg-info → catocli-2.1.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: catocli
-Version: 2.1.6
+Version: 2.1.8
 Summary: Cato Networks cli wrapper for the GraphQL API.
 Home-page: https://github.com/Cato-Networks/cato-cli
 Author: Cato Networks

{catocli-2.1.6 → catocli-2.1.8}/catocli/Utils/cliutils.py

@@ -85,10 +85,10 @@ def load_cli_settings():
     settings_locations = [
         # Try package resource first (for installed packages)
         lambda: json.loads(get_package_resource('catocli', 'clisettings.json')),
-        # Try repository location (for development)
-        lambda: json.load(open(os.path.join(os.path.dirname(__file__), '../../clisettings.json'), 'r', encoding='utf-8')),
         # Try adjacent file location (for development - new location)
-        lambda: json.load(open(os.path.join(os.path.dirname(__file__), '../clisettings.json'), 'r', encoding='utf-8'))
+        lambda: json.load(open(os.path.join(os.path.dirname(__file__), '../clisettings.json'), 'r', encoding='utf-8')),
+        # Try repository location (for development - fallback)
+        lambda: json.load(open(os.path.join(os.path.dirname(__file__), '../../clisettings.json'), 'r', encoding='utf-8'))
     ]
     
     for i, load_func in enumerate(settings_locations):
@@ -112,7 +112,9 @@ def get_cli_settings_path():
         str or None: Path to the settings file if found, None otherwise
     """
     possible_paths = [
-        # Repository location (for development)
+        # Adjacent file location (for development - new location)
+        os.path.join(os.path.dirname(__file__), '../clisettings.json'),
+        # Repository location (for development - fallback)
         os.path.join(os.path.dirname(__file__), '../../clisettings.json'),
         # Current directory
         os.path.join(os.getcwd(), 'clisettings.json'),

catocli-2.1.8/catocli/__init__.py

@@ -0,0 +1,2 @@
+__version__ = "2.1.8"
+__cato_host__ = "https://api.catonetworks.com/api/v1/graphql2"

{catocli-2.1.6 → catocli-2.1.8}/catocli/parsers/custom/export_sites/export_sites.py

@@ -237,7 +237,7 @@ def export_socket_site_to_json(args, configuration):
                 nr_relay_group_name = nr_helper_fields.get('XXXXX', None)
                 nr_gateway = nr_helper_fields.get('XXXXX', None)
                 nr_translated_subnet = nr_helper_fields.get('XXXXX', None)
-                # nr_internet_only = nr_helper_fields.get('XXXXX', None)
+                nr_internet_only = nr_helper_fields.get('XXXXX', None)  # Default to None for JSON
                 nr_local_ip = nr_helper_fields.get('XXXXX', None)
                 nr_range_type = nr_helper_fields.get('XXXXX', None)
                 # Adding logic to pre-populate with default value
@@ -271,8 +271,7 @@ def export_socket_site_to_json(args, configuration):
                             cur_range['gateway'] = nr_gateway
                             cur_range['range_type'] = nr_range_type
                             cur_range['translated_subnet'] = nr_translated_subnet
-                            # # Not available to set for native_range via API today
-                            # cur_range['internet_only'] = nr_internet_only
+                            cur_range['internet_only'] = nr_internet_only
                             cur_range['local_ip'] = nr_local_ip  # Use the calculated or original value
                             cur_range['dhcp_settings'] = {
                                 'dhcp_type': nr_dhcp_type,
@@ -291,8 +290,7 @@ def export_socket_site_to_json(args, configuration):
                         site_native_range['gateway'] = nr_gateway
                         site_native_range['range_type'] = nr_range_type
                         site_native_range['translated_subnet'] = nr_translated_subnet
-                        # # Not available to set for native_range via API today
-                        # site_native_range['internet_only'] = nr_internet_only
+                        site_native_range['internet_only'] = nr_internet_only
                         site_native_range['local_ip'] = nr_local_ip
                         site_native_range['dhcp_settings'] = {
                             'dhcp_type': nr_dhcp_type,
@@ -314,8 +312,7 @@ def export_socket_site_to_json(args, configuration):
                         cur_range['gateway'] = nr_gateway
                         cur_range['range_type'] = nr_range_type
                         cur_range['translated_subnet'] = nr_translated_subnet
-                        # # Not available to set for native_range via API today
-                        # cur_range['internet_only'] = nr_internet_only
+                        cur_range['internet_only'] = nr_internet_only
                         cur_range['local_ip'] = nr_local_ip  # Use the calculated or original value
                         cur_range['dhcp_settings'] = {
                             'dhcp_type': nr_dhcp_type,
@@ -594,6 +591,7 @@ def get_processed_site_data(args, configuration):
             nr_relay_group_name = None
             nr_gateway = None
             nr_translated_subnet = None
+            nr_internet_only = None  # Default to None for JSON
             nr_local_ip = None
             nr_range_type = "VLAN" if nr_vlan != None else "Direct"
             
@@ -612,7 +610,7 @@ def get_processed_site_data(args, configuration):
                         cur_range = {
                             'id': range_id, 'name': range_name, 'subnet': nr_subnet, 'vlan': nr_vlan,
                             'mdns_reflector': nr_mdns_reflector, 'gateway': nr_gateway, 'range_type': nr_range_type,
-                            'translated_subnet': nr_translated_subnet, 'local_ip': nr_local_ip,
+                            'translated_subnet': nr_translated_subnet, 'internet_only': nr_internet_only, 'local_ip': nr_local_ip,
                             'dhcp_settings': {
                                 'dhcp_type': nr_dhcp_type, 'ip_range': nr_ip_range, 'relay_group_id': None,
                                 'relay_group_name': nr_relay_group_name, 'dhcp_microsegmentation': nr_dhcp_microsegmentation
@@ -627,6 +625,7 @@ def get_processed_site_data(args, configuration):
                     site_native_range['gateway'] = nr_gateway
                     site_native_range['range_type'] = nr_range_type
                     site_native_range['translated_subnet'] = nr_translated_subnet
+                    site_native_range['internet_only'] = nr_internet_only
                     site_native_range['local_ip'] = nr_local_ip
                     site_native_range['dhcp_settings'] = {
                         'dhcp_type': nr_dhcp_type, 'ip_range': nr_ip_range, 'relay_group_id': None,
@@ -638,7 +637,7 @@ def get_processed_site_data(args, configuration):
                     cur_range = {
                         'id': range_id, 'name': range_name, 'subnet': nr_subnet, 'vlan': nr_vlan,
                         'mdns_reflector': nr_mdns_reflector, 'gateway': nr_gateway, 'range_type': nr_range_type,
-                        'translated_subnet': nr_translated_subnet, 'local_ip': nr_local_ip,
+                        'translated_subnet': nr_translated_subnet, 'internet_only': nr_internet_only, 'local_ip': nr_local_ip,
                         'dhcp_settings': {
                             'dhcp_type': nr_dhcp_type, 'ip_range': nr_ip_range, 'relay_group_id': None,
                             'relay_group_name': nr_relay_group_name, 'dhcp_microsegmentation': nr_dhcp_microsegmentation
@@ -857,7 +856,7 @@ def export_network_ranges_to_csv(site, args, account_id):
         'lan_interface_id', 'lan_interface_name', 'lan_interface_dest_type', 'is_native_range', 'lan_interface_index',
         # Network Range columns (populated on all rows)
         'network_range_id', 'network_range_name', 'subnet', 'vlan', 'mdns_reflector', 
-        'gateway', 'range_type', 'translated_subnet', 'local_ip', 
+        'gateway', 'range_type', 'translated_subnet', 'internet_only', 'local_ip', 
         'dhcp_type', 'dhcp_ip_range', 'dhcp_relay_group_id', 'dhcp_relay_group_name', 'dhcp_microsegmentation'
     ]
     
@@ -911,6 +910,7 @@ def export_network_ranges_to_csv(site, args, account_id):
                         'gateway': network_range.get('gateway', ''),
                         'range_type': network_range.get('range_type', ''),
                         'translated_subnet': network_range.get('translated_subnet', ''),
+                        'internet_only': network_range.get('internet_only', ''),
                         'local_ip': network_range.get('local_ip', ''),
                         'dhcp_type': network_range.get('dhcp_settings', {}).get('dhcp_type', ''),
                         'dhcp_ip_range': network_range.get('dhcp_settings', {}).get('ip_range', ''),
@@ -958,6 +958,7 @@ def export_network_ranges_to_csv(site, args, account_id):
                 'gateway': '',
                 'range_type': '',
                 'translated_subnet': '',
+                'internet_only': '',
                 'local_ip': '',
                 'dhcp_type': '',
                 'dhcp_ip_range': '',
@@ -992,8 +993,9 @@ def export_network_ranges_to_csv(site, args, account_id):
                     'gateway': network_range.get('gateway', ''),
                     'range_type': network_range.get('range_type', ''),
                     'translated_subnet': network_range.get('translated_subnet', ''),
-                        'local_ip': network_range.get('local_ip', ''),
-                        'dhcp_type': network_range.get('dhcp_settings', {}).get('dhcp_type', ''),
+                    'internet_only': network_range.get('internet_only', ''),
+                    'local_ip': network_range.get('local_ip', ''),
+                    'dhcp_type': network_range.get('dhcp_settings', {}).get('dhcp_type', ''),
                     'dhcp_ip_range': network_range.get('dhcp_settings', {}).get('ip_range', ''),
                     'dhcp_relay_group_id': network_range.get('dhcp_settings', {}).get('relay_group_id', ''),
                     'dhcp_relay_group_name': network_range.get('dhcp_settings', {}).get('relay_group_name', ''),

{catocli-2.1.6 → catocli-2.1.8}/catocli/parsers/customParserApiClient.py

@@ -695,23 +695,30 @@ def loadJSON(file):
     Enhanced JSON loading with better error handling and path resolution
     """
     module_dir = os.path.dirname(__file__)
-    # Navigate up two directory levels (from parsers/ to catocli/ to root)
-    module_dir = os.path.dirname(module_dir)  # Go up from parsers/
-    module_dir = os.path.dirname(module_dir)  # Go up from catocli/
     
-    try:
+    # Special handling for clisettings.json - it's in catocli/ directory
+    if file == "clisettings.json":
+        # From parsers/ go up to catocli/ and look for clisettings.json
+        catocli_dir = os.path.dirname(module_dir)  # Go up from parsers/ to catocli/
+        file_path = os.path.join(catocli_dir, file)
+    else:
+        # For other files (like models), navigate up two directory levels (from parsers/ to catocli/ to root)
+        module_dir = os.path.dirname(module_dir)  # Go up from parsers/
+        module_dir = os.path.dirname(module_dir)  # Go up from catocli/
         file_path = os.path.join(module_dir, file)
+    
+    try:
         with open(file_path, 'r') as data:
             config = json.load(data)
             return config
     except FileNotFoundError:
-        logging.error(f"File \"{os.path.join(module_dir, file)}\" not found.")
+        logging.error(f"File \"{file_path}\" not found.")
         raise
     except json.JSONDecodeError as e:
-        logging.error(f"Invalid JSON in file \"{os.path.join(module_dir, file)}\": {e}")
+        logging.error(f"Invalid JSON in file \"{file_path}\": {e}")
         raise
     except Exception as e:
-        logging.error(f"Error loading file \"{os.path.join(module_dir, file)}\": {e}")
+        logging.error(f"Error loading file \"{file_path}\": {e}")
         raise
 
 

{catocli-2.1.6 → catocli-2.1.8/catocli.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: catocli
-Version: 2.1.6
+Version: 2.1.8
 Summary: Cato Networks cli wrapper for the GraphQL API.
 Home-page: https://github.com/Cato-Networks/cato-cli
 Author: Cato Networks

{catocli-2.1.6 → catocli-2.1.8}/schema/catolib.py

@@ -94,7 +94,7 @@ def parseSchema(schema):
     print("  - Loading settings and initializing...")
     
     # Load settings to get childOperationParent and childOperationObjects configuration
-    settings = loadJSON("../clisettings.json")
+    settings = loadJSON("../catocli/clisettings.json")
     childOperationParent = settings.get("childOperationParent", {})
     childOperationObjects = settings.get("childOperationObjects", {})
     
@@ -968,7 +968,7 @@ def writeOperationParsers(catoApiSchema):
     parserMapping = {"query":{},"mutation":{}}
     
     # Load settings to get CSV-supported operations
-    settings = loadJSON("../clisettings.json")
+    settings = loadJSON("../catocli/clisettings.json")
     csv_supported_operations = settings.get("queryOperationCsvOutput", {})
     
     ## Write the raw query parser ##
@@ -1363,7 +1363,7 @@ def renderSubParser(subParser, parentParserPath):
     if not isinstance(subParser, dict):
         return ""
         
-    settings = loadJSON("../clisettings.json")
+    settings = loadJSON("../catocli/clisettings.json")
     csv_supported_operations = settings.get("queryOperationCsvOutput", {})
 
     cliDriverStr = f"""

catocli-2.1.6/catocli/__init__.py

@@ -1,2 +0,0 @@
-__version__ = "2.1.6"
-__cato_host__ = "https://api.catonetworks.com/api/v1/graphql2"