python之路-paramiko模塊

 paramiko

該模塊基於SSH用於連接遠程服務器並執行相關操作

SSHClient

遠程連接服務器有兩種方法：SSHClient、tranceport

用於連接遠程服務器並執行基本命令

基於用戶名密碼連接：

import paramiko
# 創建SSH對象
ssh = paramiko.SSHClient()
# 允許連接不在know_hosts文件中的主機
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
# 連接服務器
ssh.connect(hostname='192.168.247.131', port=22, username='root', password='111111')
# 執行命令
stdin, stdout, stderr = ssh.exec_command('pwd')
# 獲取命令結果
result = stdout.read()
print(result)
# 關閉連接
ssh.close() 

SSHClient封裝transport

import paramiko

transport = paramiko.Transport(('hostname', 22))
transport.connect(username='wupeiqi', password='123')

ssh = paramiko.SSHClient()
ssh._transport = transport

stdin, stdout, stderr = ssh.exec_command('df')
print stdout.read()

transport.close()

基於公鑰密鑰連接：

import paramiko
private_key = paramiko.RSAKey.from_private_key_file('/home/auto/.ssh/id_rsa')
# 創建SSH對象
ssh = paramiko.SSHClient()
# 允許連接不在know_hosts文件中的主機
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
# 連接服務器
ssh.connect(hostname='192.168.247.131', port=22, username='root', key=private_key)
# 執行命令
stdin, stdout, stderr = ssh.exec_command('df')
# 獲取命令結果
result = stdout.read()



# 關閉連接

ssh.close()

封裝

import paramiko

private_key = paramiko.RSAKey.from_private_key_file('/home/auto/.ssh/id_rsa')

transport = paramiko.Transport(('hostname', 22))
transport.connect(username='wupeiqi', pkey=private_key)

ssh = paramiko.SSHClient()
ssh._transport = transport

stdin, stdout, stderr = ssh.exec_command('df')

transport.close()

SFTPClient

實現文件下載的方法：Tranceport

用於連接遠程服務器並執行上傳下載

基於用戶名密碼上傳下載

import paramiko

 

transport = paramiko.Transport(('hostname',22))

transport.connect(username='wupeiqi',password='123')

 

sftp = paramiko.SFTPClient.from_transport(transport)

# 將location.py 上傳至服務器 /tmp/test.py

sftp.put('/tmp/location.py', '/tmp/test.py')

# 將remove_path 下載到本地 local_path

sftp.get('remove_path', 'local_path')

 

transport.close()

基於公鑰密鑰上傳下載

import paramiko

 

private_key = paramiko.RSAKey.from_private_key_file('/home/auto/.ssh/id_rsa')

 

transport = paramiko.Transport(('hostname', 22))

transport.connect(username='wupeiqi', pkey=private_key )

 

sftp = paramiko.SFTPClient.from_transport(transport)

# 將location.py 上傳至服務器 /tmp/test.py

sftp.put('/tmp/location.py', '/tmp/test.py')

# 將remove_path 下載到本地 local_path

sftp.get('remove_path', 'local_path')

 

transport.close()

通過以上分析，可以知道，如果想實現遠程連接執行命令，只能使用tranceport方法。

通過利用transport和SSHClient實現在一次鏈接中多次執行命令和上傳、下載文件。

import paramiko
import uuid
class SSHConnection(object):
    def __init__(self,host='192.168.247.131',port=22,username='root',pwd='111111'):
        self.host = host
        self.port = port
        self.username = username
        self.pwd = pwd
        self._k = None

    def run(self):
        self.connect()
        pass
        self.close()

    def connect(self):
        transport = paramiko.Transport((self.host,self.port))
        transport.connect(username=self.username,password=self.pwd)
        self.__transport = transport
    def close(self):
        self.__transport.close()
    def cmd(self,command):
        ssh = paramiko.SSHClient()
        ssh._transport = self.__transport
        stdin,stdout,stderr = ssh.exec_command(command)
        result = stdout.read()
        return result
    def upload(self,local_path,target_path):
        sftp = paramiko.SFTPClient.from_transport(self.__transport)
        sftp.put(local_path,target_path)

def create_file(self):
    file_name = str(uuid.uuid4())
    with open(file_name, 'w') as f:
        f.write('sb')
    return file_name
ssh = SSHConnection()
#上傳文件、執行命令
ssh.connect()
r1 = ssh.cmd('df')
print(r1)
ssh.upload('s2.py','/home/s7.py')
ssh.close()

 上面的例子是將命令寫入代碼中，並返回執行結果，下面我們就來介紹通過終端輸入命令，等待返回結果：

1.輸入命令點回車，執行命令，等待結果（linux終端默認只有點擊回車，io中的stdin才會有變化），不支持tab補全

2.每輸入一個字就給服務器傳過去，然后再點回車，服務器執行傳來的命令。（改變終端的模式）

第一個實例

 

import paramiko
import sys
import os
import socket
import select
import getpass
# from paramiko.py3compat import u   py27需要注釋
tran = paramiko.Transport(('192.168.247.128',22,))
tran.start_client()
tran.auth_password('root','111111')
#打開一個通道
chan = tran.open_session()
#獲取一個通道
chan.get_pty()
#激活器
chan.invoke_shell()
while True:
    #監視用戶輸入和服務器返回數據
    #sys.stdin 處理用戶輸入
    #chan是之前創建的通道，用於就收服務器返回信息
    readable,writeable,error = select.select([chan,sys.stdin,],[],[],1)
    if chan in readable:        #此處的chan是服務器通過執行命令，返回的結果
        try:
            #x = u(chan.recv(1024))
            x = chan.recv(1024)      #py27寫法
            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 readable:
        inp = sys.stdin.readline()
        chan.sendall(inp)
chan.close()
tran.close()

 

第二個實例

import paramiko
import sys
import os
import socket
import select
import getpass
import tty
import termios
# from paramiko.py3compat import u   py27需要注釋
tran = paramiko.Transport(('192.168.247.128',22,))
tran.start_client()
tran.auth_password('lwq','123')
#打開一個通道
chan = tran.open_session()
#獲取一個通道
chan.get_pty()
#激活器
chan.invoke_shell()
#獲取原tty屬性，目的是為了在操作完以后恢復終端原型
oldtty = termios.tcgetattr(sys.stdin)
try:
    #為tty設置新屬性
    #默認當前tty設備屬性
    #輸入一行回車，執行
    #ctrl +c 進程退出，遇到特殊字符，特殊處理
    #這是為原始模式，不認識特殊字符號
    #放置特殊字符應用在當前終端，如此設置，將所有的用戶輸入均發送到 遠程服務器
    tty.setraw(sys.stdin.fileno())
    chan.settimeout(0.0)
    while True:
        #監視用戶輸入和服務器返回數據
        #阻塞，直到句柄可讀
        readable,writeable,error = select.select([chan,sys.stdin,],[],[],1)
        if chan in readable:
            try:
                #x = u(chan.recv(1024))
                x = chan.recv(1024)      #py27寫法
                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 readable:
            inp = sys.stdin.read(1)
            if len(inp) == 0:
                break
            chan.send(inp)
finally:
    #重新設置終端屬性
    termios.tcsetattr(sys.stdin,termios.TCSADRAIN,oldtty)
chan.close()
tran.close()