pysawsim/invoke.py

   1 # Copyright (C) 2009-2010  W. Trevor King <wking@drexel.edu>
   2 #
   3 # This program is free software: you can redistribute it and/or modify
   4 # it under the terms of the GNU General Public License as published by
   5 # the Free Software Foundation, either version 3 of the License, or
   6 # (at your option) any later version.
   7 #
   8 # This program is distributed in the hope that it will be useful,
   9 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11 # GNU General Public License for more details.
  12 #
  13 # You should have received a copy of the GNU General Public License
  14 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  15 #
  16 # The author may be contacted at <wking@drexel.edu> on the Internet, or
  17 # write to Trevor King, Drexel University, Physics Dept., 3141 Chestnut St.,
  18 # Philadelphia PA 19104, USA.
  19
  20 """Functions for running external commands in subprocesses.
  21 """
  22
  23 from subprocess import Popen, PIPE
  24 import sys
  25
  26
  27 class CommandError(Exception):
  28     """Represent errors in command execution.
  29
  30     Instances are picklable (for passing through `multiprocessing.Queue`\s).
  31
  32     >>> import pickle
  33     >>> a = CommandError('somefunc', 1, '', 'could not find "somefunc"')
  34     >>> x = pickle.dumps(a)
  35     >>> b = pickle.loads(x)
  36     >>> print b
  37     Command failed (1):
  38       could not find "somefunc"
  39     <BLANKLINE>
  40     while executing
  41       somefunc
  42     """
  43     def __init__(self, command=None, status=None, stdout=None, stderr=None):
  44         self.command = command
  45         self.status = status
  46         self.stdout = stdout
  47         self.stderr = stderr
  48         Exception.__init__(self, self.__str__())
  49
  50     def __getstate__(self):
  51         return self.__dict__
  52
  53     def __setstate__(self, data):
  54         self.__dict__.update(data)
  55
  56     def __str__(self):
  57         return "\n".join([
  58                 "Command failed (%s):\n  %s\n" % (self.status, self.stderr),
  59                 "while executing\n  %s" % self.command,
  60                 ])
  61
  62
  63 def invoke(cmd_string, stdin=None, expect=(0,), cwd=None, verbose=False):
  64     """
  65     expect should be a tuple of allowed exit codes.  cwd should be
  66     the directory from which the command will be executed.
  67     """
  68     if cwd == None:
  69         cwd = "."
  70     if verbose == True:
  71         print >> sys.stderr, "%s$ %s" % (cwd, cmd_string)
  72     try :
  73         if sys.platform != "win32" and False:
  74             q = Popen(cmd_string, stdin=PIPE, stdout=PIPE, stderr=PIPE, cwd=cwd)
  75         else:
  76             # win32 don't have os.execvp() so have to run command in a shell
  77             q = Popen(cmd_string, stdin=PIPE, stdout=PIPE, stderr=PIPE,
  78                       shell=True, cwd=cwd)
  79     except OSError, e :
  80         raise CommandError(cmd_string, status=e.args[0], stdout="", stderr=e)
  81     stdout,stderr = q.communicate(input=stdin)
  82     status = q.wait()
  83     if verbose == True:
  84         print >> sys.stderr, "%d\n%s%s" % (status, stdout, stderr)
  85     if status not in expect:
  86         raise CommandError(cmd_string, status, stdout, stderr)
  87     return status, stdout, stderr