Python模块学习之Paramiko

Posted 2020-10-03

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