使用os.forkpty（）創建一個僞終端ssh到遠程服務器，並與它通信

Question

我想寫一個python腳本，可以ssh到遠程服務器，並可以執行簡單的命令，如ls，cd從python客戶端。但是，在成功將ssh傳入服務器後，我無法從僞終端讀取輸出。任何人都可以在這裏幫助我，這樣我就可以在服務器上執行一些命令。

下面是示例代碼：

#!/usr/bin/python2.6 
import os,sys,time,thread 
pid,fd = os.forkpty() 
if pid == 0: 
    os.execv('/usr/bin/ssh',['/usr/bin/ssh','user@host',]) 
    sys.exit(0) 
else: 
    output = os.read(fd,1024) 
    print output 
    data = output 
    os.write(fd,'password\n') 
    time.sleep(1) 
    output = os.read(fd,1024) 
    print output 
    os.write(fd,'ls\n') 
    output = os.read(fd,1024) 
    print output 

輸出示例：

user@host's password: 

Last login: Wed Aug 24 03:16:57 2011 from 1x.x.x.xxxx 

-bash: ulimit: open files: cannot modify limit: Operation not permitted 
host: /home/user>ls 

Answer 1

我建議嘗試模塊Pexpect的，而這正是專爲這樣的事情（與其他應用程序接口通過僞TTY）或Fabric，這是爲了更抽象地構建此類事物（使用SSH在遠程服務器上自動執行系統管理任務）。

Pexpect的：http://pypi.python.org/pypi/pexpect/

面料：http://docs.fabfile.org/en/1.11/

Answer 2

如前所述，最好使用公共密鑰。正如我通常使用它們，我已經改變了你的程序，以便它在這裏工作。

#!/usr/bin/python2.6 
import os,sys,time,thread 
pid,fd = os.forkpty() 
if pid == 0: 
    os.execv('/usr/bin/ssh',['/usr/bin/ssh','localhost',]) 
    sys.exit(0) 
else: 
    output = os.read(fd,1024) 
    print output 
    os.write(fd,'ls\n') 
    time.sleep(1) # this is new! 
    output = os.read(fd,1024) 
    print output 

隨着加入sleep(1)，我給遠程主機（或者，在我的情況下，不那麼遠程主機）的時間來處理ls命令併產生其輸出。

如果您發送ls並立即閱讀，您只能閱讀當前的內容。也許你應該閱讀一個循環左右。

，或者你應該這樣來做：

import subprocess 
sp = subprocess.Popen(("ssh", "localhost", "ls"), stdout=subprocess.PIPE) 
print sp.stdout.read()