簡介
最近需要遠(yuǎn)程操作一個服務(wù)器并執(zhí)行該服務(wù)器上的一個python腳本,查到可以使用Pexpect這個庫。記錄一下。
什么是Pexpect?Pexpect能夠產(chǎn)生子應(yīng)用程序,并控制他們,并能夠通過期望模式對子應(yīng)用的輸出做出反應(yīng)。Pexpect允許你的腳本產(chǎn)生子應(yīng)用、控制他們像一個人類在輸入命令一樣。
Pexpect使用在自動交互的應(yīng)用,例如SSH、SFTP、PASSWD、TELNET。它可以被應(yīng)用在使用自動設(shè)置腳本為不同的服務(wù)器自動地重復(fù)的安裝軟件包。也可以被應(yīng)用在自動的軟件測試。
Pexpect的主要特點(diǎn)是需要Python的基本庫pty,這個庫只有在類Unix系統(tǒng)上才有
Pexpect關(guān)于SSH的使用
注:測試,我們直接用虛擬機(jī)本機(jī)ssh本機(jī)
環(huán)境
1. win10 物理機(jī)
2. Vmware Centos 虛擬機(jī)
3. Xshell
4. 虛擬機(jī)python安裝pexpect:pip install pexpect
在虛擬機(jī)創(chuàng)建一個 python文件
#-*- coding:UTF-8 -*- import pexpect # 定義ssh連接 def ssh(user,host,password,command): #創(chuàng)建子應(yīng)用,命令是 ssh -l root 127.0.0.1 python /home/python/test.py child = pexpect.spawn('ssh -l %s %s %s'%(user,host,command)) # 期待開啟的子程序的顯示,子程序的不同顯示會匹配到不同key然后我們定義不同的操作 # 0 : 連接超時 # 1 :ssh有時候提示你是否確認(rèn)連接 # 2 :提示輸入密碼 # 3 :匹配到#號,表示命令已經(jīng)執(zhí)行完畢。沒用到 i = child.expect([pexpect.TIMEOUT, 'Are you sure you want to continue connecting','password:',r"([^-]>|#)"]) # 如果登錄超時,renturn none if i == 0: # Timeout print "Timeout" return None # 提示是否確認(rèn)連接 if i == 1: child.sendline ('yes') # 我們輸入yes child.expect ('password: ')# 輸入yes后 子程序應(yīng)該提示輸入密碼,我們再次期待password i = child.expect([pexpect.TIMEOUT, 'password: ']) #超時 if i == 0: # Timeout return None # 不考慮其他情況,走到此處時,要么timeout 已經(jīng)return ,要么等待輸入密碼 #輸入密碼 child.sendline(password) # 返回子程序 return child if __name__ =='__main__': try: # 配置數(shù)據(jù) host='127.0.0.1' user="root" password = '********' command = 'python /home/python/test.py' #command="ls -l" child = ssh(user,host,password,command) #這句是將子程序的命令行拉到末端 test = child.expect(pexpect.EOF) #child中before就是我們要的數(shù)據(jù),有時候還會在 after中 print child.before print child.after except Exception,e: print str(e) # 最終的顯示結(jié)果是 test.py中打印的hahaha結(jié)果, [root@localhost python]# python test_pexpect.py hahaha