使用Selenium抓取zabbix性能监控图 - 51Testing软件测试论坛

#!/usr/bin/env python
　　# -*- coding:utf-8 -*-
　　# name image.py
　　from selenium import webdriver
　　import os
　　import sys
　　reload(sys)
　　sys.setdefaultencoding('utf-8')
　　def get_item_graph(itemid,flag,eventid):
　　 temp_name = "/tmp/"+eventid+".png"
　　 #save_screenshot仅能保存png格式图片，所以文件名定义需要以png结尾
　　 driver = webdriver.PhantomJS("/usr/local/zabbix-agent-ops/phantomjs-2.1.1/bin/phantomjs",service_log_path=os.path.devnull)
　　 #使用PhantomJS可以模拟浏览器进行访问
　　 driver.get("http://127.0.0.1/zabbix/")
　　 driver.set_window_size(640,480)
　　 driver.find_element_by_id("name").send_keys("armo")
　　 driver.find_element_by_id("password").send_keys("123456")
　　 driver.find_element_by_id("enter").click()
　　 #模拟访问url，在对应的元素element处输入用户名密码后click登陆
　　 if flag:
　　 driver.get("http://127.0.0.1/zabbix/history.php?action=showgraph&fullscreen=1&itemids[]="+itemid)
　　 else:
　　 driver.get("http://127.0.0.1/zabbix/history.php?action=showvalues&fullscreen=1&itemids[]="+itemid)
　　 #flag如果是1，则对应Item有对应的graph获取；如果是0，则获取最新一段时间的值
　　 driver.save_screenshot(temp_name)
　　 #将网页内容保存为png图片
　　 driver.close()
　　 driver.quit()
　　if __name__ == "__main__":
　　 if len(sys.argv) > 1:
　　 itemid = sys.argv[1] #脚本传递的第一个参数 Item ID
　　 flag = sys.argv[2] #脚本传递的第二个参数 Flag，从zabbix数据库item和graph的对应表查询item是否具有对应的graph，如果有则传递1到脚本，无传递0
　　 eventid = sys.argv[3] #脚本传递的第三个参数告警信息的Event ID，用来命名png图片
　　 get_item_graph(itemid,flag,eventid)
　　 #以上三个参数为前置调用脚本分析zabbix的发送信息后得到，之后会更新在后续文章中

复制代码