作為一個python小白，在查找expect的替代中發現了依云用python寫得expect.py。閱讀后，發現很多都不明白，遂決定一一百度分析一下，另外還需要更改功能，遂記錄之。
依云：expect.py
import fcntl #文件鎖
import os
import re
from select import select #異步socket
import signal
import sys
import termios #for tty io control
import tty
class Expect:
buffer = b''
encoding = sys.getdefaultencoding()
def __init__(self):
self.master, self.slave = os.openpty()
self._old_winch = signal.signal(signal.SIGWINCH, self._size_changed)
self._size_changed()
try:
os.set_inheritable(self.slave, True)
except AttributeError:
pass
def spawn(self, cmd, *, executable=None, preexec_fn=None):
if executable is None:
executable = cmd[0]
pid = os.fork()
if pid == 0:
slave = self.slave
os.dup2(slave, 0)
os.dup2(slave, 1)
os.dup2(slave, 2)
os.close(slave)
if preexec_fn is not None:
preexec_fn()
os.execvp(executable, cmd)
elif pid > 0:
os.close(self.slave)
return
else:
raise RuntimeError('os.fork() return negative value?')
def expect_line(self, regex):
if isinstance(regex, str):
regex = regex.encode(self.encoding)
if isinstance(regex, bytes):
regex = re.compile(regex)
if not isinstance(getattr(regex, 'pattern', b''), bytes):
raise TypeError('regex must be one of type str, bytes or regex of bytes')
fd = self.master
outfd = sys.stdout.fileno()
buf = self.buffer
try:
while True:
if buf:
data = buf
else:
data = os.read(fd, 4096)
os.write(outfd, data)
data = data.split(b'\n', 1)
if len(data) == 2:
data, buf = data
else:
data = data[0]
if regex.search(data):
return
finally:
self.buffer = buf
def _read_write(self, data):
fd = self.master
outfd = sys.stdout.fileno()
to_write = len(data)
while to_write:
try:
r, w, e = select([fd], [fd], [])
if r:
rdata = os.read(fd, 4096)
os.write(outfd, rdata)
if w:
written = os.write(fd, data)
data = data[written:]
to_write -= written
except InterruptedError:
continue
def send(self, data):
if isinstance(data, str):
data = data.encode(self.encoding)
self._read_write(data)
def interact(self):
pfd = sys.stdin.fileno()
cfd = self.master
outfd = sys.stdout.fileno()
old = termios.tcgetattr(pfd)
old_signal = signal.signal(signal.SIGCHLD, self._interact_leave)
tty.setraw(pfd)
try:
while True:
try:
r, w, e = select([pfd, cfd], [], [])
for fd in r:
data = os.read(fd, 4096)
other = outfd if fd == cfd else cfd
os.write(other, data)
except InterruptedError:
continue
except OSError as e:
if e.errno == 5: # Input/output error: no clients run
break
else:
raise
except ChildProcessError:
pass
finally:
termios.tcsetattr(pfd, termios.TCSAFLUSH, old)
signal.signal(signal.SIGCHLD, old_signal)
def _interact_leave(self, signum, sigframe):
while os.waitid(os.P_ALL, 0, os.WEXITED | os.WNOHANG) is not None:
pass
def _size_changed(self, signum=0, sigframe=None):
size = fcntl.ioctl(sys.stdin.fileno(), termios.TIOCGWINSZ, '1234')
fcntl.ioctl(self.master, termios.TIOCSWINSZ, size)
def __del__(self):
if self._old_winch:
signal.signal(signal.SIGWINCH, self._old_winch)
解剖
依云自己給的用法是
import sys, expect
def main(host):
p = expect.Expect()
p.spawn(['ssh',host])
p.expect_line('# ')
p.send('ls\n')
p.interact()
if __name__ == '__main__':
host = '10.17.17.17'
if len(sys.argv) == 2:
host += sys.argv[1]
main(host)
這里需要手動鍵入ssh host的密碼，且不退出當前連接。功能與我的需要有差異，就從這里入手分析，然后修改。
- 分析下類的初始化中都做了什么
os.openpty 用于打開一個新的偽終端時，返回pty與tty的文件描述符。
tty是各種類型設備的終端設備的簡稱
pty是虛擬終端。遠程telnet登陸的就是pty
signal.signal用于綁定信號處理函數，修改進程收到信號以后的行為
SIGWINCH是窗口信號改變的信號
_size_changed中調用fcntl.ioctl(sys.stdin.fileno,termios.TIOCGWINSZ,'1234')獲取當前窗口的長寬。
os.set_inheritable用于為文件標識符設定可繼承標識
fcntl
參考
2.spawn方法開啟任務
spawn函數聲明中位置參數與關鍵字參數混合的方法，起初查找時，只知道a,*a這種用法，不明白放在中間什么意思，后來查找后才明白，*號只是起到分隔的作用，*號本身并沒有被用到。
python位置參數，關鍵字參數
os.fork創建子進程
python fork
os.dup2重定向pty輸入輸出至當前輸入輸出
os.execvp開啟一個新進程，執行相應的cmd
python dup2
3.expect_line方法接收關鍵信息
isinstance判斷變量類型，支持判斷"int,float,str,bool,list,tuple,dict,bytes"。其中bytes類型是python3.x開始支持
python bytes
