上周,看到部门同事在做设备反复重启的qtp脚本,很显然,在ui层面完成硬件设备的反复重启是极其痛苦的,原因有如下几点:
1. 在ui层完成对设备的重启本身具有高难度,启用的都是低级录制,失败率达到90%。
2. 硬件设备反复重启这个事情如果在ui层做有一个最大的弊端就是,永远无法知道设备什么时候起来了,我要继续重启它。
ui层的设备反复重启?!简直就是坑爹,为此,我们必须改造这个非常笨拙的测试方法。
反复重启设备,从代码角度来说,简直就是一个活脱脱的多线程协同模型(c/p-消费者,生产者)
我的思路如下:
1.开启两个线程,一个线程专门用来嗅探设备状态(并全局一个state)
2.两个线程互相通知,分别负责连接设备和重启设备
3.线程间共享一个state
4.利用python强大的paramiko模块
使用方法:
1.直接复制下面的代码到xxx.py
2.运行方式python xxx.py
使用前需要安装:
paramiko,由于我是在mac下配置的,这里不给出教程了,paramiko安装前需要手动编译安装pycrypto。
代码如下:
# -*- coding:utf-8 -*-
#About
__author__ = 'archer'
__version__=1.0.0
'''
设备重启长拷验证
'''
from threading import Condition,Thread
from time import sleep as wait
import paramiko
#condition需要全局共享,以保证线程之间顺利工作
condition=Condition()
#sshclient
devicecon=paramiko.SSHClient()
#设备默认状态为关闭
devicestate=False
class MyError(paramiko.SSHException):
pass
#设备通知线程
class DeviceNotify(Thread):
def run(self):
global devicestate
while True:
#拿锁
condition.acquire()
#如果设备处于开启状态,则通知DeviceDoReboot
if devicestate:
condition.notify()
condition.release()
else:
print [Thread-Notify]设备状态:关闭;正在等待设备重启......
try:
#set_missing_host_Key
devicecon.set_missing_host_key_policy(paramiko.AutoAddPolicy())
devicecon.connect('172.16.121.140',2222,root,kedacom,timeout=6000)
except:
pass
#等待时间
wait(10)
#将设备置为开并释放锁
devicestate=True
condition.release()
class DeviceDoReboot(Thread):
def run(self):
global devicestate
while True:
#拿锁
condition.acquire()
#如果设备处于开启状态,则进行关闭,
if devicestate:
print [Thread-DeviceReboot]线程已获取通知,正在重启设备......
#发送重启命令
try:
devicecon.exec_command(reboot)
except:
pass
#重启命令发送之后等待30秒,可以手动控制长拷频率
wait(30)
#将设备状态置为关闭
devicestate=False
#释放
condition.release()
else:
#通知DeviceNotify线程
condition.notify()
condition.release()
if __name__==__main__:
DeviceNotify().start()
DeviceDoReboot().start()