网络编程
位置:首页>> 网络编程>> Python编程>> Python Paramiko上传下载sftp文件及远程执行命令详解

Python Paramiko上传下载sftp文件及远程执行命令详解

作者:py3study  发布时间:2021-06-20 16:35:09 

标签:Python,Paramiko,上传下载,sftp,远程

一、简介

Paramiko模块是基于Python实现的SSH远程安全连接,用于SSH远程执行命令、文件传输等功能。

安装模块

默认Python没有自带,需要手动安装:

pip3 install paramiko

二、上传文件

#!/usr/bin/env python3
# coding: utf-8
import paramiko
def sftp_upload_file(host,user,password,server_path, local_path,timeout=10):
   """
   上传文件,注意:不支持文件夹
   :param host: 主机名
   :param user: 用户名
   :param password: 密码
   :param server_path: 远程路径,比如:/home/sdn/tmp.txt
   :param local_path: 本地路径,比如:D:/text.txt
   :param timeout: 超时时间(默认),必须是int类型
   :return: bool
   """
   try:
       t = paramiko.Transport((host, 22))
       t.banner_timeout = timeout
       t.connect(username=user, password=password)
       sftp = paramiko.SFTPClient.from_transport(t)
       sftp.put(local_path, server_path)
       t.close()
       return True
   except Exception as e:
       print(e)
       return False

测试一下上传,完整代码如下:

#!/usr/bin/env python3
# coding: utf-8
import paramiko
def sftp_upload_file(host, user, password, server_path, local_path, timeout=10):
   """
   上传文件,注意:不支持文件夹
   :param host: 主机名
   :param user: 用户名
   :param password: 密码
   :param server_path: 远程路径,比如:/home/sdn/tmp.txt
   :param local_path: 本地路径,比如:D:/text.txt
   :param timeout: 超时时间(默认),必须是int类型
   :return: bool
   """
   try:
       t = paramiko.Transport((host, 22))
       t.banner_timeout = timeout
       t.connect(username=user, password=password)
       sftp = paramiko.SFTPClient.from_transport(t)
       sftp.put(local_path, server_path)
       t.close()
       return True
   except Exception as e:
       print(e)
       return False
if __name__ == '__main__':
   host = '192.168.10.1'
   user = 'xiao'
   password = 'xiao@1234'
   server_path = '/tmp/tmp.txt'
   local_path = 'D:/text.txt'
   res = sftp_upload_file(host, user, password, server_path, local_path)
   if not res:
       print("上传文件: %s 失败"%local_path)
   else:
       print("上传文件: %s 成功" % local_path)

执行输出:

上传文件: D:/text.txt 成功

三、下载文件

def sftp_down_file(host,user,password,server_path, local_path,timeout=10):
   """
   下载文件,注意:不支持文件夹
   :param host: 主机名
   :param user: 用户名
   :param password: 密码
   :param server_path: 远程路径,比如:/home/sdn/tmp.txt
   :param local_path: 本地路径,比如:D:/text.txt
   :param timeout: 超时时间(默认),必须是int类型
   :return: bool
   """
   try:
       t = paramiko.Transport((host,22))
       t.banner_timeout = timeout
       t.connect(username=user,password=password)
       sftp = paramiko.SFTPClient.from_transport(t)
       sftp.get(server_path, local_path)
       t.close()
       return True
   except Exception as e:
       print(e)
       return False

测试一下,下载文件功能,完整代码如下:

#!/usr/bin/env python3
# coding: utf-8
import paramiko
def sftp_down_file(host,user,password,server_path, local_path,timeout=10):
   """
   下载文件,注意:不支持文件夹
   :param host: 主机名
   :param user: 用户名
   :param password: 密码
   :param server_path: 远程路径,比如:/home/sdn/tmp.txt
   :param local_path: 本地路径,比如:D:/text.txt
   :param timeout: 超时时间(默认),必须是int类型
   :return: bool
   """
   try:
       t = paramiko.Transport((host,22))
       t.banner_timeout = timeout
       t.connect(username=user,password=password)
       sftp = paramiko.SFTPClient.from_transport(t)
       sftp.get(server_path, local_path)
       t.close()
       return True
   except Exception as e:
       print(e)
       return False
if __name__ == '__main__':
   host = '192.168.10.1'
   user = 'xiao'
   password = 'xiao@1234'
   server_path = '/tmp/tmp.txt'
   local_path = 'D:/text.txt'
   res = sftp_down_file(host, user, password, server_path, local_path)
   if not res:
       print("下载文件: %s 失败"%server_path)
   else:
       print("下载文件: %s 成功" % server_path)

执行输出:

下载文件: /tmp/tmp.txt 成功

