ClearML/wexpect-venv
1
0
Fork 0
mirror of https://github.com/clearml/wexpect-venv synced 2025-04-25 08:44:09 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

[ADD] First working version on python-3.x

Browse Source
This commit is contained in:
Benedek Racz 2019-05-03 13:49:05 +02:00
commit 257e9de8b4
6 changed files with 77 additions and 76 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

0
__init__.py
View File

8
examples/cmd.py
View File

@ -1,6 +1,14 @@
# A simple example code for wexpect written in python-2.7
from __future__ import print_function
import sys
import os
here = os.path.dirname(os.path.abspath(__file__))
wexpectPath = os.path.dirname(here)
sys.path.insert(0, wexpectPath)
import wexpect
# Path of cmd executable:

1
requirements.txt Normal file
View File

@ -0,0 +1 @@
pywin32>=220

14
setup.cfg
View File

@ -1,8 +1,8 @@
[metadata]
name = wexpect
author = Noah Spurrier, Richard Holden, Marco Molteni, Kimberley Burchett, Robert Stone, Hartmut Goebel, Chad Schroeder, Erick Tryzelaar, Dave Kirby, Ids vander Molen, George Todd, Noel Taylor, Nicolas D. Cesar, Alexander Gattin, Geoffrey Marshall, Francisco Lourenco, Glen Mabey, Karthik Gurusamy, Fernando Perez, Corey Minyard, Jon Cohen, Guillaume Chazarain, Andrew Ryan, Nick Craig-Wood, Andrew Stone, Jorgen Grahn, Benedek Racz
author-email = betontalpfa@gmail.com
summary = Windows alternative of pexpect
name = wexpect
author = Noah Spurrier, Richard Holden, Marco Molteni, Kimberley Burchett, Robert Stone, Hartmut Goebel, Chad Schroeder, Erick Tryzelaar, Dave Kirby, Ids vander Molen, George Todd, Noel Taylor, Nicolas D. Cesar, Alexander Gattin, Geoffrey Marshall, Francisco Lourenco, Glen Mabey, Karthik Gurusamy, Fernando Perez, Corey Minyard, Jon Cohen, Guillaume Chazarain, Andrew Ryan, Nick Craig-Wood, Andrew Stone, Jorgen Grahn, Benedek Racz
author-email = betontalpfa@gmail.com
summary = Windows alternative of pexpect
description-file = README.md
description-content-type = text/markdown; charset=UTF-8
requires-python = >=2.6
@ -15,7 +15,11 @@ classifier =
Intended Audience :: Developers
Intended Audience :: Information Technology
Operating System :: Microsoft :: Windows
Programming Language :: Python
Programming Language :: Python :: 3.3
Programming Language :: Python :: 3.4
Programming Language :: Python :: 3.5
Programming Language :: Python :: 3.6
Programming Language :: Python :: 3.7
keywords =
scripting, automation, expect, pexpect, wexpect

2
setup.py
View File

@ -6,6 +6,6 @@ setup(
setup_requires=['pbr'],
pbr=True,
packages=[''],
# packages=[''],
py_modules=['wexpect'],
)

128
wexpect.py
View File

