def get_vmss_batch_resource_ids()

in src/hpcadvisor/batch_handler.py [0:0]

• 21 lines of code
• 6 McCabe index (conditional complexity)

def get_vmss_batch_resource_ids(batch_client, poolid):
    batch_resource_group = _get_batch_resource_group(poolid)
    if not batch_resource_group:
        return []

    resource_ids = []
    credentials = DefaultAzureCredential()
    subscription_id = env["SUBSCRIPTION"]

    rclient = ResourceManagementClient(credentials, subscription_id)
    items = rclient.resources.list_by_resource_group(batch_resource_group)
    vmss = None
    for resource in items:
        if resource.type == "Microsoft.Compute/virtualMachineScaleSets":
            vmss = resource
        break

    if vmss:
        compute_client = ComputeManagementClient(credentials, subscription_id)
        instances = compute_client.virtual_machine_scale_set_vms.list(
            batch_resource_group, vmss.name
        )
        resource_ids = [instance.id for instance in instances]
    return resource_ids