zabbix自动截图留档_python版
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()').split(':')
50 hostname = hostname_itemname
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
页:
[1]