def main():
'''Main routine.'''
# process arguments
if len(sys.argv) < 3:
usage()
rgname = sys.argv[1]
vmss_name = sys.argv[2]
# Load Azure app defaults
try:
with open('azurermconfig.json') as config_file:
config_data = json.load(config_file)
except FileNotFoundError:
sys.exit('Error: Expecting azurermconfig.json in current folder')
tenant_id = config_data['tenantId']
app_id = config_data['appId']
app_secret = config_data['appSecret']
subscription_id = config_data['subscriptionId']
access_token = azurerm.get_access_token(tenant_id, app_id, app_secret)
print('Printing VMSS details\n')
vmssget = azurerm.get_vmss(access_token, subscription_id, rgname,
vmss_name)
name = vmssget['name']
capacity = vmssget['sku']['capacity']
location = vmssget['location']
offer = \
vmssget['properties']['virtualMachineProfile']['storageProfile']['imageReference']['offer']
sku = vmssget['properties']['virtualMachineProfile']['storageProfile'][
'imageReference']['sku']
print(
json.dumps(vmssget, sort_keys=False, indent=2, separators=(',', ': ')))
print('Name: ' + name + ', capacity: ' + str(capacity) + ', ' + location +
', ' + offer + ', ' + sku)
print('Printing VMSS instance view')
instance_view = azurerm.get_vmss_instance_view(access_token,
subscription_id, rgname,
vmss_name)
print(
json.dumps(instance_view,
sort_keys=False,
indent=2,
separators=(',', ': ')))
'''
access_token = azurerm.get_access_token(tenant_id, app_id, app_secret)
print('Printing VMSS details\n')
vmssget = azurerm.get_vmss(access_token, subscription_id, rg, vmss)
name = vmssget['name']
capacity = vmssget['sku']['capacity']
location = vmssget['location']
offer = vmssget['properties']['virtualMachineProfile']['storageProfile'][
'imageReference']['offer']
sku = vmssget['properties']['virtualMachineProfile']['storageProfile'][
'imageReference']['sku']
print(json.dumps(vmssget, sort_keys=False, indent=2, separators=(',', ': ')))
print('Name: ' + name + ', capacity: ' + str(capacity) + ', ' + location +
', ' + offer + ', ' + sku)
print('\nPrinting VMSS instance view\n')
instanceView = azurerm.get_vmss_instance_view(access_token, subscription_id,
rg, vmss)
print(
json.dumps(instanceView, sort_keys=False, indent=2,
separators=(',', ': ')))
'''
print('\nListing VMSS VMs\n')
vmss_vms = azurerm.list_vmss_vms(access_token, subscription_id, rg, vmss)
#print(vmss_vms)
print(json.dumps(vmss_vms, sort_keys=False, indent=2, separators=(',', ': ')))
for vm in vmss_vms['value']:
instanceId = vm['instanceId']
vmInstanceView = azurerm.get_vmss_vm_instance_view(access_token, subscription_id, rg, vmss, instanceId)
print('\nVM ' + str(instanceId) + ' instance view\n')
print(json.dumps(vmInstanceView, sort_keys=False, indent=2, separators=(',', ': ')))
'''
def main():
# create parser
argParser = argparse.ArgumentParser()
argParser.add_argument('--vmssname',
'-s',
required=True,
action='store',
help='VM Scale Set name')
argParser.add_argument('--resourcegroup',
'-r',
required=True,
dest='resource_group',
action='store',
help='Resource group name')
argParser.add_argument('--newversion',
'-n',
dest='newversion',
action='store',
help='New platform image version string')
argParser.add_argument('--customuri',
'-c',
dest='customuri',
action='store',
help='New custom image URI string')
argParser.add_argument('--updatedomain',
'-u',
dest='updatedomain',
action='store',
type=int,
help='Update domain (int)')
argParser.add_argument('--vmid',
'-i',
dest='vmid',
action='store',
type=int,
help='Single VM ID (int)')
argParser.add_argument('--vmlist',
'-l',
dest='vmlist',
action='store',
help='List of VM IDs e.g. "["1", "2"]"')
argParser.add_argument('--nowait',
'-w',
action='store_true',
default=False,
help='Start upgrades and then exit without waiting')
argParser.add_argument('--verbose',
'-v',
action='store_true',
default=False,
help='Show additional information')
argParser.add_argument('-y',
dest='noprompt',
action='store_true',
default=False,
help='Do not prompt for confirmation')
args = argParser.parse_args()
# switches to determine program behavior
noprompt = args.noprompt # go ahead and upgrade without waiting for confirmation when True
nowait = args.nowait # don't loop waiting for upgrade provisioning to complete when True
verbose = args.verbose # print extra status information when True
vmssname = args.vmssname
resource_group = args.resource_group
if args.newversion is not None:
newversion = args.newversion
storagemode = 'platform'
elif args.customuri is not None:
customuri = args.customuri
storagemode = 'custom'
else:
argParser.error(
'You must specify a new version for platform images or a custom uri for custom images'
)
if args.updatedomain is not None:
updatedomain = args.updatedomain
upgrademode = 'updatedomain'
elif args.vmid is not None:
vmid = args.vmid
upgrademode = 'vmid'
elif args.vmlist is not None:
vmlist = args.vmlist
upgrademode = 'vmlist'
else:
argParser.error(
'You must specify an update domain, a vm id, or a vm list')
# Load Azure app defaults
try:
with open('vmssconfig.json') as configFile:
configdata = json.load(configFile)
except FileNotFoundError:
print("Error: Expecting vmssconfig.json in current folder")
sys.exit()
tenant_id = configdata['tenantId']
app_id = configdata['appId']
app_secret = configdata['appSecret']
subscription_id = configdata['subscriptionId']
access_token = azurerm.get_access_token(tenant_id, app_id, app_secret)
# get the vmss model
vmssmodel = azurerm.get_vmss(access_token, subscription_id, resource_group,
vmssname)
# print(json.dumps(vmssmodel, sort_keys=False, indent=2, separators=(',', ': ')))
if storagemode == 'platform':
# check current version
imagereference = vmssmodel['properties']['virtualMachineProfile'][
'storageProfile']['imageReference']
print('Current image reference in Scale Set model:')
print(
json.dumps(imagereference,
sort_keys=False,
indent=2,
separators=(',', ': ')))
# compare current version with new version
if imagereference['version'] == newversion:
print('Scale Set model version is already set to ' + newversion +
', skipping model update.')
else:
if not noprompt:
response = input('Confirm version upgrade to: ' + newversion +
' (y/n)')
if response.lower() != 'y':
sys.exit(1)
# change the version
vmssmodel['properties']['virtualMachineProfile']['storageProfile'][
'imageReference']['version'] = newversion
# put the vmss model
updateresult = azurerm.update_vmss(access_token, subscription_id,
resource_group, vmssname,
json.dumps(vmssmodel))
if verbose:
print(updateresult)
print('OS version updated to ' + newversion +
' in model for VM Scale Set: ' + vmssname)
else: # storagemode = custom
# check current uri
oldimageuri = vmssmodel['properties']['virtualMachineProfile'][
'storageProfile']['osDisk']['image']['uri']
print('Current image URI in Scale Set model:' + oldimageuri)
# compare current uri with new uri
if oldimageuri == customuri:
print('Scale Set model version is already set to ' + customuri +
', skipping model update.')
else:
if not noprompt:
response = input('Confirm uri upgrade to: ' + customuri +
' (y/n)')
if response.lower() != 'y':
sys.exit(1)
# change the version
vmssmodel['properties']['virtualMachineProfile']['storageProfile'][
'osDisk']['image']['uri'] = customuri
# put the vmss model
updateresult = azurerm.update_vmss(access_token, subscription_id,
resource_group, vmssname,
json.dumps(vmssmodel))
if verbose:
print(updateresult)
print('Image URI updated to ' + customuri +
' in model for VM Scale Set: ' + vmssname)
# build the list of VMs to upgrade depending on the upgrademode setting
if upgrademode == 'updatedomain':
# list the VMSS VM instance views to determine their update domains
print('Examining the scale set..')
udinstancelist = get_vm_ids_by_ud(access_token, subscription_id,
resource_group, vmssname,
updatedomain)
print('VM instances in UD: ' + str(updatedomain) + ' to upgrade:')
print(udinstancelist)
vmids = json.dumps(udinstancelist)
print('Upgrading VMs in UD: ' + str(updatedomain))
elif upgrademode == 'vmid':
vmids = json.dumps([str(vmid)])
print('Upgrading VM ID: ' + str(vmid))
else: # upgrademode = vmlist
vmids = vmlist
print('Upgrading VM IDs: ' + vmlist)
# do manualupgrade on the VMs in the list
upgraderesult = azurerm.upgrade_vmss_vms(access_token, subscription_id,
resource_group, vmssname, vmids)
print(upgraderesult)
# now wait for upgrade to complete
# query VM scale set instance view
if not nowait:
updatecomplete = False
provisioningstate = ''
while not updatecomplete:
vmssinstanceview = azurerm.get_vmss_instance_view(
access_token, subscription_id, resource_group, vmssname)
for status in vmssinstanceview['statuses']:
provisioningstate = status['code']
if provisioningstate == 'ProvisioningState/succeeded':
updatecomplete = True
if verbose:
print(provisioningstate)
time.sleep(5)
print(status['code'])
else:
print(
'Check Scale Set provisioning state to determine when upgrade is complete.'
)
def main():
# create parser
argParser = argparse.ArgumentParser()
argParser.add_argument('--vmssname', '-s', required=True, action='store', help='VM Scale Set name')
argParser.add_argument('--resourcegroup', '-r', required=True, dest='resource_group', action='store',
help='Resource group name')
argParser.add_argument('--newversion', '-n', dest='newversion', action='store',
help='New platform image version string')
argParser.add_argument('--customuri', '-c', dest='customuri', action='store', help='New custom image URI string')
argParser.add_argument('--updatedomain', '-u', dest='updatedomain', action='store', type=int,
help='Update domain (int)')
argParser.add_argument('--vmid', '-i', dest='vmid', action='store', type=int, help='Single VM ID (int)')
argParser.add_argument('--vmlist', '-l', dest='vmlist', action='store', help='List of VM IDs e.g. "["1", "2"]"')
argParser.add_argument('--nowait', '-w', action='store_true', default=False,
help='Start upgrades and then exit without waiting')
argParser.add_argument('--verbose', '-v', action='store_true', default=False, help='Show additional information')
argParser.add_argument('-y', dest='noprompt', action='store_true', default=False,
help='Do not prompt for confirmation')
args = argParser.parse_args()
# switches to determine program behavior
noprompt = args.noprompt # go ahead and upgrade without waiting for confirmation when True
nowait = args.nowait # don't loop waiting for upgrade provisioning to complete when True
verbose = args.verbose # print extra status information when True
vmssname = args.vmssname
resource_group = args.resource_group
if args.newversion is not None:
newversion = args.newversion
storagemode = 'platform'
elif args.customuri is not None:
customuri = args.customuri
storagemode = 'custom'
else:
argParser.error('You must specify a new version for platform images or a custom uri for custom images')
if args.updatedomain is not None:
updatedomain = args.updatedomain
upgrademode = 'updatedomain'
elif args.vmid is not None:
vmid = args.vmid
upgrademode = 'vmid'
elif args.vmlist is not None:
vmlist = args.vmlist
upgrademode = 'vmlist'
else:
argParser.error('You must specify an update domain, a vm id, or a vm list')
# Load Azure app defaults
try:
with open('vmssconfig.json') as configFile:
configdata = json.load(configFile)
except FileNotFoundError:
print("Error: Expecting vmssconfig.json in current folder")
sys.exit()
tenant_id = configdata['tenantId']
app_id = configdata['appId']
app_secret = configdata['appSecret']
subscription_id = configdata['subscriptionId']
access_token = azurerm.get_access_token(tenant_id, app_id, app_secret)
# get the vmss model
vmssmodel = azurerm.get_vmss(access_token, subscription_id, resource_group, vmssname)
# print(json.dumps(vmssmodel, sort_keys=False, indent=2, separators=(',', ': ')))
if storagemode == 'platform':
# check current version
imagereference = vmssmodel['properties']['virtualMachineProfile']['storageProfile']['imageReference']
print('Current image reference in Scale Set model:')
print(json.dumps(imagereference, sort_keys=False, indent=2, separators=(',', ': ')))
# compare current version with new version
if imagereference['version'] == newversion:
print('Scale Set model version is already set to ' + newversion + ', skipping model update.')
else:
if not noprompt:
response = input('Confirm version upgrade to: ' + newversion + ' (y/n)')
if response.lower() != 'y':
sys.exit(1)
# change the version
vmssmodel['properties']['virtualMachineProfile']['storageProfile']['imageReference']['version'] = newversion
# put the vmss model
updateresult = azurerm.update_vmss(access_token, subscription_id, resource_group, vmssname,
json.dumps(vmssmodel))
if verbose:
print(updateresult)
print('OS version updated to ' + newversion + ' in model for VM Scale Set: ' + vmssname)
else: # storagemode = custom
# check current uri
oldimageuri = vmssmodel['properties']['virtualMachineProfile']['storageProfile']['osDisk']['image']['uri']
print('Current image URI in Scale Set model:' + oldimageuri)
# compare current uri with new uri
if oldimageuri == customuri:
print('Scale Set model version is already set to ' + customuri + ', skipping model update.')
else:
if not noprompt:
response = input('Confirm uri upgrade to: ' + customuri + ' (y/n)')
if response.lower() != 'y':
sys.exit(1)
# change the version
vmssmodel['properties']['virtualMachineProfile']['storageProfile']['osDisk']['image']['uri'] = customuri
# put the vmss model
updateresult = azurerm.update_vmss(access_token, subscription_id, resource_group, vmssname,
json.dumps(vmssmodel))
if verbose:
print(updateresult)
print('Image URI updated to ' + customuri + ' in model for VM Scale Set: ' + vmssname)
# build the list of VMs to upgrade depending on the upgrademode setting
if upgrademode == 'updatedomain':
# list the VMSS VM instance views to determine their update domains
print('Examining the scale set..')
udinstancelist = get_vm_ids_by_ud(access_token, subscription_id, resource_group, vmssname, updatedomain)
print('VM instances in UD: ' + str(updatedomain) + ' to upgrade:')
print(udinstancelist)
vmids = json.dumps(udinstancelist)
print('Upgrading VMs in UD: ' + str(updatedomain))
elif upgrademode == 'vmid':
vmids = json.dumps([str(vmid)])
print('Upgrading VM ID: ' + str(vmid))
else: # upgrademode = vmlist
vmids = vmlist
print('Upgrading VM IDs: ' + vmlist)
# do manualupgrade on the VMs in the list
upgraderesult = azurerm.upgrade_vmss_vms(access_token, subscription_id, resource_group, vmssname, vmids)
print(upgraderesult)
# now wait for upgrade to complete
# query VM scale set instance view
if not nowait:
updatecomplete = False
provisioningstate = ''
while not updatecomplete:
vmssinstanceview = azurerm.get_vmss_instance_view(access_token, subscription_id, resource_group, vmssname)
for status in vmssinstanceview['statuses']:
provisioningstate = status['code']
if provisioningstate == 'ProvisioningState/succeeded':
updatecomplete = True
if verbose:
print(provisioningstate)
time.sleep(5)
print(status['code'])
else:
print('Check Scale Set provisioning state to determine when upgrade is complete.')
app_secret = configData['appSecret']
subscription_id = configData['subscriptionId']
rg = configData['resourceGroup']
vmss = configData['vmssName']
access_token = azurerm.get_access_token(tenant_id, app_id, app_secret)
print('Printing VMSS details\n')
vmssget = azurerm.get_vmss(access_token, subscription_id, rg, vmss)
name = vmssget['name']
capacity = vmssget['sku']['capacity']
location = vmssget['location']
offer = vmssget['properties']['virtualMachineProfile']['storageProfile']['imageReference']['offer']
sku = vmssget['properties']['virtualMachineProfile']['storageProfile']['imageReference']['sku']
print(json.dumps(vmssget, sort_keys=False, indent=2, separators=(',', ': ')))
print('Name: ' + name + ', capacity: ' + str(capacity) + ', ' + location + ', ' + offer + ', ' + sku)
print('\nPrinting VMSS instance view\n')
instanceView = azurerm.get_vmss_instance_view(access_token, subscription_id, rg, vmss)
print(json.dumps(instanceView, sort_keys=False, indent=2, separators=(',', ': ')))
'''
print('\nListing VMSS VMs\n')
vmss_vms = azurerm.list_vmss_vms(access_token, subscription_id, rg, vmss)
#print(vmss_vms)
print(json.dumps(vmss_vms, sort_keys=False, indent=2, separators=(',', ': ')))
for vm in vmss_vms['value']:
instanceId = vm['instanceId']
vmInstanceView = azurerm.get_vmss_vm_instance_view(access_token, subscription_id, rg, vmss, instanceId)
print('\nVM ' + str(instanceId) + ' instance view\n')
print(json.dumps(vmInstanceView, sort_keys=False, indent=2, separators=(',', ': ')))
'''