@ -83,7 +83,7 @@ try:
import resource
import fcntl
else:
from StringIO import StringIO
from io import StringIO
try:
from ctypes import windll
import pywintypes
@ -96,11 +96,11 @@ try:
import win32api
import win32file
import winerror
except ImportError, e:
except ImportError as e:
raise ImportError(str(e) + "\nThis package requires the win32 python packages.")
screenbufferfillchar = u'\4'
screenbufferfillchar = '\4'
maxconsoleY = 8000
except ImportError, e:
except ImportError as e:
raise ImportError (str(e) + """
A critical module was not found. Probably this operating system does not
@ -233,8 +233,8 @@ def run (command, timeout=-1, withexitstatus=False, events=None, extra_args=None
else:
child = spawn(command, timeout=timeout, maxread=2000, logfile=logfile, cwd=cwd, env=env)
if events is not None:
patterns = events.keys()
responses = events.values()
patterns = list(events.keys())
responses = list(events.values())
else:
patterns=None # We assume that EOF or TIMEOUT will save us.
responses=None
@ -243,26 +243,26 @@ def run (command, timeout=-1, withexitstatus=False, events=None, extra_args=None
while 1:
try:
index = child.expect (patterns)
if type(child.after) in types.StringTypes:
if type(child.after) in (str,):
child_result_list.append(child.before + child.after)
else: # child.after may have been a TIMEOUT or EOF, so don't cat those.
child_result_list.append(child.before)
if type(responses[index]) in types.StringTypes:
if type(responses[index]) in (str,):
child.send(responses[index])
elif type(responses[index]) is types.FunctionType:
callback_result = responses[index](locals())
sys.stdout.flush()
if type(callback_result) in types.StringTypes:
if type(callback_result) in (str,):
child.send(callback_result)
elif callback_result:
break
else:
raise TypeError ('The callback must be a string or function type.')
event_count = event_count + 1
except TIMEOUT, e:
except TIMEOUT as e:
child_result_list.append(child.before)
break
except EOF, e:
except EOF as e:
child_result_list.append(child.before)
break
child_result = ''.join(child_result_list)
@ -443,7 +443,7 @@ class spawn_unix (object):
self.name = '<' + repr(self) + '>' # File-like object.
self.encoding = None # File-like object.
self.closed = True # File-like object.
self.ocwd = os.getcwdu()
self.ocwd = os.getcwd()
self.cwd = cwd
self.env = env
self.__irix_hack = (sys.platform.lower().find('irix')>=0) # This flags if we are running on irix
@ -559,7 +559,7 @@ class spawn_unix (object):
if self.use_native_pty_fork:
try:
self.pid, self.child_fd = pty.fork()
except OSError, e:
except OSError as e:
raise ExceptionPexpect('Error! pty.fork() failed: ' + str(e))
else: # Use internal __fork_pty
self.pid, self.child_fd = self.__fork_pty()
@ -619,11 +619,11 @@ class spawn_unix (object):
parent_fd, child_fd = os.openpty()
if parent_fd < 0 or child_fd < 0:
raise ExceptionPexpect, "Error! Could not open pty with os.openpty()."
raise ExceptionPexpect("Error! Could not open pty with os.openpty().")
pid = os.fork()
if pid < 0:
raise ExceptionPexpect, "Error! Failed os.fork()."
raise ExceptionPexpect("Error! Failed os.fork().")
elif pid == 0:
# Child.
os.close(parent_fd)
@ -661,7 +661,7 @@ class spawn_unix (object):
fd = os.open("/dev/tty", os.O_RDWR | os.O_NOCTTY);
if fd >= 0:
os.close(fd)
raise ExceptionPexpect, "Error! We are not disconnected from a controlling tty."
raise ExceptionPexpect("Error! We are not disconnected from a controlling tty.")
except:
# Good! We are disconnected from a controlling tty.
pass
@ -669,14 +669,14 @@ class spawn_unix (object):
# Verify we can open child pty.
fd = os.open(child_name, os.O_RDWR);
if fd < 0:
raise ExceptionPexpect, "Error! Could not open child pty, " + child_name
raise ExceptionPexpect("Error! Could not open child pty, " + child_name)
else:
os.close(fd)
# Verify we now have a controlling tty.
fd = os.open("/dev/tty", os.O_WRONLY)
if fd < 0:
raise ExceptionPexpect, "Error! Could not open controlling tty, /dev/tty"
raise ExceptionPexpect("Error! Could not open controlling tty, /dev/tty")
else:
os.close(fd)
@ -864,7 +864,7 @@ class spawn_unix (object):
if self.child_fd in r:
try:
s = os.read(self.child_fd, size)
except OSError, e: # Linux does this
except OSError as e: # Linux does this
self.flag_eof = True
raise EOF ('End Of File (EOF) in read_nonblocking(). Exception style platform.')
if s == '': # BSD style
@ -935,7 +935,7 @@ class spawn_unix (object):
return self
def next (self): # File-like object.
def __next__ (self): # File-like object.
"""This is to support iterators over a file-like object.
"""
@ -1109,7 +1109,7 @@ class spawn_unix (object):
else:
return False
return False
except OSError, e:
except OSError as e:
# I think there are kernel timing issues that sometimes cause
# this to happen. I think isalive() reports True, but the
# process is dead to the kernel.
@ -1168,7 +1168,7 @@ class spawn_unix (object):
try:
pid, status = os.waitpid(self.pid, waitpid_options)
except OSError, e: # No child processes
except OSError as e: # No child processes
if e[0] == errno.ECHILD:
raise ExceptionPexpect ('isalive() encountered condition where "terminated" is 0, but there was no child process. Did someone else call waitpid() on our process?')
else:
@ -1180,7 +1180,7 @@ class spawn_unix (object):
if pid == 0:
try:
pid, status = os.waitpid(self.pid, waitpid_options) ### os.WNOHANG) # Solaris!
except OSError, e: # This should never happen...
except OSError as e: # This should never happen...
if e[0] == errno.ECHILD:
raise ExceptionPexpect ('isalive() encountered condition that should never happen. There was no child process. Did someone else call waitpid() on our process?')
else:
@ -1247,7 +1247,7 @@ class spawn_unix (object):
if patterns is None:
return []
if type(patterns) is not types.ListType:
if type(patterns) is not list:
patterns = [patterns]
compile_flags = re.DOTALL # Allow dot to match \n
@ -1255,7 +1255,7 @@ class spawn_unix (object):
compile_flags = compile_flags | re.IGNORECASE
compiled_pattern_list = []
for p in patterns:
if type(p) in types.StringTypes:
if type(p) in (str,):
compiled_pattern_list.append(re.compile(p, compile_flags))
elif p is EOF:
compiled_pattern_list.append(EOF)
@ -1376,7 +1376,7 @@ class spawn_unix (object):
This method is also useful when you don't want to have to worry about
escaping regular expression characters that you want to match."""
if type(pattern_list) in types.StringTypes or pattern_list in (TIMEOUT, EOF):
if type(pattern_list) in (str,) or pattern_list in (TIMEOUT, EOF):
pattern_list = [pattern_list]
return self.expect_loop(searcher_string(pattern_list), timeout, searchwindowsize)
@ -1413,13 +1413,14 @@ class spawn_unix (object):
if timeout < 0 and timeout is not None:
raise TIMEOUT ('Timeout exceeded in expect_any().')
# Still have time left, so read more data
c = self.read_nonblocking (self.maxread, timeout)
c = self.read_nonblocking(self.maxread, timeout)
# c = c.decode("utf-8")
freshlen = len(c)
time.sleep (0.0001)
incoming = incoming + c
incoming += c
if timeout is not None:
timeout = end_time - time.time()
except EOF, e:
except EOF as e:
self.buffer = ''
self.before = incoming
self.after = EOF
@ -1432,7 +1433,7 @@ class spawn_unix (object):
self.match = None
self.match_index = None
raise EOF (str(e) + '\n' + str(self))
except TIMEOUT, e:
except TIMEOUT as e:
self.buffer = incoming
self.before = incoming
self.after = TIMEOUT
@ -1457,7 +1458,7 @@ class spawn_unix (object):
"""This returns the terminal window size of the child tty. The return
value is a tuple of (rows, cols). """
TIOCGWINSZ = getattr(termios, 'TIOCGWINSZ', 1074295912L)
TIOCGWINSZ = getattr(termios, 'TIOCGWINSZ', 1074295912)
s = struct.pack('HHHH', 0, 0, 0, 0)
x = fcntl.ioctl(self.fileno(), TIOCGWINSZ, s)
return struct.unpack('HHHH', x)[0:2]
@ -1479,7 +1480,7 @@ class spawn_unix (object):
# Newer versions of Linux have totally different values for TIOCSWINSZ.
# Note that this fix is a hack.
TIOCSWINSZ = getattr(termios, 'TIOCSWINSZ', -2146929561)
if TIOCSWINSZ == 2148037735L: # L is not required in Python >= 2.2.
if TIOCSWINSZ == 2148037735: # L is not required in Python >= 2.2.
TIOCSWINSZ = -2146929561 # Same bits, but with sign.
# Note, assume ws_xpixel and ws_ypixel are zero.
s = struct.pack('HHHH', r, c, 0, 0)
@ -1556,7 +1557,7 @@ class spawn_unix (object):
if self.child_fd in r:
try:
data = self.__interact_read(self.child_fd)
except OSError, e:
except OSError as e:
break
if output_filter: data = output_filter(data)
if self.logfile is not None:
@ -1587,7 +1588,7 @@ class spawn_unix (object):
while True:
try:
return select.select (iwtd, owtd, ewtd, timeout)
except select.error, e:
except select.error as e:
if e[0] == errno.EINTR:
# if we loop back we have to subtract the amount of time we already waited.
if timeout is not None:
@ -1655,7 +1656,7 @@ class spawn_windows (spawn_unix, object):
self.name = '<' + repr(self) + '>' # File-like object.
self.encoding = None # File-like object.
self.closed = True # File-like object.
self.ocwd = os.getcwdu()
self.ocwd = os.getcwd()
self.cwd = cwd
self.env = env
self.codepage = codepage
@ -1785,19 +1786,9 @@ class spawn_windows (spawn_unix, object):
This is a wrapper around Wtty.read(). """
if self.closed:
raise ValueError ('I/O operation on closed file in read_nonblocking().')
if timeout == -1:
timeout = self.timeout
@ -1971,7 +1962,7 @@ class Wtty:
if childPid:
try:
self.__childProcess = win32api.OpenProcess(win32con.PROCESS_TERMINATE | win32con.PROCESS_QUERY_INFORMATION, False, childPid)
except pywintypes.error, e:
except pywintypes.error as e:
if time.time() > ts + self.timeout:
break
else:
@ -2066,11 +2057,11 @@ class Wtty:
win32console.AttachConsole(self.conpid)
self.__consin = win32console.GetStdHandle(win32console.STD_INPUT_HANDLE)
self.__consout = self.getConsoleOut()
except Exception, e:
except Exception as e:
#e = traceback.format_exc()
try:
win32console.AttachConsole(self.__parentPid)
except Exception, ex:
except Exception as ex:
pass
#log(e)
#log(ex)
@ -2144,7 +2135,7 @@ class Wtty:
try:
if s[-1] == '\n':
s = s[:-1]
records = [self.createKeyEvent(c) for c in unicode(s)]
records = [self.createKeyEvent(c) for c in str(s)]
if not self.__consout:
return ""
consinfo = self.__consout.GetConsoleScreenBufferInfo()
@ -2228,11 +2219,8 @@ class Wtty:
strlist.append(c)
s = ''.join(strlist)
try:
return s.encode(str(self.codepage), 'replace')
except LookupError:
return s.encode(getattr(sys.stdout, 'encoding', None) or
sys.getdefaultencoding(), 'replace')
return s
def readConsoleToCursor(self):
"""Reads from the current read position to the current cursor
@ -2317,6 +2305,7 @@ class Wtty:
return s
def read_nonblocking(self, timeout, size):
"""Reads data from the console if available, otherwise
waits timeout seconds, and writes the string 'None'
@ -2348,9 +2337,7 @@ class Wtty:
end = time.clock()
timeout -= end - start
except Exception, e:
except Exception as e:
log(e)
log('End Of File (EOF) in Wtty.read_nonblocking().')
self.switchBack()
@ -2359,6 +2346,7 @@ class Wtty:
self.switchBack()
return s
def refreshConsole(self):
"""Clears the console after pausing the child and
reading all the data currently on the console."""
@ -2489,7 +2477,7 @@ class ConsoleReader:
log("Setting console output code page to %s" % cp, 'consolereader', logdir)
try:
win32console.SetConsoleOutputCP(cp)
except Exception, e:
except Exception as e:
log(e, 'consolereader', logdir)
else:
log("Console output code page: %s" % windll.kernel32.GetConsoleOutputCP(), 'consolereader', logdir)
@ -2502,7 +2490,7 @@ class ConsoleReader:
si = win32process.GetStartupInfo()
self.__childProcess, _, childPid, self.__tid = win32process.CreateProcess(None, path, None, None, False,
0, None, None, si)
except Exception, e:
except Exception as e:
log(e, 'consolereader', logdir)
time.sleep(.1)
win32api.PostThreadMessage(int(tid), win32con.WM_USER, 0, 0)
@ -2523,7 +2511,7 @@ class ConsoleReader:
time.sleep(.1)
try:
win32process.TerminateProcess(self.__childProcess, 0)
except pywintypes.error, e:
except pywintypes.error as e:
# 'Access denied' happens always? Perhaps if not
# running as admin (or UAC enabled under Vista/7).
# Don't log. Child process will exit regardless when
@ -2547,7 +2535,7 @@ class ConsoleReader:
paused = False
time.sleep(.1)
except Exception, e:
except Exception as e:
log(e, 'consolereader', logdir)
time.sleep(.1)
@ -2613,7 +2601,7 @@ class searcher_string (object):
self.eof_index = -1
self.timeout_index = -1
self._strings = []
for n, s in zip(range(len(strings)), strings):
for n, s in zip(list(range(len(strings))), strings):
if s is EOF:
self.eof_index = n
continue
@ -2634,7 +2622,7 @@ class searcher_string (object):
if self.timeout_index >= 0:
ss.append ((self.timeout_index,' %d: TIMEOUT' % self.timeout_index))
ss.sort()
ss = zip(*ss)[1]
ss = list(zip(*ss))[1]
return '\n'.join(ss)
def search(self, buffer, freshlen, searchwindowsize=None):
@ -2711,7 +2699,7 @@ class searcher_re (object):
self.eof_index = -1
self.timeout_index = -1
self._searches = []
for n, s in zip(range(len(patterns)), patterns):
for n, s in zip(list(range(len(patterns))), patterns):
if s is EOF:
self.eof_index = n
continue
@ -2732,7 +2720,7 @@ class searcher_re (object):
if self.timeout_index >= 0:
ss.append ((self.timeout_index,' %d: TIMEOUT' % self.timeout_index))
ss.sort()
ss = zip(*ss)[1]
ss = list(zip(*ss))[1]
return '\n'.join(ss)
def search(self, buffer, freshlen, searchwindowsize=None):
@ -2806,12 +2794,12 @@ def log(e, suffix='', logdir=None):
if os.path.isfile(logfile):
try:
logstat = os.stat(logfile)
except Exception, exception:
except Exception as exception:
pass
else:
try:
mtime = time.localtime(logstat.st_mtime)
except ValueError, exception:
except ValueError as exception:
# This can happen on Windows because localtime() is buggy on
# that platform. See:
# http://stackoverflow.com/questions/4434629/zipfile-module-in-python-runtime-problems
@ -2823,7 +2811,7 @@ def log(e, suffix='', logdir=None):
# do rollover
try:
os.remove(logfile)
except Exception, exception:
except Exception as exception:
pass
try:
fout = open(logfile, 'a')
@ -2852,7 +2840,7 @@ def which (filename):
if os.access (filename, os.X_OK):
return filename
if not os.environ.has_key('PATH') or os.environ['PATH'] == '':
if 'PATH' not in os.environ or os.environ['PATH'] == '':
p = os.defpath
else:
p = os.environ['PATH']
@ -2874,7 +2862,7 @@ def join_args(args):
commandline = []
for arg in args:
if re.search('[\^!$%&()[\]{}=;\'+,`~\s]', arg):
arg = '"%s"' % arg
arg = '"%s"' % arg
commandline.append(arg)
return ' '.join(commandline)