python模塊-paramiko

paramiko

COPY FROM https://blog.csdn.net/forever_wen/article/details/82556154

paramiko 是一個用於做遠程控制的模塊，使用該模塊可以對遠程服務器進行命令或文件操作,paramiko是用python語言寫的一個模塊，遵循SSH2協議，支持以加密和認證的方式，進行遠程服務器的連接。由於使用的是python這樣的能夠跨平台運行的語言，所以所有python支持的平台，如Linux, Solaris, BSD, MacOS X, Windows等，paramiko都可以支持，因此，如果需要使用SSH從一個平台連接到另外一個平台，進行一系列的操作時，paramiko是最佳工具之一

• 安裝
  由於 paramiko 模塊內部依賴pycrypto，所以先下載安裝pycrypto
  yum -y install python-devel
  pip3 install pycrypto
  pip3 install paramiko

• 代碼案例

1. 遠程連接

import paramiko

##1.創建一個ssh對象
client = paramiko.SSHClient()

#2.解決問題:如果之前沒有，連接過的ip，會出現選擇yes或者no的操作，
##自動選擇yes
client.set_missing_host_key_policy(paramiko.AutoAddPolicy())

#3.連接服務器
client.connect(hostname='172.25.254.31',
               port=22,
               username='root',
               password='westos')

#4.執行操作
stdin,stdout, stderr = client.exec_command('hostname')

#5.獲取命令執行的結果
result=stdout.read().decode('utf-8')
print(result)

#6.關閉連接
client.close()

2. 使用sftp上傳文件

import paramiko
#獲取Transport實例
tran = paramiko.Transport("172.25.254.31",22)
#連接SSH服務端
tran.connect(username = "root", password = "westos")
#獲取SFTP實例
sftp = paramiko.SFTPClient.from_transport(tran)
#設置上傳的本地/遠程文件路徑
localpath="passwd.html"   ##本地文件路徑
remotepath="/home/kiosk/Desktop/fish"   ##上傳對象保存的文件路徑
#執行上傳動作
sftp.put(localpath,remotepath)

tran.close()

3. 使用sftps遠程讀取Linux服務器上的文件方法

import paramiko
 
#服務器信息，主機名（IP地址）、端口號、用戶名及密碼
hostname = ""
port = 22
username = ""
password = ""
 
 
client = paramiko.SSHClient()
client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
client.connect(hostname, port, username, password, compress=True)
sftp_client = client.open_sftp()
remote_file = sftp_client.open("/home/...txt")#文件路徑
try:
  for line in remote_file:
    print(line)
finally:
  remote_file.close()

4. 使用sftp下載文件

import paramiko

#獲取SSHClient實例
client = paramiko.SSHClient()
client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
#連接SSH服務端
client.connect("172.25.254.31",username="root",password="westos")
#獲取Transport實例
tran = client.get_transport()
#獲取SFTP實例
sftp = paramiko.SFTPClient.from_transport(tran)

remotepath='/home/kiosk/Desktop/fish'
localpath='/home/kiosk/Desktop/fish'

sftp.get(remotepath, localpath)

client.close()

5. 批量遠程密碼連接

from paramiko.ssh_exception import NoValidConnectionsError
from paramiko.ssh_exception import AuthenticationException


def connect(cmd,hostname,port=22,username='root',passwd='westos'):
    import paramiko

    ##1.創建一個ssh對象
    client = paramiko.SSHClient()

    #2.解決問題:如果之前沒有，連接過的ip，會出現選擇yes或者no的操作，
    ##自動選擇yes
    client.set_missing_host_key_policy(paramiko.AutoAddPolicy())

    #3.連接服務器
    try:
        client.connect(hostname=hostname,
                       port=port,
                       username=username,
                       password=passwd)
        print('正在連接主機%s......'%(hostname))
    except NoValidConnectionsError as e:   ###用戶不存在時的報錯
        print("連接失敗")
    except AuthenticationException as t:   ##密碼錯誤的報錯
        print('密碼錯誤')

    else:
        #4.執行操作
        stdin,stdout, stderr = client.exec_command(cmd)

        #5.獲取命令執行的結果
        result=stdout.read().decode('utf-8')
        print(result)

        #6.關閉連接
    finally:
        client.close()

with open('ip.txt') as f:  #ip.txt為本地局域網內的一些用戶信息
    for line in f:
        line = line.strip()   ##去掉換行符
        hostname,port,username,passwd= line.split(':')
        print(hostname.center(50,'*'))
        connect('uname', hostname, port,username,passwd)

6. paramiko基於公鑰密鑰連接

import  paramiko

from paramiko.ssh_exception import NoValidConnectionsError, AuthenticationException


def connect(cmd, hostname, port=22, user='root'):

    client = paramiko.SSHClient()  

    private_key = paramiko.RSAKey.from_private_key_file('id_rsa')
     ###id_rsa為本地局域網密鑰文件

    client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    try:
        client.connect(hostname=hostname,
                       port=port,
                       username=user,
                       pkey=private_key
                      )

        stdin, stdout, stderr = client.exec_command(cmd)
    except NoValidConnectionsError as e:
        print("連接失敗")
    except AuthenticationException as e:
        print("密碼錯誤")
    else:

        result = stdout.read().decode('utf-8')
        print(result)
    finally:

        client.close()
for count in range(254):
    host = '172.25.254.%s' %(count+1)
    print(host.center(50, '*'))
    connect('uname', host)

