python的paramiko模塊

    paramiko是用python語言寫的一個模塊，遵循SSH2協議，支持以加密和認證的方式，進行遠程服務器的連接。paramiko支持Linux, Solaris, BSD, MacOS X, Windows等平台通過SSH從一個平台連接到另外一個平台。利用該模塊，可以方便的進行ssh連接和sftp協議進行sftp文件傳輸。

 

paramiko常用的類與方法：

 

1、SSHClient類

SHClient類是SSH服務會話的高級表示，封裝了傳輸、通道以及SFTPClient的校驗、建立方法，通常用於執行命令。

 

1）connect方法

connect(self,hostname,port=22,username=None,password=None,pkey=None,key_filename=None,timeout=None,allow_agent=True,look_for_keys=True,compress=False)

參數說明：

hostname：連接目標的主機地址

port：連接目錄的端口，默認為22

username：用戶名

password：密碼

pkey：私鑰方式用戶驗證

key_filename：私鑰文件名

timeout：連接超時時間

allow_agent：是否允許使用ssh代理

look_for_keys：是否允許搜索私鑰文件

compress：打開時是否壓縮

 

2）exec_command方法

exec_command(self,command,bufsize=-1)

參數說明：

command：執行的的指令

bufsize：文件緩沖區大小，-1不限制

 

3）load_system_host_keys方法

load_system_host_keys(self,filename=None)

參數說明：

filename：指定遠程主機的公鑰文件，默認為.ssh目錄下的known_hosts文件

 

4）set_missing_host_key_policy方法

ssh = paramiko.SSHClient()

ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())

參數說明：

AutoAddPolicy：自動添加主機名及密鑰到本地並保存，不依賴load_system_host_keys()配置，即如果known_hosts里沒有遠程主機的公鑰時，默認連接會提示yes/no，自動yes

RejectPolicy：自動拒絕未知主機名和密鑰，依賴load_system_host_keys()

WarnningPlicy：功能與AutoAddPolicy相同，但是未知主機會提示yes/no

 

2、SFTPClient類

根據SSH傳輸協議的sftp會話，實現遠程文件上傳、下載等操作。

 

1）from_transport方法

 

classmethod from_transport(cls,t)

參數說明：

t：一個已通過驗證的傳輸對象

 

示例：

>>> import paramiko
>>> a = paramiko.Transport((“127.0.0.1″,2222))
>>> a.connect(username=”root”, password=’123456′)
>>> sftp = paramiko.SFTPClient.from_transport(a)

 

2)put方法

 

put(self,localpath,remotepath,callback=None,confirm=True)

參數說明：

localpath：上傳源文件的本地路徑

remotepath：目標路徑

callback：獲取接收與總傳輸字節數

confirm：上傳完畢后是否調用stat()方法，以便確認文件大小

 

示例：

>>> localpath=’ftp-test.log’
>>> remotepath=’/data/ftp-test.log’
>>> sftp.put(localpath,remotepath)

 

 

3）get方法

 

get(self, remotepath, localpath, callback=None)

參數說明：

remotepath：需要下載的遠程文件

localpath：本地存儲路徑

callback：同put方法

 

4）其他方法

 

mkdir：用於創建目錄

remove：刪除目錄

rename：重命名

stat：獲取文件信息

listdir：獲取目錄列表

 

代碼示例

Paramiko ssh客戶端：

#!/usr/bin/python
import paramiko
import os,sys
ssh_host = sys.argv[1]
ssh_port = 22
user = 'root'
password = 'xxxxxx'
cmd = sys.argv[2]
paramiko.util.log_to_file('/tmp/test')    #使用paramiko記錄日志
s = paramiko.SSHClient()    #綁定一個實例
s.load_system_host_keys()    #加載known_hosts文件
s.set_missing_host_key_policy(paramiko.AutoAddPolicy())    #遠程連接如果提示yes/no時，默認為yes
s.connect(ssh_host,ssh_port,user,password,timeout=5)    #連接遠程主機
stdin,stdout,stderr = s.exec_command(cmd)    #執行指令，並將命令本身及命令的執行結果賦值到標准辦入、標准輸出或者標准錯誤
cmd_result = stdout.read(),stderr.read()    #取得執行的輸出
for line in cmd_result:
　　print line
s.close()

 

使用ssh密鑰連接：

pkey_file = '/home/breeze/.ssh/id_rsa'
key = paramiko.RSAKey.from_private_key_file(pkey_file)
s.connect(host,port,username,pkey=key,timeout=5)
stdin,stdout,stderr = s.exec_command(cmd)

 

Paramiko SFTP傳送文件：

#!/usr/bin/python
import os,sys
import paramiko
host = sys.argv[1]
rfilename = sys.argv[2]
lfilename = os.path.basename(rfilename)
user = 'root'
password = 'xxxx'
paramiko.util.log_to_file('/tmp/test') 
t = paramiko.Transport((host,22))
t.connect(username=user,password=password)
sftp = paramiko.SFTPClient.from_transport(t)
sftp.get(rfilename,lfilename)
#sftp.put('paramiko1.py','/tmp/paramiko1.py')
t.close()

 

 

使用interactive模塊實現SSH交互：

interactive.py模塊內容如下：

#!/usr/bin/python
import socket
import sys   
# windows does not have termios...
try:
import termios
import tty
    has_termios = True
except ImportError:
    has_termios = False
def interactive_shell(chan):
if has_termios:
        posix_shell(chan)
else:
        windows_shell(chan)   
def posix_shell(chan):
import select
    oldtty = termios.tcgetattr(sys.stdin)
try:
        tty.setraw(sys.stdin.fileno())
        tty.setcbreak(sys.stdin.fileno())
        chan.settimeout(0.0)
while True:
            r, w, e = select.select([chan, sys.stdin], [], [])
if chan in r:
try:
                    x = chan.recv(1024)
if len(x) == 0:
print '\r\n*** EOF\r\n',
break
sys.stdout.write(x)
                    sys.stdout.flush()
except socket.timeout:
pass
            if sys.stdin in r:
                x = sys.stdin.read(1)
if len(x) == 0:
break
chan.send(x)
finally:
        termios.tcsetattr(sys.stdin, termios.TCSADRAIN, oldtty)    
# thanks to Mike Looijmans for this code
def windows_shell(chan):
import threading
    sys.stdout.write("Line-buffered terminal emulation. Press F6 or ^Z to send EOF.\r\n\r\n")         
def writeall(sock):
while True:
            data = sock.recv(256)
if not data:
                sys.stdout.write('\r\n*** EOF ***\r\n\r\n')
                sys.stdout.flush()
break
sys.stdout.write(data)
            sys.stdout.flush()       
    writer = threading.Thread(target=writeall, args=(chan,))
    writer.start()    
try:
while True:
            d = sys.stdin.read(1)
if not d:
break
chan.send(d)
except EOFError:
# user hit ^Z or F6
pass

 

inter_ssh.py交互腳本如下：

#!/usr/bin/python
#_*_coding:utf8_*_
import paramiko
import interactive
#記錄日志
paramiko.util.log_to_file('/tmp/test')
#建立ssh連接
ssh=paramiko.SSHClient()
ssh.load_system_host_keys()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh.connect('192.168.128.82',port=22,username='root',password='cheyian')
#建立交互式shell連接
channel=ssh.invoke_shell()
#建立交互式管道
interactive.interactive_shell(channel)
#關閉連接
channel.close()
ssh.close()