Expose stdin from **kwargs

Signed-off-by: Aleksei Stepanov <penguinolog@gmail.com>

(cherry picked from commit 442ac75)
Signed-off-by: Aleksei Stepanov <penguinolog@gmail.com>

Author: penguinolog

exec_helpers/api.py

@@ -314,6 +314,7 @@ def execute(
         timeout: typing.Union[int, float, None] = constants.DEFAULT_TIMEOUT,
         *,
         log_mask_re: typing.Optional[str] = None,
+        stdin: typing.Union[bytes, str, bytearray, None] = None,
         **kwargs: typing.Any
     ) -> exec_result.ExecResult:
         """Execute command and wait for return code.
@@ -327,6 +328,8 @@ def execute(
         :param log_mask_re: regex lookup rule to mask command for logger.
                             all MATCHED groups will be replaced by '<*masked*>'
         :type log_mask_re: typing.Optional[str]
+        :param stdin: pass STDIN text to the process
+        :type stdin: typing.Union[bytes, str, bytearray, None]
         :param kwargs: additional parameters for call.
         :type kwargs: typing.Any
         :return: Execution result
@@ -337,7 +340,7 @@ def execute(
         .. versionchanged:: 2.1.0 Allow parallel calls
         """
         async_result = self.execute_async(
-            command, verbose=verbose, log_mask_re=log_mask_re, **kwargs
+            command, verbose=verbose, log_mask_re=log_mask_re, stdin=stdin, **kwargs
         )  # type: ExecuteAsyncResult
 
         result = self._exec_command(
@@ -346,6 +349,7 @@ def execute(
             timeout=timeout,
             verbose=verbose,
             log_mask_re=log_mask_re,
+            stdin=stdin,
             **kwargs
         )  # type: exec_result.ExecResult
         message = "Command {result.cmd!r} exit code: {result.exit_code!s}".format(result=result)
@@ -359,6 +363,7 @@ def __call__(
         timeout: typing.Union[int, float, None] = constants.DEFAULT_TIMEOUT,
         *,
         log_mask_re: typing.Optional[str] = None,
+        stdin: typing.Union[bytes, str, bytearray, None] = None,
         **kwargs: typing.Any
     ) -> exec_result.ExecResult:
         """Execute command and wait for return code.
@@ -372,6 +377,8 @@ def __call__(
         :param log_mask_re: regex lookup rule to mask command for logger.
                             all MATCHED groups will be replaced by '<*masked*>'
         :type log_mask_re: typing.Optional[str]
+        :param stdin: pass STDIN text to the process
+        :type stdin: typing.Union[bytes, str, bytearray, None]
         :param kwargs: additional parameters for call.
         :type kwargs: typing.Any
         :return: Execution result
@@ -380,7 +387,9 @@ def __call__(
 
         .. versionadded:: 2.9.4
         """
-        return self.execute(command=command, verbose=verbose, timeout=timeout, log_mask_re=log_mask_re, **kwargs)
+        return self.execute(
+            command=command, verbose=verbose, timeout=timeout, log_mask_re=log_mask_re, stdin=stdin, **kwargs
+        )
 
     def check_call(
         self,
@@ -392,6 +401,7 @@ def check_call(
         raise_on_err: bool = True,
         *,
         log_mask_re: typing.Optional[str] = None,
+        stdin: typing.Union[bytes, str, bytearray, None] = None,
         exception_class: "typing.Type[exceptions.CalledProcessError]" = exceptions.CalledProcessError,
         **kwargs: typing.Any
     ) -> exec_result.ExecResult:
@@ -412,6 +422,8 @@ def check_call(
         :param log_mask_re: regex lookup rule to mask command for logger.
                             all MATCHED groups will be replaced by '<*masked*>'
         :type log_mask_re: typing.Optional[str]
+        :param stdin: pass STDIN text to the process
+        :type stdin: typing.Union[bytes, str, bytearray, None]
         :param exception_class: Exception class for errors. Subclass of CalledProcessError is mandatory.
         :type exception_class: typing.Type[exceptions.CalledProcessError]
         :param kwargs: additional parameters for call.
@@ -427,7 +439,8 @@ def check_call(
         """
         expected_codes = proc_enums.exit_codes_to_enums(expected)
         result = self.execute(
-            command, verbose, timeout, log_mask_re=log_mask_re, **kwargs
+            command, verbose, timeout, log_mask_re=log_mask_re, stdin=stdin, **kwargs
+
         )  # type: exec_result.ExecResult
         if result.exit_code not in expected_codes:
             message = (
@@ -449,8 +462,9 @@ def check_stderr(
         error_info: typing.Optional[str] = None,
         raise_on_err: bool = True,
         *,
-        log_mask_re: typing.Optional[str] = None,
         expected: typing.Iterable[typing.Union[int, proc_enums.ExitCodes]] = (proc_enums.EXPECTED,),
+        log_mask_re: typing.Optional[str] = None,
+        stdin: typing.Union[bytes, str, bytearray, None] = None,
         exception_class: "typing.Type[exceptions.CalledProcessError]" = exceptions.CalledProcessError,
         **kwargs: typing.Any
     ) -> exec_result.ExecResult:
@@ -466,11 +480,13 @@ def check_stderr(
         :type error_info: typing.Optional[str]
         :param raise_on_err: Raise exception on unexpected return code
         :type raise_on_err: bool
+        :param expected: expected return codes (0 by default)
+        :type expected: typing.Iterable[typing.Union[int, proc_enums.ExitCodes]]
         :param log_mask_re: regex lookup rule to mask command for logger.
                             all MATCHED groups will be replaced by '<*masked*>'
         :type log_mask_re: typing.Optional[str]
-        :param expected: expected return codes (0 by default)
-        :type expected: typing.Iterable[typing.Union[int, proc_enums.ExitCodes]]
+        :param stdin: pass STDIN text to the process
+        :type stdin: typing.Union[bytes, str, bytearray, None]
         :param exception_class: Exception class for errors. Subclass of CalledProcessError is mandatory.
         :type exception_class: typing.Type[exceptions.CalledProcessError]
         :param kwargs: additional parameters for call.
@@ -493,6 +509,7 @@ def check_stderr(
             expected=expected,
             exception_class=exception_class,
             log_mask_re=log_mask_re,
+            stdin=stdin,
             **kwargs
         )
         append = error_info + "\n" if error_info else ""  # type: str