Fix execpton handle

Author: AndreMarcel99

plugins/modules/zos_fetch.py

@@ -314,11 +314,12 @@
 
 
 try:
-    from zoautil_py import datasets, mvscmd, ztypes, gdgs
+    from zoautil_py import datasets, mvscmd, ztypes, gdgs, zoau_exceptions
 except Exception:
     datasets = ZOAUImportError(traceback.format_exc())
     mvscmd = ZOAUImportError(traceback.format_exc())
     ztypes = ZOAUImportError(traceback.format_exc())
+    zoau_exceptions = ZOAUImportError(traceback.format_exc())
 
 
 class FetchHandler:
@@ -621,7 +622,7 @@ def _fetch_pdse(self, src, is_binary, temp_dir=None, encoding=None):
 
         Raises
         ------
-        fail_json
+        ZOSFetchError
             Error copying partitioned dataset to USS.
         fail_json
             Error converting encoding of the member.
@@ -635,21 +636,23 @@ def _fetch_pdse(self, src, is_binary, temp_dir=None, encoding=None):
         if is_binary:
             copy_args["options"] = "-B"
 
-        rc = datasets.copy(source=src, target=dir_path, **copy_args)
+        try:
+            rc = datasets.copy(source=src, target=dir_path, **copy_args)
 
-        if rc != 0:
+        except zoau_exceptions.ZOAUException as copy_exception:
             rmtree(dir_path)
-            self._fail_json(
+            raise ZOSFetchError(
                 msg=(
                     "Error copying partitioned data set {0} to USS. Make sure it is"
                     " not empty".format(src)
                 ),
-                stdout="",
-                stderr="Error copying partitioned data set {0} to USS. Make sure it is not empty",
-                stdout_lines="",
-                stderr_lines="Error copying partitioned data set {0} to USS. Make sure it is not empty".splitlines(),
-                rc=rc,
+                rc=copy_exception.response.rc,
+                stdout=copy_exception.response.stdout_response,
+                stderr=copy_exception.response.stderr_response,
+                stdout_lines=copy_exception.response.stdout_response.splitlines(),
+                stderr_lines=copy_exception.response.stderr_response.splitlines(),
             )
+
         if (not is_binary) and encoding:
             enc_utils = encode.EncodeUtils()
             from_code_set = encoding.get("from")
@@ -746,7 +749,7 @@ def _fetch_mvs_data(self, src, is_binary, temp_dir=None, file_override=None, enc
 
         Raises
         ------
-        fail_json
+        ZOSFetchError
             Unable to copy to USS.
         fail_json
             Error converting encoding of the dataset.
@@ -767,18 +770,20 @@ def _fetch_mvs_data(self, src, is_binary, temp_dir=None, file_override=None, enc
         if is_binary:
             copy_args["options"] = "-B"
 
-        rc = datasets.copy(source=src, target=file_path, **copy_args)
+        try:
+            rc = datasets.copy(source=src, target=file_path, **copy_args)
 
-        if rc != 0:
+        except zoau_exceptions.ZOAUException as copy_exception:
             os.remove(file_path)
-            self._fail_json(
+            raise ZOSFetchError(
                 msg="Unable to copy {0} to USS".format(src),
-                stdout="",
-                stderr="Unable to copy {0} to USS".format(src),
-                rc=rc,
-                stdout_lines="",
-                stderr_lines="Unable to copy {0} to USS".format(src),
+                rc=copy_exception.response.rc,
+                stdout=copy_exception.response.stdout_response,
+                stderr=copy_exception.response.stderr_response,
+                stdout_lines=copy_exception.response.stdout_response.splitlines(),
+                stderr_lines=copy_exception.response.stderr_response.splitlines(),
             )
+
         if (not is_binary) and encoding:
             enc_utils = encode.EncodeUtils()
             from_code_set = encoding.get("from")
@@ -1045,6 +1050,36 @@ def run_module():
     module.exit_json(**res_args)
 
 
+class ZOSFetchError(Exception):
+    def __init__(self, msg, rc="", stdout="", stderr="", stdout_lines="", stderr_lines=""):
+        """Error in a copy operation.
+
+        Parameters
+        ----------
+        msg : str
+            Human readable string describing the exception.
+        rc : int
+            Result code.
+        stdout : str
+            Standart output.
+        stderr : str
+            Standart error.
+        stdout_lines : str
+            Standart output divided in lines.
+        stderr_lines : str
+            Standart error divided in lines.
+        """
+        self.json_args = dict(
+            msg=msg,
+            rc=rc,
+            stdout=stdout,
+            stderr=stderr,
+            stdout_lines=stdout_lines,
+            stderr_lines= stderr_lines,
+        )
+        super().__init__(self.msg)
+
+
 def main():
     run_module()