概述

ssh是一個協議，OpenSSH是其中一個開源實現，paramiko是Python的一個庫，實現了SSHv2協議(底層使用cryptography)。

有了Paramiko以後，我們就可以在Python代碼中直接使用SSH協議對遠程服務器執行操作，而不是通過ssh命令對遠程服務器進行操作。今天主要介紹一下Paramiko的一些相關概念。

Paramiko介紹

paramiko包含兩個核心組件：SSHClient和SFTPClient。

• SSHClient的作用類似於Linux的ssh命令，是對SSH會話的封裝，該類封裝了傳輸(Transport)，通道(Channel)及SFTPClient建立的方法(open_sftp)，通常用於執行遠程命令。
• SFTPClient的作用類似與Linux的sftp命令，是對SFTP客戶端的封裝，用以實現遠程文件操作，如文件上傳、下載、修改文件權限等操作。

# Paramiko中的幾個基礎名詞：
1、Channel：是一種類Socket，一種安全的SSH傳輸通道；
2、Transport：是一種加密的會話，使用時會同步創建了一個加密的Tunnels(通道)，這個Tunnels叫做Channel；
3、Session：是client與Server保持連接的對象，用connect()/start_client()/start_server()開始會話。

Paramiko的基本使用

SSHClient常用的方法介紹

connect()：實現遠程服務器的連接與認證，對於該方法只有hostname是必傳參數。

常用參數
hostname 連接的目標主機
port=SSH_PORT 指定端口 

username=None 驗證的用戶名
password=None 驗證的用戶密碼
pkey=None 私鑰方式用於身份驗證
key_filename=None 一個文件名或文件列表，指定私鑰文件
timeout=None 可選的tcp連接超時時間
allow_agent=True, 是否允許連接到ssh代理，默認為True 允許
look_for_keys=True 是否在~/.ssh中搜索私鑰文件，默認為True 允許
compress=False, 是否打開壓縮

set_missing_host_key_policy()：設置遠程服務器沒有在know_hosts文件中記錄時的應對策略。目前支持三種策略：

設置連接的遠程主機沒有本地主機密鑰或HostKeys對象時的策略，目前支持三種：
 
AutoAddPolicy 自動添加主機名及主機密鑰到本地HostKeys對象，不依賴load_system_host_key的配置。即新建立ssh連接時不需要再輸入yes或no進行確認
WarningPolicy 用於記錄一個未知的主機密鑰的python警告。並接受，功能上和AutoAddPolicy類似，但是會提示是新連接
RejectPolicy 自動拒絕未知的主機名和密鑰，依賴load_system_host_key的配置。此為默認選項

exec_command()：在遠程服務器執行Linux命令的方法。

open_sftp()：在當前ssh會話的基礎上創建一個sftp會話。該方法會返回一個SFTPClient對象。

# 利用SSHClient對象的open_sftp()方法，可以直接返回一個基於當前連接的sftp對象，可以進行文件的上傳等操作.
 
sftp = client.open_sftp()
sftp.put('test.txt','text.txt')

七大案例

1.paramiko遠程密碼連接

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.使用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()

4.批量遠程密碼連接

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)

5.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)

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

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)

7.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() 

覺得有用的朋友多幫忙轉發哦！後面會分享更多devops和DBA方面的內容，感興趣的朋友可以關注下~