diff --git a/wetb/functions/process_exec.py b/wetb/functions/process_exec.py index 509459eb9afb097b9c30ccec1a5edc86920f48e8..caeb49750b88a33e96961aec34bef5eacc9f6c7f 100644 --- a/wetb/functions/process_exec.py +++ b/wetb/functions/process_exec.py @@ -5,6 +5,7 @@ Created on 10/03/2014 ''' import os +import psutil DEBUG = False def pexec(args, cwd=None): @@ -28,3 +29,26 @@ def pexec(args, cwd=None): errorcode = proc.returncode return errorcode, stdout.decode(), stderr.decode(), cmd + + +def process(args, cwd=None): + import subprocess + if not isinstance(args, (list, tuple)): + args = [args] + args = [str(arg) for arg in args] + for i in range(len(args)): + if os.path.exists(args[i]): + args[i] = str(args[i]).replace('/', os.path.sep).replace('\\', os.path.sep).replace('"', '') + + cmd = "%s" % '{} /c "{}"'.format (os.environ.get("COMSPEC", "cmd.exe"), subprocess.list2cmdline(args)) + if cwd is not None and os.path.isfile(cwd): + cwd = os.path.dirname(cwd) + return subprocess.Popen(args, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=False, cwd=cwd) + +def exec_process(process): + stdout, stderr = process.communicate() + errorcode = process.returncode + + return errorcode, stdout.decode(), stderr.decode() + +