Python模塊學習之Paramiko

paramiko模塊提供了ssh及sft進行遠程登錄服務器執行命令和上傳下載文件的功能。這是一個第三方的軟件包，使用之前需要安裝。

1  基於用戶名和密碼的 sshclient 方式登錄

import paramiko
import getpass

hostip = input("請輸入主機ip:")
username = input("請輸入用戶名:")
passwd = getpass.getpass("請輸入密碼:")

while True:

    cmd = input('>>:')
    if cmd == 'exit':break
    else:
         # 建立一個sshclient對象
        ssh = paramiko.SSHClient()
        # 允許將信任的主機自動加入到host_allow 列表，此方法必須放在connect方法的前面
        ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        # 調用connect方法連接服務器
        ssh.connect(hostname=hostip,port=22,username=username,password=passwd)
        stdin,stdout,stderr = ssh.exec_command(cmd)
        print(stdout.read().decode())
        ssh.close()

2 基於用戶名和密碼的 SFTP傳送文件

import paramiko

# 實例化一個trans對象# 實例化一個transport對象
trans = paramiko.Transport(('192.168.1.129',22))
trans.connect(username='xxxx',password='xxxx')
sftp = paramiko.SFTPClient.from_transport(trans)
sftp.put(localpath = r'E:\tttt\test2.py',remotepath =r'/root/test2.py')
sftp.get(localpath = r'/tmp/hello.txt',remotepath = r'hello.txt') 
trans.close()

3 實現輸入命令立馬返回結果的功能

以上操作都是基本的連接，如果我們想實現一個類似xshell工具的功能，登錄以後可以輸入命令回車後就返回結果：

import paramikoimport osimport selectimport sys# 建立一個sockettrans = paramiko.Transport(('192.168.2.129', 22))# 啓動一個客戶端trans.start_client()# 如果使用rsa密鑰登錄的話'''
default_key_file = os.path.join(os.environ['HOME'], '.ssh', 'id_rsa')
prikey = paramiko.RSAKey.from_private_key_file(default_key_file)
trans.auth_publickey(username='super', key=prikey)
'''# 如果使用用戶名和密碼登錄trans.auth_password(username='super', password='super')# 打開一個通道channel = trans.open_session()# 獲取終端channel.get_pty()# 激活終端，這樣就可以登錄到終端了，就和我們用類似於xshell登錄系統一樣channel.invoke_shell()# 下面就可以執行你所有的操作，用select實現# 對輸入終端sys.stdin和 通道進行監控,# 當用戶在終端輸入命令後，將命令交給channel通道，這個時候sys.stdin就發生變化，select就可以感知# channel的發送命令、獲取結果過程其實就是一個socket的發送和接受信息的過程while True:
    readlist, writelist, errlist = select.select([channel, sys.stdin,], [], [])    # 如果是用戶輸入命令了,sys.stdin發生變化
    if sys.stdin in readlist:        # 獲取輸入的內容
        input_cmd = sys.stdin.read(1)        # 將命令發送給服務器
        channel.sendall(input_cmd)    # 服務器返回了結果,channel通道接受到結果,發生變化 select感知到
    if channel in readlist:        # 獲取結果
        result = channel.recv(1024)        # 斷開連接後退出
        if len(result) == 0:
            print("\r\n**** EOF **** \r\n")            break
        # 輸出到屏幕
        sys.stdout.write(result.decode())
        sys.stdout.flush()# 關閉通道channel.close()# 關閉鏈接trans.close()

4 支持tab自動補全

import paramiko
import os
import select
import sys
import tty
import termios

'''
實現一個xshell登錄系統的效果，登錄到系統就不斷輸入命令同時返回結果
支持自動補全，直接調用服務器終端

'''
# 建立一個socket
trans = paramiko.Transport(('192.168.2.129', 22))
# 啓動一個客戶端
trans.start_client()

# 如果使用rsa密鑰登錄的話
'''
default_key_file = os.path.join(os.environ['HOME'], '.ssh', 'id_rsa')
prikey = paramiko.RSAKey.from_private_key_file(default_key_file)
trans.auth_publickey(username='super', key=prikey)
'''
# 如果使用用戶名和密碼登錄
trans.auth_password(username='super', password='super')
# 打開一個通道
channel = trans.open_session()
# 獲取終端
channel.get_pty()
# 激活終端，這樣就可以登錄到終端了，就和我們用類似於xshell登錄系統一樣
channel.invoke_shell()

# 獲取原操作終端屬性
oldtty = termios.tcgetattr(sys.stdin)
try:
    # 將現在的操作終端屬性設置爲服務器上的原生終端屬性,可以支持tab了
    tty.setraw(sys.stdin)
    channel.settimeout(0)

    while True:
        readlist, writelist, errlist = select.select([channel, sys.stdin,], [], [])
        # 如果是用戶輸入命令了,sys.stdin發生變化
        if sys.stdin in readlist:
            # 獲取輸入的內容，輸入一個字符發送1個字符
            input_cmd = sys.stdin.read(1)
            # 將命令發送給服務器
            channel.sendall(input_cmd)

        # 服務器返回了結果,channel通道接受到結果,發生變化 select感知到
        if channel in readlist:
            # 獲取結果
            result = channel.recv(1024)
            # 斷開連接後退出
            if len(result) == 0:
                print("\r\n**** EOF **** \r\n")
                break
            # 輸出到屏幕
            sys.stdout.write(result.decode())
            sys.stdout.flush()
finally:
    # 執行完後將現在的終端屬性恢復爲原操作終端屬性
    termios.tcsetattr(sys.stdin, termios.TCSADRAIN, oldtty)

# 關閉通道
channel.close()
# 關閉鏈接
trans.close()