python学习之---- paramiko 模块

paramiko 模块

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

1  基于用户名和密码的 sshclient 方式登录

执行过程：传统的连接服务器、执行命令、关闭的一个操作

 import paramiko

 # 实例化一个客户端
 ssh_client = paramiko.SSHClient()
 # 实现自动添加ssh信任关系knows_hosts
 ssh_client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
 # 连接远程linux目标机
 ssh_client.connect(hostname='ip',port=22,username='root',password='paswd')
 # 在远程服务器执行命令
 stdin,stdout,stderr = ssh_client.exec_command('df -h')
 # 读取标准输出和标准错误
 out,err = stdout.read(),stderr.read()
 # 打印输出的信息
 mess = out if out else err
 print(mess.decode())
 # 关闭客户端连接
 ssh_client.close()

2 基于用户名和密码的 transport 方式登录

执行过程：连接服务器，执行上传，下载，关闭操作

 import paramiko
 # 定义传输的ip和端口
 transfer = paramiko.Transport('10.10.2.12',22)
 # 定义传输的账号，密码
 transfer.connect(username='root',password='passwd')
 # 建立sftp连接
 sftp = paramiko.SFTPClient.from_transport(transfer)
 # 上传
 sftp.put('源文件','目标文件')
 # 下载
 sftp.get('服务器文件,'本地文件')
 # 关闭连接
 transfer.close()

3  基于公钥密钥的 SSHClient 方式登录

 # 指定本地的RSA私钥文件,如果建立密钥对时设置的有密码，password为设定的密码，如无不用指定password参数
 pkey = paramiko.RSAKey.from_private_key_file('/root/.ssh/id_rsa', password='')
 # 建立连接
 ssh = paramiko.SSHClient()
 ssh.connect(hostname='ip',port=22,username='user',pkey=pkey)
 # 执行命令
 stdin, stdout, stderr = ssh.exec_command('df -hl')
 # 结果放到stdout中，如果有错误将放到stderr中
 print(stdout.read().decode())
 # 关闭连接
 ssh.close()

4  基于密钥的 Transport 方式登录

 # 指定本地的RSA私钥文件,如果建立密钥对时设置的有密码，password为设定的密码，如无不用指定password参数
 pkey = paramiko.RSAKey.from_private_key_file('/root/.ssh/id_rsa', password='')
 # 建立连接
 trans = paramiko.Transport(('ip', 22))
 trans.connect(username='user', pkey=pkey)

 # 将sshclient的对象的transport指定为以上的trans
 ssh = paramiko.SSHClient()
 ssh._transport = trans

 # 执行命令，和传统方法一样
 stdin, stdout, stderr = ssh.exec_command('df -hl')
 print(stdout.read().decode())

 #
 sftp = paramiko.SFTPClient.from_transport(trans)
 sftp.put('src_file,'dst_file')
 sftp.get('dst_file,'src_file)
 # 关闭连接
 trans.close()

小技巧：

修改 paramiko模块的源码，将用户的输入进行文件保存

首先从github下载模块的源码包，打开 demmos\interactive.py文件，跳转到49行，添加如下：
        # 这里开始输入命令的监听
        cmd_list = []  # 用于记录用户的输入的每个键
        while True:
            r, w, e = select.select([chan, sys.stdin], [], [])  # 进入监听状态
            if chan in r:
                try:
                    x = u(chan.recv(1024))
                    if len(x) == 0:
                        sys.stdout.write('\r\n*** EOF\r\n')
                        break
                    sys.stdout.write(x)
                    sys.stdout.flush()
                except socket.timeout:
                    pass
            if sys.stdin in r:
                x = sys.stdin.read(1)  # 这里的1 代表读取用户敲击的每一个键
                if len(x) == 0:
                    break
                # 将用户输入的命令进行写文件保存，当遇到键盘输入为回车，则开发文件写入用户输入的命令，如果没有回车，则
                # 将用户键盘输入的每一个键进行列表添加
                if x == '\r':  # 为回车则写文件
                    f = open('./stdin.log','w')
                    cmd_str = ''.join(cmd_list)
                    f.write(cmd_str+'\r')
                    f.close()
                else:
                    cmd_list.append(x)
                chan.send(x)
这样就把用户的输入保存到stdin.log文件中了。