catocli 2.1.2__tar.gz → 2.1.3__tar.gz

{catocli-2.1.2 → catocli-2.1.3}/MANIFEST.in

@@ -2,5 +2,6 @@ include catocli/*/README.md
 include catocli/parsers/*/README.md
 include models/*.json
 include vendor/*
+include clisettings.json
 prune venv/
-prune test/
+prune test/

{catocli-2.1.2/catocli.egg-info → catocli-2.1.3}/PKG-INFO

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

{catocli-2.1.2 → catocli-2.1.3}/catocli/Utils/clidriver.py

@@ -59,8 +59,8 @@ from ..parsers.mutation_admin import mutation_admin_parse
 from ..parsers.mutation_policy import mutation_policy_parse
 from ..parsers.mutation_accountManagement import mutation_accountManagement_parse
 from ..parsers.mutation_sandbox import mutation_sandbox_parse
-from ..parsers.mutation_groups import mutation_groups_parse
 from ..parsers.mutation_hardware import mutation_hardware_parse
+from ..parsers.mutation_groups import mutation_groups_parse
 from ..parsers.mutation_enterpriseDirectory import mutation_enterpriseDirectory_parse
 
 def show_version_info(args, configuration=None):
@@ -85,15 +85,6 @@ def show_version_info(args, configuration=None):
         else:
             print("Unable to check for updates (check your internet connection)")
     return [{"success": True, "current_version": catocli.__version__, "latest_version": latest_version if not args.current_only else None}]
-
-def load_private_settings():
-    # Load private settings from ~/.cato/settings.json
-    settings_file = os.path.expanduser("~/.cato/settings.json")
-    try:
-        with open(settings_file, 'r') as f:
-            return json.load(f)
-    except (FileNotFoundError, json.JSONDecodeError):
-        return {}
         
 def get_configuration(skip_api_key=False):
     configuration = Configuration()
@@ -200,8 +191,8 @@ mutation_admin_parser = mutation_admin_parse(mutation_subparsers)
 mutation_policy_parser = mutation_policy_parse(mutation_subparsers)
 mutation_accountManagement_parser = mutation_accountManagement_parse(mutation_subparsers)
 mutation_sandbox_parser = mutation_sandbox_parse(mutation_subparsers)
-mutation_groups_parser = mutation_groups_parse(mutation_subparsers)
 mutation_hardware_parser = mutation_hardware_parse(mutation_subparsers)
+mutation_groups_parser = mutation_groups_parse(mutation_subparsers)
 mutation_enterpriseDirectory_parser = mutation_enterpriseDirectory_parse(mutation_subparsers)
 
 

catocli-2.1.3/catocli/Utils/cliutils.py

@@ -0,0 +1,178 @@
+#!/usr/bin/env python3
+"""
+General CLI utilities for catocli
+
+This module contains general-purpose utility functions used across the catocli
+package, including settings loading and configuration management.
+"""
+
+import os
+import json
+
+
+# Import for resource handling
+try:
+    # Python 3.9+
+    from importlib.resources import files
+    def get_package_resource(package, filename):
+        return files(package).joinpath(filename).read_text(encoding='utf-8')
+except ImportError:
+    try:
+        # Python 3.7-3.8
+        from importlib.resources import read_text
+        def get_package_resource(package, filename):
+            return read_text(package, filename, encoding='utf-8')
+    except ImportError:
+        try:
+            # Fallback to pkg_resources
+            import pkg_resources
+            def get_package_resource(package, filename):
+                return pkg_resources.resource_string(package, filename).decode('utf-8')
+        except ImportError:
+            # Final fallback - no package resources available
+            def get_package_resource(package, filename):
+                raise ImportError("No resource handling module available")
+
+
+def load_cli_settings():
+    """
+    Load clisettings.json from multiple possible locations:
+    1. Package resource (for installed packages)
+    2. Repository location (for development)
+    
+    Returns:
+        dict: The loaded settings or default settings if loading fails
+    """
+    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'))
+    ]
+    
+    for i, load_func in enumerate(settings_locations):
+        try:
+            settings = load_func()
+            if settings:
+                return settings
+        except (FileNotFoundError, json.JSONDecodeError, ImportError, OSError, ModuleNotFoundError) as e:
+            # Continue to next location
+            continue
+    
+    # If all locations fail, return default settings for socket export
+    return {
+        "export_by_socket_type": {
+            "SOCKET_X1500": True,
+            "SOCKET_X1600": True,
+            "SOCKET_X1600_LTE": True,
+            "SOCKET_X1700": True
+        },
+        "default_socket_interface_map": {
+            "SOCKET_X1500": "LAN1",
+            "SOCKET_X1600": "INT_5",
+            "SOCKET_X1600_LTE": "INT_5",
+            "SOCKET_X1700": "INT_3"
+        }
+    }
+
+
+def get_cli_settings_path():
+    """
+    Get the path to the CLI settings file, trying different locations.
+    
+    Returns:
+        str or None: Path to the settings file if found, None otherwise
+    """
+    possible_paths = [
+        # Repository location (for development)
+        os.path.join(os.path.dirname(__file__), '../../clisettings.json'),
+        # Current directory
+        os.path.join(os.getcwd(), 'clisettings.json'),
+    ]
+    
+    for path in possible_paths:
+        if os.path.exists(path):
+            return path
+    
+    return None
+
+
+def load_json_file(file_path, encoding='utf-8'):
+    """
+    Load a JSON file with error handling.
+    
+    Args:
+        file_path (str): Path to the JSON file
+        encoding (str): File encoding (default: utf-8)
+    
+    Returns:
+        dict or None: The loaded JSON data, or None if loading fails
+    """
+    try:
+        with open(file_path, 'r', encoding=encoding) as f:
+            return json.load(f)
+    except (FileNotFoundError, json.JSONDecodeError, OSError) as e:
+        return None
+
+
+def save_json_file(data, file_path, encoding='utf-8', indent=2):
+    """
+    Save data to a JSON file with error handling.
+    
+    Args:
+        data: Data to save
+        file_path (str): Path to save the JSON file
+        encoding (str): File encoding (default: utf-8)
+        indent (int): JSON indentation (default: 2)
+    
+    Returns:
+        bool: True if successful, False otherwise
+    """
+    try:
+        # Ensure the directory exists
+        os.makedirs(os.path.dirname(file_path), exist_ok=True)
+        
+        with open(file_path, 'w', encoding=encoding) as f:
+            json.dump(data, f, indent=indent, ensure_ascii=False)
+        return True
+    except (OSError, TypeError, ValueError) as e:
+        return False
+
+
+def load_private_settings():
+    """
+    Load private settings from ~/.cato/settings.json in an OS-compatible way.
+    
+    This function constructs the path to ~/.cato/settings.json using os.path methods
+    for cross-platform compatibility (Windows, Mac, Linux).
+    
+    Returns:
+        dict: The privateCommands section from the settings file, or empty dict if not found
+    """
+    # Use os.path.join for OS-compatible path construction
+    cato_dir = os.path.join(os.path.expanduser("~"), ".cato")
+    settings_file = os.path.join(cato_dir, "settings.json")
+    
+    try:
+        with open(settings_file, 'r', encoding='utf-8') as f:
+            settings = json.load(f)
+            return settings.get('privateCommands', {})
+    except (FileNotFoundError, json.JSONDecodeError, KeyError, OSError):
+        return {}
+
+
+def ensure_directory_exists(directory_path):
+    """
+    Ensure a directory exists, creating it if necessary.
+    
+    Args:
+        directory_path (str): Path to the directory
+        
+    Returns:
+        bool: True if directory exists or was created successfully
+    """
+    try:
+        os.makedirs(directory_path, exist_ok=True)
+        return True
+    except OSError:
+        return False

{catocli-2.1.2 → catocli-2.1.3}/catocli/__init__.py

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

{catocli-2.1.2 → catocli-2.1.3}/catocli/parsers/custom/export_sites/export_sites.py

@@ -6,6 +6,7 @@ from datetime import datetime
 from graphql_client.api.call_api import ApiClient, CallApi
 from graphql_client.api_client import ApiException
 from ..customLib import writeDataToFile, makeCall, getAccountID
+from ....Utils.cliutils import load_cli_settings
 
 def export_socket_site_to_json(args, configuration):
     """
@@ -20,9 +21,8 @@ def export_socket_site_to_json(args, configuration):
     }
 
     try:
-        settings = {}
-        with open(os.path.join(os.path.dirname(__file__), '../../../../settings.json'), 'r', encoding='utf-8') as f:
-            settings = json.load(f)
+        # Load CLI settings using the robust function
+        settings = load_cli_settings()
 
         account_id = getAccountID(args, configuration)
         # Get account snapshot with siteIDs if provided

{catocli-2.1.2 → catocli-2.1.3}/catocli/parsers/custom_private/__init__.py

@@ -4,21 +4,9 @@ Private commands parser for custom GraphQL payloads
 Dynamically loads commands from ~/.cato/settings.json
 """
 
-import os
-import json
 import argparse
 from ..customParserApiClient import createPrivateRequest, get_private_help
-
-
-def load_private_settings():
-    """Load private settings from ~/.cato/settings.json"""
-    settings_file = os.path.expanduser("~/.cato/settings.json")
-    try:
-        with open(settings_file, 'r') as f:
-            settings = json.load(f)
-            return settings.get('privateCommands', {})
-    except (FileNotFoundError, json.JSONDecodeError, KeyError):
-        return {}
+from ...Utils.cliutils import load_private_settings
 
 
 def private_parse(subparsers):

{catocli-2.1.2 → catocli-2.1.3/catocli.egg-info}/PKG-INFO

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

{catocli-2.1.2 → catocli-2.1.3}/catocli.egg-info/SOURCES.txt

@@ -1,6 +1,7 @@
 LICENSE
 MANIFEST.in
 README.md
+clisettings.json
 setup.py
 catocli/__init__.py
 catocli/__main__.py
@@ -11,6 +12,7 @@ catocli.egg-info/entry_points.txt
 catocli.egg-info/requires.txt
 catocli.egg-info/top_level.txt
 catocli/Utils/clidriver.py
+catocli/Utils/cliutils.py
 catocli/Utils/profile_manager.py
 catocli/Utils/version_checker.py
 catocli/parsers/customParserApiClient.py

catocli-2.1.3/clisettings.json

@@ -0,0 +1,29 @@
+{
+    "export_by_socket_type": {
+        "SOCKET_X1500":true,
+        "SOCKET_X1600":true,
+        "SOCKET_X1600_LTE":true,
+        "SOCKET_X1700":true
+    },
+    "default_socket_interface_map": {
+        "SOCKET_X1500":"LAN1",
+        "SOCKET_X1600":"INT_5",
+        "SOCKET_X1600_LTE":"INT_5",
+        "SOCKET_X1700":"INT_3"
+    },
+    "childOperationParent": {
+        "xdr": true,
+        "policy": true,
+        "groups": true,
+        "newGroups": true,
+        "site": true
+    },
+    "childOperationObjects": {
+        "ipAddressRange": true,
+        "fqdn": true,
+        "PolicyQueries": true,
+        "GroupsQueries": true,
+        "ContainerQueries": true,
+        "SiteQueries": true
+    }
+}

{catocli-2.1.2 → catocli-2.1.3}/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("../settings.json")
+    settings = loadJSON("../clisettings.json")
     childOperationParent = settings.get("childOperationParent", {})
     childOperationObjects = settings.get("childOperationObjects", {})
     
@@ -786,15 +786,6 @@ def show_version_info(args, configuration=None):
         else:
             print("Unable to check for updates (check your internet connection)")
     return [{"success": True, "current_version": catocli.__version__, "latest_version": latest_version if not args.current_only else None}]
-
-def load_private_settings():
-    # Load private settings from ~/.cato/settings.json
-    settings_file = os.path.expanduser("~/.cato/settings.json")
-    try:
-        with open(settings_file, 'r') as f:
-            return json.load(f)
-    except (FileNotFoundError, json.JSONDecodeError):
-        return {}
         
 def get_configuration(skip_api_key=False):
     configuration = Configuration()

{catocli-2.1.2 → catocli-2.1.3}/setup.py

@@ -13,7 +13,7 @@ setuptools.setup(
     },
     install_requires=['urllib3', 'certifi', 'six'],
     package_data={
-        '': ['vendor/*'],
+        '': ['vendor/*', 'clisettings.json'],
     },
     python_requires='>=3.6',
     url='https://github.com/Cato-Networks/cato-cli',