文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

利用paramiko库执行命令时如何在超出给定的时间后强制退出

2023-06-06 17:31

关注

利用paramiko库执行命令时如何在超出给定的时间后强制退出?很多新手对此不是很清楚,为了帮助大家解决这个难题,下面小编将为大家详细讲解,有这方面需求的人可以来学习下,希望你能有所收获。

实现方式:

线程+事件,在线程中执行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 paramikossh = 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 paramikotransport = paramiko.Transport(('192.168.2.103', 22))transport.connect(username='root', password='123456')ssh = paramiko.SSHClient()ssh._transport = transportstdin, stdout, stderr = ssh.exec_command('df')print (str(stdout.read(),encoding='utf-8'))transport.close()

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

#coding:utf-8import paramikoimport uuidclass SSHConnection(object):def __init__(self, host='192.168.2.103', port=22, username='root',pwd='123456'):self.host = hostself.port = portself.username = usernameself.pwd = pwdself.__k = Nonedef connect(self):transport = paramiko.Transport((self.host,self.port))transport.connect(username=self.username,password=self.pwd)self.__transport = transportdef 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.pysftp.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 resultssh = SSHConnection()ssh.connect()ssh.cmd("ls")ssh.upload('s1.py','/tmp/ks77.py')ssh.download('/tmp/test.py','kkkk',)ssh.cmd("df")ssh.close()

看完上述内容是否对您有帮助呢?如果还想对相关知识有进一步的了解或阅读更多相关文章,请关注编程网行业资讯频道,感谢您对编程网的支持。

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     807人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     351人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     314人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     433人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     221人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