tools/dftestlib/util.py

   1 # Copyright (c) 2013 by Gilbert Ramirez <gram@alumni.rice.edu>
   2 #
   3 # $Id$
   4 #
   5 # This program is free software; you can redistribute it and/or
   6 # modify it under the terms of the GNU General Public License
   7 # as published by the Free Software Foundation; either version 2
   8 # of the License, or (at your option) any later version.
   9 #
  10 # This program is distributed in the hope that it will be useful,
  11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 # GNU General Public License for more details.
  14 #
  15 # You should have received a copy of the GNU General Public License
  16 # along with this program; if not, write to the Free Software
  17 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18
  19
  20 import subprocess
  21
  22 SUCCESS = 0
  23 def exec_cmdv(cmdv, cwd=None, stdin=None):
  24     """Run the commands in cmdv, returning (retval, output),
  25     where output is stdout and stderr combined.
  26     If cwd is given, the child process runs in that directory.
  27     If a filehandle is passed as stdin, it is used as stdin.
  28     If there is an OS-level error, None is the retval."""
  29
  30     try:
  31         output = subprocess.check_output(cmdv, stderr=subprocess.STDOUT,
  32                 cwd=cwd, stdin=stdin)
  33         retval = SUCCESS
  34
  35     # If file isn't executable
  36     except OSError, e:
  37         output = str(e)
  38         retval = None
  39
  40     # If process returns non-zero
  41     except subprocess.CalledProcessError, e:
  42         output = e.output
  43         retval = e.returncode
  44
  45     return (retval, output)
  46