zabbix自动截图留档_python版

liujjun

1 # -*- coding: utf-8 -*-　　
2 __author__ = 'xinysu'
　　
3 __date__ = '2017/10/12 14:38'
　　
4 import sys
　　
5 import datetime
　　
6 import http.cookiejar, urllib.request, urllib
　　
7 from lxml import etree
　　
8 import requests

　　
9>　　
10     def __init__(self, name, password):
　　
11         url="http://company.monitor.com/index.php";
　　
12         self.url = url
　　
13         self.name = name
　　
14         self.password = password
　　
15         cookiejar = http.cookiejar.CookieJar()
　　
16         urlOpener = urllib.request.build_opener(urllib.request.HTTPCookieProcessor(cookiejar))
　　
17         values = {"name": self.name, 'password': self.password, 'autologin': 1, "enter": 'Sign in'}
　　
18         data = urllib.parse.urlencode(values).encode(encoding='UTF8')
　　
19         request = urllib.request.Request(url, data)
　　
20         try:
　　
21             urlOpener.open(request, timeout=10)
　　
22             self.urlOpener = urlOpener
　　
23         except urllib.request.HTTPError as e:
　　
24             print(e)
　　
25     def download_chart(self, image_dir,itemids,stime,etime):
　　
26         # 此url是获取图片是的，请注意饼图的URL 和此URL不一样，请仔细观察！
　　
27         url="http://company.monitor.com/chart.php";
　　
28         # 折线图的大小
　　
30         url_par={}
　　
31         url_par={"width":1778, "height":300,"itemids":itemids}
　　
32         # 开始日期、结束日期从str转换为datetime
　　
33         stime = datetime.datetime.strptime(stime, "%Y-%m-%d")
　　
34         etime=datetime.datetime.strptime(etime, "%Y-%m-%d")
　　
35         # 计算period
　　
36         diff_sec = etime - stime
　　
37         period = diff_sec.days*24*3600 + diff_sec.seconds
　　
38         url_par["period"] = period
　　
39         # stime转换str
　　
40         stime = stime.strftime('%Y%m%d%H%M%S')
　　
41         url_par["stime"] = stime
　　
42         key = url_par.keys()
　　
43         data = urllib.parse.urlencode(url_par).encode(encoding='UTF8')
　　
44         request = urllib.request.Request(url, data)
　　
45         url = self.urlOpener.open(request)
　　
46         image = url.read()
　　
47         html = requests.get('http://company.monitor.com/history.php?action=showgraph&itemids[]={}'.format(itemids)).text
　　
48         page = etree.HTML(html)
　　
49         hostname_itemname = page.xpath('//div[@class="header-title"]/h1/text()')[0].split(':')
　　
50         hostname = hostname_itemname[0]
　　
51         hostname_itemname.pop(0)
　　
52         itemname = '_'.join(hostname_itemname).replace('/','_')
　　
53         imagename = "{}\{}_{}_{}_({}).png".format(image_dir,hostname,stime,etime.strftime('%Y%m%d%H%M%S'),itemname)
　　
54         f = open(imagename, 'wb')
　　
55         f.write(image)
　　
56