def get_vm_managed_identity()

in playbooks/roles/cyclecloud/files/configure.py [0:0]

• 18 lines of code
• 4 McCabe index (conditional complexity)

def get_vm_managed_identity():
    # Managed Identity may  not be available immediately at VM startup...
    # Test/Pause/Retry to see if it gets assigned
    metadata_url = ('http://169.254.169.254/metadata/identity/oauth2/token'
                    '?api-version=2018-02-01'
                    '&resource=https://management.azure.com/')
    metadata_req = Request(metadata_url, headers={"Metadata": True})

    for _ in range(30):
        print("Fetching managed identity")
        metadata_response = urlopen(metadata_req, timeout=2)

        try:
            return json.load(metadata_response)
        except ValueError as e:
            print("Failed to get managed identity %s" % e)
            print("    Retrying")
            sleep(10)
            continue
        except:
            print("Unable to obtain managed identity after 30 tries")
            raise