comparison gcp_batch_netcat.py @ 7:fcfb703748b1 draft

planemo upload for repository https://github.com/afgane/gcp_batch_netcat commit ece227052d14d755b0d0b07a827152b2e98fb94b-dirty

6:d25792770df8

    format='%(asctime)s - %(levelname)s - %(message)s',
    stream=sys.stdout
)
logger = logging.getLogger(__name__)

def discover_nfs_loadbalancer_ip():
    """
    Try to discover NFS LoadBalancer IP via Kubernetes API
    Returns the external IP if found, None otherwise
    """
    try:
        import subprocess
        logger.info("Attempting to discover NFS LoadBalancer IP via kubectl...")
        result = subprocess.run(['kubectl', 'get', 'svc', '-n', 'nfs-provisioner', '-o', 'json'], capture_output=True, text=True)
        if result.returncode == 0:
            services = json.loads(result.stdout)
            for item in services.get('items', []):
                name = item.get('metadata', {}).get('name', '')
                # Look for NFS-related service names
                if any(keyword in name.lower() for keyword in ['nfs-provisioner-nfs-server-provisioner']):
                    spec = item.get('spec', {})
                    if spec.get('type') == 'LoadBalancer':
                        ingress = item.get('status', {}).get('loadBalancer', {}).get('ingress', [])
                        if ingress:
                            ip = ingress[0].get('ip')
                            if ip:
                                logger.info(f"Found NFS LoadBalancer service '{name}' with external IP: {ip}")
                                return ip
            logger.warning("No NFS LoadBalancer services found via kubectl")
        else:
            logger.warning(f"kubectl command failed: {result.stderr}")
    except Exception as e:
        logger.warning(f"Could not discover NFS LoadBalancer IP via kubectl: {e}")
    return None

def determine_test_target(args):
    """Determine the target host and port based on test type"""

    if args.test_type == 'nfs':
        # Extract NFS server address if not provided
        if args.nfs_address:
            nfs_address = args.nfs_address
            logger.info(f"Using provided NFS address: {nfs_address}")
        else:
            # Try to auto-discover NFS LoadBalancer IP via Kubernetes API
            nfs_address = discover_nfs_loadbalancer_ip()
            if not nfs_address:
                raise ValueError("Could not auto-detect NFS LoadBalancer IP. Please provide --nfs_address parameter with the LoadBalancer external IP.")
        return nfs_address, 2049

    else:
        raise ValueError(f"Unsupported test type: {args.test_type}")

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('--nfs_address', required=False, help='NFS server LoadBalancer IP address (if not provided, will be auto-detected via Kubernetes API)')
    parser.add_argument('--output', required=True)
    parser.add_argument('--project', required=False, help='GCP Project ID (if not provided, will be extracted from service account key)')
    parser.add_argument('--region', required=True)
    parser.add_argument('--network', default='default', help='GCP Network name')
    parser.add_argument('--subnet', default='default', help='GCP Subnet name')

7:fcfb703748b1

    format='%(asctime)s - %(levelname)s - %(message)s',
    stream=sys.stdout
)
logger = logging.getLogger(__name__)



def determine_test_target(args):
    """Determine the target host and port based on test type"""

    if args.test_type == 'nfs':
        # NFS server address is required
        if not args.nfs_address:
            raise ValueError("NFS server address is required. Please provide --nfs_address parameter with the LoadBalancer external IP.")

        nfs_address = args.nfs_address
        logger.info(f"Using provided NFS address: {nfs_address}")
        return nfs_address, 2049

    else:
        raise ValueError(f"Unsupported test type: {args.test_type}")

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('--nfs_address', required=True, help='NFS server LoadBalancer external IP address (required)')
    parser.add_argument('--output', required=True)
    parser.add_argument('--project', required=False, help='GCP Project ID (if not provided, will be extracted from service account key)')
    parser.add_argument('--region', required=True)
    parser.add_argument('--network', default='default', help='GCP Network name')
    parser.add_argument('--subnet', default='default', help='GCP Subnet name')