四、远程执行命令

def ssh_exec_command(host,user,password, cmd,timeout=10):
   """
   使用ssh连接远程服务器执行命令
   :param host: 主机名
   :param user: 用户名
   :param password: 密码
   :param cmd: 执行的命令
   :param seconds: 超时时间(默认),必须是int类型
   :return: dict
   """
   result = {'status': 1, 'data': None}  # 返回结果
   try:
       ssh = paramiko.SSHClient()  # 创建一个新的SSHClient实例
       ssh.banner_timeout = timeout
       # 设置host key,如果在"known_hosts"中没有保存相关的信息, SSHClient 默认行为是拒绝连接, 会提示yes/no
       ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
       ssh.connect(host, 22, user, password, timeout=timeout)  # 连接远程服务器,超时时间1秒
       stdin, stdout, stderr = ssh.exec_command(cmd,get_pty=True,timeout=timeout)  # 执行命令
       out = stdout.readlines()    # 执行结果,readlines会返回列表
       # 执行状态,0表示成功,1表示失败
       channel = stdout.channel
       status = channel.recv_exit_status()
       ssh.close()  # 关闭ssh连接
       # 修改返回结果
       result['status'] = status
       result['data'] = out
       return result
   except Exception as e:
       print(e)
       print("错误, 登录服务器或者执行命令超时!!! ip: {} 命令: {}".format(ip,cmd))return False

测试一下,远程执行命令功能,完整代码如下:

#!/usr/bin/env python3
# coding: utf-8
import paramiko
def ssh_exec_command(host,user,password, cmd,timeout=10):
   """
   使用ssh连接远程服务器执行命令
   :param host: 主机名
   :param user: 用户名
   :param password: 密码
   :param cmd: 执行的命令
   :param seconds: 超时时间(默认),必须是int类型
   :return: dict
   """
   result = {'status': 1, 'data': None}  # 返回结果
   try:
       ssh = paramiko.SSHClient()  # 创建一个新的SSHClient实例
       ssh.banner_timeout = timeout
       # 设置host key,如果在"known_hosts"中没有保存相关的信息, SSHClient 默认行为是拒绝连接, 会提示yes/no
       ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
       ssh.connect(host, 22, user, password, timeout=timeout)  # 连接远程服务器,超时时间1秒
       stdin, stdout, stderr = ssh.exec_command(cmd,get_pty=True,timeout=timeout)  # 执行命令
       out = stdout.readlines()    # 执行结果,readlines会返回列表
       # 执行状态,0表示成功,1表示失败
       channel = stdout.channel
       status = channel.recv_exit_status()
       ssh.close()  # 关闭ssh连接
       # 修改返回结果
       result['status'] = status
       result['data'] = out
       return result
   except Exception as e:
       print(e)
       print("错误, 登录服务器或者执行命令超时!!! ip: {} 命令: {}".format(ip,cmd))
       return False
if __name__ == '__main__':
   host = '192.168.10.1'
   user = 'xiao'
   password = 'xiao@1234'
   cmd = "cat /etc/issue | awk '{print $1,$2,$3}'"
   res = ssh_exec_command(host, user, password, cmd)
   # print(res)
   if not res or not res['data'] or res['status'] != 0:
       print("错误, ip: {} 执行命令: {} 失败".format(host, cmd), "red")
       exit()
   value = res['data'][0].strip()  # 获取实际值
   print("操作系统为: %s"%value)

执行输出:

操作系统为: Ubuntu 16.04.2 LTS

五、错误集锦

1. EllipticCurvePublicKey.public_bytes

Please use EllipticCurvePublicKey.public_bytes to obtain both compressed and uncompressed point encoding.

原因

paramiko 2.4.2 依赖 cryptography,而最新的cryptography==2.5里有一些弃用的API。

解决

删掉cryptography,安装2.4.2,就不会报错了。

pip uninstall cryptography
pip install cryptography==2.4.2

本文参考链接:https://www.jb51.net/article/257218.htm

 

2. Error reading SSH protocol banner

Traceback (most recent call last):
  File "/python3/lib/python3.5/site-packages/paramiko/transport.py", line 1966, in run
    self._check_banner()
  File "/python3/lib/python3.5/site-packages/paramiko/transport.py", line 2143, in _check_banner
    "Error reading SSH protocol banner" + str(e)
paramiko.ssh_exception.SSHException: Error reading SSH protocol banner
Error reading SSH protocol banner

要解决这个问题, 需要将paramiko的响应等待时间调长。  修改paramiko/transport.py文件中的

self.banner_timeout

值, 将其设为300或者其他较长的值即可解决这个问题。

来源:https://cloud.tencent.com/developer/article/1592375

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com