catocli 2.0.2__py3-none-any.whl → 2.0.3__py3-none-any.whl

catocli/parsers/custom/import_rules_to_tf/import_rules_to_tf.py

@@ -13,6 +13,7 @@ import re
 import time
 import glob
 from pathlib import Path
+from ..customLib import validate_terraform_environment
 
 
 def load_json_data(json_file):
@@ -191,143 +192,6 @@ def import_rules(rules, module_name, verbose=False,
     return successful_imports, failed_imports
 
 
-def check_terraform_binary():
-    """Check if terraform binary is available"""
-    try:
-        result = subprocess.run(['terraform', '--version'], capture_output=True, text=True)
-        if result.returncode == 0:
-            return True, result.stdout.strip().split('\n')[0]
-        else:
-            return False, "Terraform binary not found or not working"
-    except FileNotFoundError:
-        return False, "Terraform binary not found in PATH"
-    except Exception as e:
-        return False, f"Error checking terraform binary: {e}"
-
-
-def check_terraform_config_files():
-    """Check if Terraform configuration files exist in current directory"""
-    tf_files = glob.glob('*.tf') + glob.glob('*.tf.json')
-    if tf_files:
-        return True, tf_files
-    else:
-        return False, []
-
-
-def check_terraform_init():
-    """Check if Terraform has been initialized"""
-    terraform_dir = Path('.terraform')
-    if terraform_dir.exists() and terraform_dir.is_dir():
-        # Check for providers
-        providers_dir = terraform_dir / 'providers'
-        if providers_dir.exists():
-            return True, "Terraform is initialized"
-        else:
-            return False, "Terraform directory exists but no providers found"
-    else:
-        return False, "Terraform not initialized (.terraform directory not found)"
-
-
-def check_module_exists(module_name):
-    """Check if the specified module exists in Terraform configuration"""
-    try:
-        # Remove 'module.' prefix if present
-        clean_module_name = module_name.replace('module.', '')
-        
-        # Method 1: Check .tf files directly for module definitions
-        tf_files = glob.glob('*.tf') + glob.glob('*.tf.json')
-        for tf_file in tf_files:
-            try:
-                with open(tf_file, 'r') as f:
-                    content = f.read()
-                    # Look for module "module_name" blocks
-                    if f'module "{clean_module_name}"' in content or f"module '{clean_module_name}'" in content:
-                        return True, f"Module '{clean_module_name}' found in {tf_file}"
-            except Exception as e:
-                print(f"Warning: Could not read {tf_file}: {e}")
-                continue
-        
-        # Method 2: Try terraform show -json as fallback
-        try:
-            result = subprocess.run(
-                ['terraform', 'show', '-json'],
-                capture_output=True,
-                text=True,
-                cwd=Path.cwd()
-            )
-            
-            if result.returncode == 0:
-                state_data = json.loads(result.stdout)
-                
-                # Check if module exists in configuration
-                if 'configuration' in state_data and state_data['configuration']:
-                    modules = state_data.get('configuration', {}).get('root_module', {}).get('module_calls', {})
-                    if clean_module_name in modules:
-                        return True, f"Module '{clean_module_name}' found in Terraform state"
-                
-                # Also check in planned_values for modules
-                if 'planned_values' in state_data and state_data['planned_values']:
-                    modules = state_data.get('planned_values', {}).get('root_module', {}).get('child_modules', [])
-                    for module in modules:
-                        module_addr = module.get('address', '')
-                        if clean_module_name in module_addr:
-                            return True, f"Module '{clean_module_name}' found in planned values"
-        except (subprocess.SubprocessError, json.JSONDecodeError) as e:
-            print(f"Warning: Could not check terraform state: {e}")
-        
-        return False, f"Module '{clean_module_name}' not found in Terraform configuration files"
-            
-    except Exception as e:
-        return False, f"Error checking module existence: {e}"
-
-
-def validate_terraform_environment(module_name, verbose=False):
-    """Validate the complete Terraform environment"""
-    print("\n Validating Terraform environment...")
-    
-    # 1. Check terraform binary
-    print("\n Checking Terraform binary...")
-    has_terraform, terraform_msg = check_terraform_binary()
-    if not has_terraform:
-        raise Exception(f" Terraform not available: {terraform_msg}")
-    if verbose:
-        print(f" {terraform_msg}")
-    else:
-        print(" Terraform binary found")
-    
-    # 2. Check for configuration files
-    print("\n Checking Terraform configuration files...")
-    has_config, config_files = check_terraform_config_files()
-    if not has_config:
-        raise Exception(" No Terraform configuration files (.tf or .tf.json) found in current directory")
-    if verbose:
-        print(f" Found {len(config_files)} configuration files: {', '.join(config_files)}")
-    else:
-        print(f" Found {len(config_files)} Terraform configuration files")
-    
-    # 3. Check if terraform is initialized
-    print("\n Checking Terraform initialization...")
-    is_initialized, init_msg = check_terraform_init()
-    if not is_initialized:
-        raise Exception(f" {init_msg}. Run 'terraform init' first.")
-    if verbose:
-        print(f" {init_msg}")
-    else:
-        print(" Terraform is initialized")
-    
-    # 4. Check if the specified module exists
-    print(f"\n Checking if module '{module_name}' exists...")
-    module_exists, module_msg = check_module_exists(module_name)
-    if not module_exists:
-        raise Exception(f" {module_msg}. Please add the module to your Terraform configuration first.")
-    if verbose:
-        print(f" {module_msg}")
-    else:
-        print(f" Module '{module_name}' found")
-    
-    print("\n All Terraform environment checks passed!")
-
-
 def import_if_rules_to_tf(args, configuration):
     """Main function to orchestrate the import process"""
     try:

catocli/parsers/custom/import_sites_to_tf/__init__.py

@@ -0,0 +1,45 @@
+import catocli.parsers.custom.import_sites_to_tf.import_sites_to_tf as import_sites_to_tf
+
+def site_import_parse(subparsers, import_parser):
+    """Add socket sites import command to existing import parser"""
+    
+    if import_parser is None:
+        raise ValueError("Import parser not found. Make sure rule_import_parse is called before site_import_parse.")
+    
+    # Get the existing subparsers from the import parser
+    import_subparsers = None
+    for action in import_parser._subparsers._group_actions:
+        if hasattr(action, 'choices'):
+            import_subparsers = action
+            break
+    
+    if import_subparsers is None:
+        raise ValueError("Import subparsers not found in existing import parser.")
+    
+    # Add socket_sites_to_tf command
+    socket_sites_parser = import_subparsers.add_parser(
+        'socket_sites_to_tf', 
+        help='Import socket sites to Terraform state',
+        usage='catocli import socket_sites_to_tf <json_file> --module-name <module_name> [options]\n\nexample: catocli import socket_sites_to_tf config_data/socket_sites_11484.json --module-name module.sites'
+    )
+    
+    socket_sites_parser.add_argument('json_file', help='Path to the JSON file containing socket sites data')
+    socket_sites_parser.add_argument('--module-name', required=True, 
+                                help='Terraform module name to import resources into')
+    socket_sites_parser.add_argument('-accountID', help='Account ID (required by CLI framework but not used for import)', required=False)
+    socket_sites_parser.add_argument('--batch-size', type=int, default=10, 
+                                help='Number of imports per batch (default: 10)')
+    socket_sites_parser.add_argument('--delay', type=int, default=2, 
+                                help='Delay between batches in seconds (default: 2)')
+    socket_sites_parser.add_argument('--sites-only', action='store_true', 
+                                help='Import only sites, skip interfaces and network ranges')
+    socket_sites_parser.add_argument('--interfaces-only', action='store_true', 
+                                help='Import only WAN interfaces, skip sites and network ranges')
+    socket_sites_parser.add_argument('--network-ranges-only', action='store_true', 
+                                help='Import only network ranges, skip sites and interfaces')
+    socket_sites_parser.add_argument('-v', '--verbose', action='store_true', help='Verbose output')
+    socket_sites_parser.add_argument('--auto-approve', action='store_true', help='Skip confirmation prompt and proceed automatically')
+    
+    socket_sites_parser.set_defaults(func=import_sites_to_tf.import_socket_sites_to_tf)
+        
+    return import_parser