cleanup

Author: Sandido

src/vm-repair/azext_vm_repair/_validators.py

@@ -235,9 +235,11 @@ def _prompt_public_ip(namespace):
     except NoTTYException:
         raise ValidationError('Please specify the associate-public-ip parameter in non-interactive mode.')
 
+
 def _return_public_ip_name(namespace):
     return namespace.repair_vm_name + "PublicIP"
 
+
 def _classic_vm_exists(cmd, resource_group_name, vm_name):
     classic_vm_provider = 'Microsoft.ClassicCompute'
     vm_resource_type = 'virtualMachines'

src/vm-repair/azext_vm_repair/custom.py

@@ -162,9 +162,8 @@ def create(cmd, vm_name, resource_group_name, repair_password=None, repair_usern
         # Adding the size to the command.  
         create_repair_vm_command += ' --size {sku}'.format(sku=sku)  
 
-
         # Setting the availability zone for the repair VM.  
-        # If the source VM has availability zones, the first one is chosen for the repair VM.  
+        # If the source VM has availability zones, the first one is chosen for the repair VM.
         if source_vm.zones:  
             zone = source_vm.zones[0]  
             create_repair_vm_command += ' --zone {zone}'.format(zone=zone)  
@@ -271,7 +270,6 @@ def create(cmd, vm_name, resource_group_name, repair_password=None, repair_usern
                 # Execute the command to attach the disk.  
                 _call_az_command(attach_disk_command)  
 
-
         # Check if the source VM uses unmanaged disks.  
         # If it does, the repair VM will also be created with unmanaged disks.  
         else:  
@@ -334,7 +332,6 @@ def create(cmd, vm_name, resource_group_name, repair_password=None, repair_usern
                 .format(g=repair_group_name, disk_name=copy_disk_name, vm_name=repair_vm_name, uri=copy_disk_id)  
             _call_az_command(attach_disk_command)  
 
-
         # Check if the Nested Hyper-V needs to be enabled.  
         # If it does, run the script to install Hyper-V and create the nested VM.  
         if enable_nested:  
@@ -436,6 +433,10 @@ def create(cmd, vm_name, resource_group_name, repair_password=None, repair_usern
 # This method is responsible for restoring the VM after repair  
 def restore(cmd, vm_name, resource_group_name, disk_name=None, repair_vm_id=None, yes=False):  
 
+    # Initialize repair VM variables to avoid "used before assignment" errors
+    repair_vm_name = None
+    repair_resource_group = None
+    
     # Create an instance of the command helper object to facilitate logging and status tracking.  
     command = command_helper(logger, cmd, 'vm repair restore')  
 
@@ -605,8 +606,7 @@ def run(cmd, vm_name, resource_group_name, run_id=None, repair_vm_id=None, custo
 
         logger.info('Running script on %s: %s', vm_string, repair_vm_name)  # Log the VM on which the script is being run  
 
-
-                # Start the timer to measure the run-time of the script  
+        # Start the timer to measure the run-time of the script  
         script_start_time = timeit.default_timer()  
 
         # Invoke the run command on the VM and capture the standard output and error  
@@ -654,7 +654,6 @@ def run(cmd, vm_name, resource_group_name, run_id=None, repair_vm_id=None, custo
         # Set the overall command status to success  
         command.set_status_success()  
 
-
     except KeyboardInterrupt:
         command.error_stack_trace = traceback.format_exc()
         command.error_message = "Command interrupted by user input."

src/vm-repair/azext_vm_repair/repair_utils.py

@@ -729,7 +729,7 @@ def _get_function_param_dict(frame):
     return values
 
 
-def _unlock_encrypted_vm_run(repair_vm_name, repair_group_name, is_linux, encrypt_recovery_key = ""):
+def _unlock_encrypted_vm_run(repair_vm_name, repair_group_name, is_linux, encrypt_recovery_key=""):
     stdout, stderr = _unlock_singlepass_encrypted_disk(repair_vm_name, repair_group_name, is_linux, encrypt_recovery_key)
     logger.debug('Unlock script STDOUT:\n%s', stdout)
     if stderr: