comparison gcp_batch_netcat.py @ 3:0ea626b10557 draft

planemo upload for repository https://github.com/afgane/gcp_batch_netcat

2:79160beab2a4

    args = parser.parse_args()

    # Set up authentication using the service account key
    os.environ['GOOGLE_APPLICATION_CREDENTIALS'] = args.service_account_key

    job_name = f'netcat-job-{uuid.uuid4()}'

    job_spec = {
        "taskGroups": [
            {
                "taskSpec": {
                    "runnables": [
                        {
                            "script": {
                                "text": f"nc -z -v {args.nfs_address} {args.port}"
                            }
                        }
                    ]
                },
                "taskCount": 1,
                "parallelism": 1
            }
        ],

    command = [
        'gcloud', 'batch', 'jobs', 'submit', job_name,
        '--location', args.region,
        '--project', args.project,
        '--config', job_spec_file,
        '--format=text'
    ]

    # Wait 4 minutes before submitting the job
    time.sleep(240)

3:0ea626b10557

    args = parser.parse_args()

    # Set up authentication using the service account key
    os.environ['GOOGLE_APPLICATION_CREDENTIALS'] = args.service_account_key

    # Ensure gcloud uses a writable config directory
    os.environ['CLOUDSDK_CONFIG'] = '/tmp/gcloud-config'

    # Create the temp config directory if it doesn't exist
    os.makedirs('/tmp/gcloud-config', exist_ok=True)

    job_name = f'netcat-job-{uuid.uuid4()}'

    job_spec = {
        "taskGroups": [
            {
                "taskSpec": {
                    "runnables": [
                        {
                            "script": {
                                "text": f"/usr/bin/nc -z -v {args.nfs_address} {args.port}"
                            }
                        }
                    ],
                    "computeResource": {
                        "cpuMilli": 1000,
                        "memoryMib": 1024
                    },
                    "environment": {
                        "imageUri": "afgane/gcp-batch-netcat:0.1.0"
                    }
                },
                "taskCount": 1,
                "parallelism": 1
            }
        ],

    command = [
        'gcloud', 'batch', 'jobs', 'submit', job_name,
        '--location', args.region,
        '--project', args.project,
        '--config', job_spec_file,
        '--format=text',
        '--verbosity=debug'
    ]

    # Wait 4 minutes before submitting the job
    time.sleep(240)