In cluster_shutdown module fix exception handling

Author: PolonaM

plugins/module_utils/cluster.py

@@ -11,7 +11,7 @@
 from time import sleep
 from ..module_utils.utils import PayloadMapper
 from ..module_utils.rest_client import RestClient
-from ..module_utils.errors import ScaleComputingError
+from ..module_utils.errors import ScaleComputingError, ScaleTimeoutError
 from ..module_utils.typed_classes import TypedClusterToAnsible, TypedTaskTag
 from typing import Any
 
@@ -84,18 +84,14 @@ def shutdown(
                 dict(forceShutdown=force_shutdown),
                 check_mode,
             )
-        except ScaleComputingError as e:
-            # To avoid timeout when there are a lot of VMs to shutdown
-            if "Request timed out" in str(e):
-                try:
-                    while True:
-                        # get cluster until "[Errno 111] Connection refused"
-                        Cluster.get(rest_client)
-                        sleep(5)
-                except ScaleComputingError as e2:
-                    if "Connection refused" in str(e2):  # cluster is shutdown
-                        pass
-                    else:
-                        raise e2  # Some other unexpected error
-            else:
-                raise e  # UnexpectedAPIResponse error
+        # To avoid timeout when there are a lot of VMs to shutdown
+        except ScaleTimeoutError:
+            pass
+
+        try:
+            while True:
+                # get cluster until "[Errno 111] Connection refused"
+                Cluster.get(rest_client)
+                sleep(5)
+        except ConnectionRefusedError:
+            pass