使paramiko库执行命令时在给定的时间强制退出功能的实现


Posted in Python onMarch 03, 2021

使用paramiko库ssh连接到远端云主机上时,非常偶现卡死现象,连接无法退出(可以是执行命令时云主机重启等造成)。需要给定一段时间,不管命令执行是否卡住,都退出连接,显示命令执行超时错误。

实现方式:

线程+事件,在线程中执行ssh命令,给事件配置超时时间。

代码示例:

1 from threading import Thread, Event

2 import paramiko

class SshClient(object):

  def __init__(self, ip, port, username, password):
    self.ip = ip
    self.host = host
    self.username = username
    self.password = password

  def exec_command(cmd, timeout):
    log.info(u"在ip:%s上执行命令%s" % (self.ip, cmd))
    sc = paramiko.SSHClient()
    sc.set_missing_host_key_policy(paramiko.AutoAddPolicy())

    # 用来接收stdout stderror status信息
    res = [None, None, None]

    def get_return(start_event, res_list):
      _, cmd_stdout, cmd_stderr = sc.exec_command(command=cmd, timeout=timeout)
      channel = cmd_stdout.channel
      cmd_status = channel.recv_exit_status()
      res_list[0] = cmd_stdout
      res_list[1] = cmd_stderr
      res_list[2] = cmd_status
      start_event.set()  # 表示线程已经执行完毕

    try:
      sc.connect(hostname=self.ip, port=self.port, username=self.username, password=self.password, timeout=30)  # 这里的timeout是连接使用的,与我们要的不同
      start_evt = Event()
      t = Thread(target=get_return, args=(start_evt, res))
      t.start()
      start_evt.wait(timeout=timeout)
      # 执行到这里说明线程已经退出
      if None in res:
        raise Exception(u"命令超时退出")
      stdout, stderr, status = res
      if status != 0:
        raise Exception(u"命令执行返回非0!返回值为%s,错误信息为%s" % (status, stdout.read() + stderr.read()))
      return stdout.read() + stderr.read()
    finally:
      sc.close()
}

知识点补充:

python paramiko的使用介绍

一: 使用paramiko

#设置ssh连接的远程主机地址和端口
t=paramiko.Transport((ip,port))
#设置登录名和密码
t.connect(username=username,password=password)
#连接成功后打开一个channel
chan=t.open_session()
#设置会话超时时间
chan.settimeout(session_timeout)
#打开远程的terminal
chan.get_pty()
#激活terminal
chan.invoke_shell()

然后就可以通过chan.send('command')和chan.recv(recv_buffer)来远程执行命令以及本地获取反馈。

 二: paramiko的两个模块介绍

paramiko有两个模块SSHClient()和SFTPClient()

SSHClient()的使用代码:

import paramiko

ssh = paramiko.SSHClient() # 创建SSH对象
# 允许连接不在know_hosts文件中的主机
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
# 连接服务器
ssh.connect(hostname='192.168.2.103', port=22, username='root', password='123456')

stdin, stdout, stderr = ssh.exec_command('ls') # 执行命令
result = stdout.read() # 获取命令结果
print (str(result,encoding='utf-8'))
ssh.close() # 关闭连接

SSHClient()里有个transport变量,是用于获取连接,我们也可单独的获取到transport变量,然后执行连接操作

import paramiko

transport = paramiko.Transport(('192.168.2.103', 22))
transport.connect(username='root', password='123456')

ssh = paramiko.SSHClient()
ssh._transport = transport

stdin, stdout, stderr = ssh.exec_command('df')
print (str(stdout.read(),encoding='utf-8'))

transport.close()

用transport实现上传下载以及命令的执行:

#coding:utf-8
import paramiko
import uuid

class SSHConnection(object):

def __init__(self, host='192.168.2.103', port=22, username='root',pwd='123456'):
self.host = host
self.port = port
self.username = username
self.pwd = pwd
self.__k = None

def connect(self):
transport = paramiko.Transport((self.host,self.port))
transport.connect(username=self.username,password=self.pwd)
self.__transport = transport

def close(self):
self.__transport.close()

def upload(self,local_path,target_path):
# 连接,上传
# file_name = self.create_file()
sftp = paramiko.SFTPClient.from_transport(self.__transport)
# 将location.py 上传至服务器 /tmp/test.py
sftp.put(local_path, target_path)

def download(self,remote_path,local_path):
sftp = paramiko.SFTPClient.from_transport(self.__transport)
sftp.get(remote_path,local_path)

def cmd(self, command):
ssh = paramiko.SSHClient()
ssh._transport = self.__transport
# 执行命令
stdin, stdout, stderr = ssh.exec_command(command)
# 获取命令结果
result = stdout.read()
print (str(result,encoding='utf-8'))
return result