7. 基於密鑰的上傳和下載

import paramiko
private_key = paramiko.RSAKey.from_private_key_file('id_rsa')
tran = paramiko.Transport('172.25.254.31',22)
tran.connect(username='root',password='westos')
#獲取SFTP實例
sftp = paramiko.SFTPClient.from_transport(tran)
remotepath='/home/kiosk/Desktop/fish8'
localpath='/home/kiosk/Desktop/fish1'
sftp.put(localpath,remotepath)
sftp.get(remotepath, localpath)

8. paramiko的再封裝

import os
import paramiko
from paramiko.ssh_exception import NoValidConnectionsError, AuthenticationException, SSHException


class SshRemoteHost(object):
    def __init__(self, hostname, port, user, passwd, cmd):
        self.hostname  = hostname
        self.port = port
        self.user = user
        self.passwd = passwd
        self.cmd = cmd
    def run(self):
        """默認調用的內容"""
        # cmd hostname
        # put
        # get
        cmd_str =  self.cmd.split()[0] # cmd
        # 類的反射，判斷類里面是否可以支持該操作
        if hasattr(self, 'do_'+ cmd_str):  # do_cmd
            getattr(self, 'do_'+cmd_str)()
        else:
            print("目前不支持該功能")

    def do_cmd(self):
        client = paramiko.SSHClient()

        client.set_missing_host_key_policy(paramiko.AutoAddPolicy())

        try:

            client.connect(hostname=self.hostname,
                           port=self.port,
                           username=self.user,
                           password=self.passwd)

            print("正在連接%s......." % (self.hostname))
        except NoValidConnectionsError as e:
            print("連接失敗")
        except AuthenticationException as e:
            print("密碼錯誤")
        else:
            # 4. 執行操作
            cmd = ''.join(self.cmd.split()[1:])  ##將輸入的后面的取出，作為
            stdin, stdout, stderr = client.exec_command(cmd)

            # 5.獲取命令執行的結果
            result = stdout.read().decode('utf-8')
            print(result)
        finally:
            # 6.關閉連接
            client.close()

    def do_put(self):
        ###put /tmp/passwd  ###將本地的/tmp/passwd上傳到遠端/tmp/passwd
        print('正在上傳...')
        try:
            #獲取Transport實例
            tran = paramiko.Transport(self.hostname,int(self.port))  ##由於端口為整形，而我們用split方法得到的是str
            #連接SSH服務端
            tran.connect(username = self.user, password = self.passwd)
        except SSHException as e:
            print('連接失敗')
        else:
            #獲取SFTP實例
            sftp = paramiko.SFTPClient.from_transport(tran)
            newCmd = self.cmd.split()[1:]
            if len(newCmd) == 2:

                #設置上傳的本地/遠程文件路徑
                localpath=newCmd[0]
                remotepath=newCmd[1]
                #執行上傳動作
                sftp.put(localpath,remotepath)
                print('%s文件上傳到%s主機的%s文件成功' %(localpath,self.hostname,remotepath))
            else:
                print('上傳文件信息錯誤')

                tran.close()


    def do_get(self):

        print('正在下載...')
        try:
            # 獲取Transport實例
            tran = paramiko.Transport(self.hostname, int(self.port))  ##由於端口為整形，而我們用split方法得到的是str
            # 連接SSH服務端
            tran.connect(username=self.user, password=self.passwd)
        except SSHException as e:
            print('連接失敗')
        else:
            # 獲取SFTP實例
            sftp = paramiko.SFTPClient.from_transport(tran)
            newCmd = self.cmd.split()[1:]
            if len(newCmd) == 2:

                # 設置下載的本地/遠程文件路徑
                localpath = newCmd[1]
                remotepath = newCmd[0]
                # 執行上傳動作
                sftp.get( remotepath,localpath)
                print('%s主機的%s文件下載到%s文件成功' % (self.hostname,remotepath,localpath))
            else:
                print('上傳文件信息錯誤')

                tran.close()


import paramiko
import os
# 1.選擇操作的主機組:eg:mysql,web,ftp
groups=[file.rstrip('.conf') for  file in os.listdir('conf')]
print("主機組顯示：".center(50,'*'))
for group in groups:
    print('\t',group)
choiceGroup = input("選擇批量操作的主機組(eg:mysql):")


##2.根據選擇的主機組，顯示包含的主機IP/主機名
# 1).打開文件conf/choiceGroup.conf
# 2).依次讀取文件每一行
# 3).只拿出
print("主機組包含的主機：".center(50,'*'))
with open('conf/%s.conf' %(choiceGroup)) as f:
    for line in f:
        print(line.split(':')[0])
    f.seek(0,0)  ##把指針移動到文件最開始
    hostinfos = [line.strip() for line in f.readlines()]

###3.讓用戶確認信息，選擇需要批量執行的命令；
##      -cmd shell 命令
##      -put 本地文件  遠程文件
##      -get 遠程文件  本地文件
print("批量執行腳本".center(50,"*"))
while True:
    cmd = input('>>:').strip()
    if cmd :
        if cmd == 'exit' or cmd == "quit":
            print("執行完畢，正在退出")
            break
        for info in hostinfos:
            host,port,user,passwd = info.split(':')
            clientObj = SshRemoteHost(host,port,user,passwd,cmd)
            clientObj.run()