Pexpect 是一个用来启动子程序并对其进行自动控制的纯 Python 模块,因此在模拟人对程序的控制上表现很好,由于项目需要,看了一些关于pexpect的api文档和部分源码,这里对立面一些关键函数使用部分进行解释说明,部分内部源码会进行截取以供分析,自己的例子也会给出,主要是针对ssh远程登录,然后执行一些hive任务,将执行期间的各种状态信息获取。这里给出个例子,可以根据业务需要封装成更通用的参数式传入的函数,这里只给出一个参数写死的示例程序。
代码举例,如下所示:
from logger import logger
try:
import pxssh
import getpass
import pexpect
except Exception,e:
logger.error('your system need install pexpect module'+str(e))
def test_case3(self):
try:
connectHandler = pxssh.pxssh(timeout=100)
fpwrite = open('./hadoop_logs/test.log','a')
connectHandler.login ('192.168.10.70', 'hadoop', 'hadoop')
connectHandler.sendline ('sudo sh /home/hadoop/.verlink/testhive.sh 2>&1')
#connectHandler.sendline('sleep 5')
connectHandler.logfile = fpwrite
flag = connectHandler.expect(['password', pexpect.EOF,pexpect.TIMEOUT],timeout=2)
if flag == 2:
logger.info('[Worker] pexpect.TIMEOUT')
logger.info('[Worker] ' + str(connectHandler))
connectHandler.prompt()
elif flag == 1:
logger.info('[Worker] pexpect.EOF')
#logger.info('[Worker] ' + str(connectHandler))
connectHandler.prompt()
else:
logger.info('[Worker] COMMAND NEED PASSWORD AND HAS GIVEN AUTOMATIC')
connectHandler.sendline("hadoop")
connectHandler.prompt()
print connectHandler.before
connectHandler.logout()
print connectHandler.pid