ssh = SSHConnection()
ssh.connect()
ssh.cmd("ls")
ssh.upload('s1.py','/tmp/ks77.py')
ssh.download('/tmp/test.py','kkkk',)
ssh.cmd("df")
ssh.close()

到此这篇关于使paramiko库执行命令时,在给定的时间强制退出的文章就介绍到这了,更多相关paramiko库执行命令内容请搜索三水点靠木以前的文章或继续浏览下面的相关文章希望大家以后多多支持三水点靠木!

Python 相关文章推荐
Python中的jquery PyQuery库使用小结
May 13 Python
python网络编程学习笔记(九):数据库客户端 DB-API
Jun 09 Python
利用Python操作消息队列RabbitMQ的方法教程
Jul 19 Python
python表格存取的方法
Mar 07 Python
python3使用smtplib实现发送邮件功能
May 22 Python
使用Python获取网段IP个数以及地址清单的方法
Nov 01 Python
pandas dataframe添加表格框线输出的方法
Feb 08 Python
python3.6.8 + pycharm + PyQt5 环境搭建的图文教程
Jun 11 Python
Keras 加载已经训练好的模型进行预测操作
Jun 17 Python
Tensorflow之MNIST CNN实现并保存、加载模型
Jun 17 Python
详解如何在PyCharm控制台中输出彩色文字和背景
Aug 17 Python
教你用python实现12306余票查询
Jun 30 Python
python上下文管理的使用场景实例讲解
Mar 03 #Python
pip install命令安装扩展库整理
Mar 02 #Python
python实现杨辉三角的几种方法代码实例
Mar 02 #Python
python爬虫破解字体加密案例详解
Mar 02 #Python
python包的导入方式总结
Mar 02 #Python
numpy实现RNN原理实现
Mar 02 #Python
解决tensorflow模型压缩的问题_踩坑无数,总算搞定
Mar 02 #Python
You might like
PHP多线程之内部多线程实例分析
2015/03/09 PHP
PHP Yii框架之表单验证规则大全
2015/11/16 PHP
谈谈从phpinfo中能获取哪些值得注意的信息
2017/03/28 PHP
PHP实现验证码校验功能
2017/11/16 PHP
从面试题学习Javascript 面向对象(创建对象)
2012/03/30 Javascript
jQuery之日期选择器的深入解析
2013/06/19 Javascript
PHP使用方法重载实现动态创建属性的get和set方法
2014/11/17 Javascript
Javascript 拖拽的一些高级的应用(逐行分析代码,让你轻松了拖拽的原理)
2015/01/23 Javascript
Javascript中数组方法汇总(推荐)
2015/04/01 Javascript
JS实现动态移动层及拖动浮层关闭的方法
2015/04/30 Javascript
JavaScript获得url查询参数的方法
2015/07/02 Javascript
jQuery实现输入框邮箱内容自动补全与上下翻动显示效果【附demo源码下载】
2016/09/20 Javascript
ReactNative页面跳转实例代码
2016/09/27 Javascript
jQuery实现动态添加tr到table的方法
2016/12/26 Javascript
angular学习之ngRoute路由机制
2017/04/12 Javascript
微信小程序wx.previewImage预览图片实例详解
2017/12/07 Javascript
详解Next.js页面渲染的优化方案
2019/01/27 Javascript
Vue data的数据响应式到底是如何实现的
2020/02/11 Javascript
vue 项目软键盘回车触发搜索事件
2020/09/09 Javascript
[01:06]DOTA2小知识课堂 Ep.02 吹风竟可解梦境缠绕
2019/12/05 DOTA
python实现列表中由数值查到索引的方法
2018/06/27 Python
Django框架使用富文本编辑器Uedit的方法分析
2018/07/31 Python
Python比较配置文件的方法实例详解
2019/06/06 Python
简单介绍python封装的基本知识
2019/08/10 Python
python脚本调用iftop 统计业务应用流量的思路详解
2019/10/11 Python
python3实现弹弹球小游戏
2019/11/25 Python
如何在mac下配置python虚拟环境
2020/07/06 Python
opencv 阈值分割的具体使用
2020/07/08 Python
python反扒机制的5种解决方法
2021/02/06 Python
AmazeUI 折叠面板的实现代码
2020/08/17 HTML / CSS
2014年学校领导班子对照检查材料
2014/09/19 职场文书
篮球友谊赛通讯稿
2014/10/10 职场文书
大二学年个人总结
2015/03/03 职场文书
大学生违纪检讨书范文
2015/05/07 职场文书
少先队入队仪式主持词
2015/07/04 职场文书
canvas绘制折线路径动画实现
2021/05/12 Javascript