Add CLI wait 1.0.6

src/vm-repair/HISTORY.rst

@@ -1,6 +1,11 @@
 
 Release History
 ===============
+1.0.6
+++++++
+Add while loop for ASG to check if the operation is done as the async 2rd operation will cancel the 1st call.
+
+
 1.0.5
 ++++++
 Bug fix ASG is not added properly when reset the nic

src/vm-repair/azext_vm_repair/custom.py

@@ -618,9 +618,11 @@ def reset_nic(cmd, vm_name, resource_group_name, yes=False):
             for item in ip_config_object[applicationSecurityGroups]:
                 application_id_tokens = item['id'].split('/')
                 if application_id_tokens[-1] is not None:
+
                     application_names+=application_id_tokens[-1]+ " "
-            logger.info('applicationSecurityGroups {application_names}...\n')    
-
+
+        logger.info('applicationSecurityGroups {application_names}...\n')
+
 
         # Dynamic | Static
         orig_ip_allocation_method = ip_config_object['privateIPAllocationMethod']
@@ -644,7 +646,20 @@ def reset_nic(cmd, vm_name, resource_group_name, yes=False):
             update_ip_command = 'az network nic ip-config update -g {g} --nic-name {nic} -n {config} --private-ip-address {ip}' \
                                 .format(g=resource_group_name, nic=primary_nic_name, config=ipconfig_name, ip=swap_ip_address)
         _call_az_command(update_ip_command)
-
+            # Check if IP is updated
+    while True:
+        get_current_ip_command = 'az network nic ip-config wait --updatede -g {g} --nic-name {nic} -n ' \
+                                .format(g=resource_group_name, nic=primary_nic_name)
+
+        current_ip_address = _call_az_command(get_current_ip_command)
+        if current_ip_address == swap_ip_address:
+            logger.info('IP address updated successfully to {ip}\n'.format(ip=swap_ip_address))
+            break
+        else:
+            logger.info('IP address not updated yet. Waiting for a few minutes before retrying...\n')
+            time.sleep(300)
+            # Sleep for 5 minutes 
+
         # 4) Change things back. This will also invoke and wait for a VM restart.
         logger.info('NIC reset is complete. Now reverting back to your original configuration...\n')
         # If user had dynamic config, change back to dynamic
@@ -785,4 +800,4 @@ def repair_and_restore(cmd, vm_name, resource_group_name, repair_password=None,
 
     logger.info('\n%s\n', command.message)
 
-    return return_dict
+    return return_dict

src/vm-repair/setup.py

@@ -8,7 +8,7 @@
 from codecs import open
 from setuptools import setup, find_packages
 
-VERSION = "1.0.5"
+VERSION = "1.0.6"
 
 CLASSIFIERS = [
     'Development Status :: 4 - Beta',