python3之路 基础-paramiko

paramiko模块介绍

paramiko模块提供了基于ssh连接，进行远程登录服务器执行命令和上传下载文件的功能。这是一个第三方的软件包，使用之前需要安装。

paramiko的安装

首先确保自己的电脑上已经安装了python

执行 pip install paramiko 安装

执行 pip uninstall paramiko 卸载

paramiko的使用方法

首先简单的使用以下

__auto__ = 'GPF'
import paramiko

cmd = input("-->>:")
# 创建一个ssh客户端
ssh = paramiko.SSHClient()

# 允许连接不在know_hosts文件中的主机
# 自动添加
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy)

# 连接的用户明密码以及端口地址
ssh.connect("localhost",22,username="root",password="123")


# 执行命令
stdin,stdout,stderr=ssh.exec_command(cmd)

out,err = stdout.read(),stderr.read()

result = out if out else err
print(result.decode())
# if not stdout :
#     print(stdout.read())
# else:
#     print(stderr.read())
ssh.close()

使用RSA秘钥认证连接，免去登录密码【公钥】

__auto__ = 'GPF'

import paramiko
# /home/root/.ssh/id_rsa 公钥位置
private_key = paramiko.RSAKey.from_private_key_file("/home/root/.ssh/id_rsa")

ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh.connect("localhost", 22, username="GPF", pkey=private_key)

stdin, stdout, stderr = ssh.exec_command("cmd")

out, err = stdout.read(), stderr.read()
result = out if out else err
print(result.decode())

ssh.close()

基于SFTP远程文件传输

__auto__ = 'GPF'
import paramiko

# scp -rp  r文件或者文件夹 p权限
transport = paramiko.Transport(("localhost", 22))
transport.connect(username="gpf", password="123")

sftp = paramiko.SFTPClient.from_transport(transport)

# 将本地文件上传到服务器上
sftp.put("local_path", "remove_path")

# 将服务器上文件下载到本地
sftp.get("remove_path", "local_path")

transport.close()

封装方法，隐藏属性：

1. 创建一个conf.ini文件

   [ssh]
   host=192.168.1.1
   port=8443
   user=root
   pwd=passwd
   timeout=1000
   

2. 封装编码

__auto__ = 'GPF'

import configparser, paramik
class Demo(object):
    def __init__(self, ini_file):
        self.config = configparser.ConfigParser()
        self.config.read(ini_file)
        self.host = self.config.get('ssh', 'host')
        self.port = self.config.get('ssh', 'port')
        self.user = self.config.get('ssh', 'user')
        self.pwd = self.config.get('ssh', 'pwd')
        self.timeout = self.config.get('ssh', 'timeout')
        self.client = paramiko.SSHClient()
        self.client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        self.client.connect(hostname=self.host, port=self.port, username=self.user, password=self.pwd,
                            timeout=float(self.timeout))

    def run_ssh(self, cmd_command):
        # 执行命令，输出结果在stdout中，如果是错误则放在stderr中
        stdin, stdout, stderr = self.client.exec_command(cmd_command)
        result = stdout.read()  # read方法读取输出结果
        if len(result) == 0:  # 判断如果输出结果长度等于0表示为错误输出
            print(stderr.read().decode())
        else:
            print(str(result, 'utf-8'))

    def close(self):
        self.client.close()


if __name__ == '__main__':
    client_cmd = Demo('conf.ini')
    while True:
        cmd_input = input('>>>:')
        client_cmd.run_ssh(cmd_input)
        if cmd_input == 'quit':
            client_cmd.close()

实现一个类似xshell工具的功能，登录以后可以输入命令回车后就返回结果

import paramiko
import os
import select
import sys

# 建立一个socket
trans = paramiko.Transport(('192.168.1.1', 8443))
# 启动一个客户端
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='root', key=prikey)
'''
# 如果使用用户名和密码登录
trans.auth_password(username='root', password='passwd')
# 打开一个通道
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()

支持tab自动补全

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

'''
实现一个xshell登录系统的效果，登录到系统就不断输入命令同时返回结果
支持自动补全，直接调用服务器终端

'''
# 建立一个socket
trans = paramiko.Transport(('192.168.1.1', 8443))
# 启动一个客户端
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='root', password='passwd')
# 打开一个通道
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()