Updated azure_rm_manageddisk.py's method to update resources (ansible…

…-collections#1666)
l3ender · Aug 13, 2024 · 6cdd057 · 6cdd057
1 parent 997e4de
commit 6cdd057
Showing 1 changed file with 12 additions and 5 deletions.
diff --git a/plugins/modules/azure_rm_manageddisk.py b/plugins/modules/azure_rm_manageddisk.py
@@ -436,13 +436,15 @@ def exec_module(self, **kwargs):
 
         result = None
         changed = False
+        update_flag = False
 
         resource_group = self.get_resource_group(self.resource_group)
         if not self.location:
             self.location = resource_group.location
 
         disk_instance = self.get_managed_disk()
         if disk_instance is not None:
+            update_flag = True
             if self.create_option is None:
                 self.create_option = disk_instance.get('create_option')
             if self.source_uri is None:
@@ -461,7 +463,7 @@ def exec_module(self, **kwargs):
             if not disk_instance or self.is_different(disk_instance, parameter):
                 changed = True
                 if not self.check_mode:
-                    result = self.create_or_update_managed_disk(parameter)
+                    result = self.create_or_update_managed_disk(parameter, update_flag)
                 else:
                     result = True
 
@@ -593,11 +595,16 @@ def generate_managed_disk_property(self):
         disk_params['creation_data'] = creation_data
         return disk_params
 
-    def create_or_update_managed_disk(self, parameter):
+    def create_or_update_managed_disk(self, parameter, update_flag):
         try:
-            poller = self.compute_client.disks.begin_create_or_update(self.resource_group,
-                                                                      self.name,
-                                                                      parameter)
+            if update_flag:
+                poller = self.compute_client.disks.begin_update(self.resource_group,
+                                                                self.name,
+                                                                parameter)
+            else:
+                poller = self.compute_client.disks.begin_create_or_update(self.resource_group,
+                                                                          self.name,
+                                                                          parameter)
             aux = self.get_poller_result(poller)
             return managed_disk_to_dict(aux)
         except Exception as e: