diff options
author | Matt A. Tobin <mattatobin@localhost.localdomain> | 2018-02-02 04:16:08 -0500 |
---|---|---|
committer | Matt A. Tobin <mattatobin@localhost.localdomain> | 2018-02-02 04:16:08 -0500 |
commit | 5f8de423f190bbb79a62f804151bc24824fa32d8 (patch) | |
tree | 10027f336435511475e392454359edea8e25895d /python/which/test/testsupport.py | |
parent | 49ee0794b5d912db1f95dce6eb52d781dc210db5 (diff) | |
download | UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.gz UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.lz UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.xz UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.zip |
Add m-esr52 at 52.6.0
Diffstat (limited to 'python/which/test/testsupport.py')
-rw-r--r-- | python/which/test/testsupport.py | 83 |
1 files changed, 83 insertions, 0 deletions
diff --git a/python/which/test/testsupport.py b/python/which/test/testsupport.py new file mode 100644 index 000000000..8ff519fad --- /dev/null +++ b/python/which/test/testsupport.py @@ -0,0 +1,83 @@ +#!/usr/bin/env python +# Copyright (c) 2002-2003 ActiveState Corp. +# Author: Trent Mick (TrentM@ActiveState.com) + +import os +import sys +import types + + +#---- Support routines + +def _escapeArg(arg): + """Escape the given command line argument for the shell.""" + #XXX There is a *lot* more that we should escape here. + return arg.replace('"', r'\"') + + +def _joinArgv(argv): + r"""Join an arglist to a string appropriate for running. + >>> import os + >>> _joinArgv(['foo', 'bar "baz']) + 'foo "bar \\"baz"' + """ + cmdstr = "" + for arg in argv: + if ' ' in arg: + cmdstr += '"%s"' % _escapeArg(arg) + else: + cmdstr += _escapeArg(arg) + cmdstr += ' ' + if cmdstr.endswith(' '): cmdstr = cmdstr[:-1] # strip trailing space + return cmdstr + + +def run(argv): + """Prepare and run the given arg vector, 'argv', and return the + results. Returns (<stdout lines>, <stderr lines>, <return value>). + Note: 'argv' may also just be the command string. + """ + if type(argv) in (types.ListType, types.TupleType): + cmd = _joinArgv(argv) + else: + cmd = argv + if sys.platform.startswith('win'): + i, o, e = os.popen3(cmd) + output = o.read() + error = e.read() + i.close() + e.close() + try: + retval = o.close() + except IOError: + # IOError is raised iff the spawned app returns -1. Go + # figure. + retval = -1 + if retval is None: + retval = 0 + else: + import popen2 + p = popen2.Popen3(cmd, 1) + i, o, e = p.tochild, p.fromchild, p.childerr + output = o.read() + error = e.read() + i.close() + o.close() + e.close() + retval = (p.wait() & 0xFF00) >> 8 + if retval > 2**7: # 8-bit signed 1's-complement conversion + retval -= 2**8 + return output, error, retval + + +def _rmtreeOnError(rmFunction, filePath, excInfo): + if excInfo[0] == OSError: + # presuming because file is read-only + os.chmod(filePath, 0777) + rmFunction(filePath) + +def rmtree(dirname): + import shutil + shutil.rmtree(dirname, 0, _rmtreeOnError) + + |