有个zabbix抓取图片脚本需要各位大神帮忙看下了

610302547

下面这个脚本是从网上找的.

功能就是从网页和数据库中获取screen上显示的服务器性能监控图, 然后通过邮件形式发送到我的邮箱中.

而我现在想要的是获取到的图片能不能导入到PDF文档中, 谢谢! 因为不懂Python, 所以麻烦cu上的大神了.

1. #! /usr/bin/env python
   
2. #coding=utf-8
   
3. # Andy_f
   
4. import time,os
   
5. import urllib
   
6. import urllib2
   
7. import cookielib
   
8. import MySQLdb
   
9. import smtplib
   
10. from email.mime.text import MIMEText
    
11. 
    
12. screens = ["zabbix"]
    
13. #
    
14. save_graph_path = "/opt/httpd-2.4.3/htdocs/reports/%s"%time.strftime("%Y-%m-%d")
    
15. if not os.path.exists(save_graph_path):
    
16.     os.makedirs(save_graph_path)
    
17. # zabbix host
    
18. zabbix_host = "192.168.2.1"
    
19. # zabbix login username
    
20. username = "admin"
    
21. # zabbix login password
    
22. password = "password"
    
23. # graph width
    
24. width = 600
    
25. # graph height
    
26. height = 100
    
27. # graph Time period, s
    
28. period = 86400
    
29. # zabbix DB
    
30. dbhost = "192.168.2.1"
    
31. dbport = 3306
    
32. dbuser = "zabbix"
    
33. dbpasswd = "zabbix"
    
34. dbname = "zabbix"
    
35. # mail
    
36. to_list = ["wangshuai@163.com"]
    
37. smtp_server = "mail.163.com"
    
38. mail_user = "zabbix@163.com"
    
39. mail_pass = "password"
    
40. domain  = "163.com
    
41. 
    
42. def mysql_query(sql):
    
43.     try:
    
44.         conn = MySQLdb.connect(host=dbhost,user=dbuser,passwd=dbpasswd,port=dbport,connect_timeout=20)
    
45.         conn.select_db(dbname)
    
46.         cur = conn.cursor()
    
47.         count = cur.execute(sql)
    
48.         if count == 0:
    
49.             result = 0
    
50.         else:
    
51.             result = cur.fetchall()
    
52.         return result
    
53.         cur.close()
    
54.         conn.close()
    
55.     except MySQLdb.Error,e:
    
56.         print "mysql error:" ,e
    
57. 
    
58. def get_graph(zabbix_host,username,password,screen,width,height,period,save_graph_path):
    
59.     screenid_list = []
    
60.     global html
    
61.     html = ''
    
62.     for i in mysql_query("select screenid from screens where name='%s'"%(screen)):
    
63.                 for screenid in i:
    
64.                     graphid_list = []
    
65.                     for c in mysql_query("select resourceid from screens_items where screenid='%s'"%(int(screenid))):
    
66.                         for d in c:
    
67.                             graphid_list.append(int(d))
    
68.                     for graphid in graphid_list:
    
69.                         login_opt = urllib.urlencode({
    
70.                         "name": username,
    
71.                         "password": password,
    
72.                         "autologin": 1,
    
73.                         "enter": "Sign in"})
    
74.                         get_graph_opt = urllib.urlencode({
    
75.                         "graphid": graphid,
    
76.                         "screenid": screenid,
    
77.                         "width": width,
    
78.                         "height": height,
    
79.                         "period": period})
    
80.                         cj = cookielib.CookieJar()
    
81.                         opener = urllib2.build_opener(urllib2.HTTPCookieProcessor(cj))
    
82.                         login_url = r"http://%s/index.php"%zabbix_host
    
83.                         save_graph_url = r"http://%s/chart2.php"%zabbix_host
    
84.                         opener.open(login_url,login_opt).read()
    
85.                         data = opener.open(save_graph_url,get_graph_opt).read()
    
86.                         filename = "%s/%s.%s.png"%(save_graph_path,screenid,graphid)
    
87.                         html += '<img width="600" height="250" src="http://%s/%s/%s/%s.%s.png">'%(zabbix_host,save_graph_path.split("/")[len(save_graph_path.split("/"))-2],save_graph_path.split("/")[len(save_graph_path.split("/"))-1],screenid,graphid)
    
88.                         f = open(filename,"wb")
    
89.                         f.write(data)
    
90.                         f.close()
    
91. 
    
92. 
    
93. 
    
94. def send_mail(username,password,smtp_server,to_list,sub,content):
    
95.     print to_list
    
96.     me = "monitor"+"<"+username+">"
    
97.     msg = MIMEText(content,_subtype="html",_charset="utf8")
    
98.     msg["Subject"] = sub
    
99.     msg["From"] = me
    
100.     msg["To"] = ";".join(to_list)
     
101.     try:
     
102.         server = smtplib.SMTP()
     
103.         server.connect(smtp_server)
     
104.         server.login(username,password)
     
105.         server.sendmail(me,to_list,msg.as_string())
     
106.         server.close()
     
107.         print "send mail Ok!"
     
108.     except Exception, e:
     
109.         print e
     
110. 
     
111. if __name__ == '__main__':
     
112.     for screen in screens:
     
113.         get_graph(zabbix_host,username,password,screen,width,height,period,save_graph_path)
     
114.     send_mail(mail_user,mail_pass,smtp_server,to_list,"PROD Server Performance Report",html)
     

复制代码

610302547

@wolfg@smallfish_xy@ning_lianjie
不能沉啊!!!

icymirror

回复 2# 610302547
下面的代码可以把jpg文件保存成pdf

1. import PIL
   
2. pic = PIL.Image.open("test.jpg")
   
3. with open("test.pdf", "w") as fh:
   
4.     pic.save(fh, "PDF")
   

复制代码

610302547

icymirror 发表于 2014-07-22 16:07
回复 2# 610302547
下面的代码可以把jpg文件保存成pdf

谢谢, 因为没有学习过python, 所以你说的不大懂. 不过我这边有几十张图片怎么让他们按照一定格式显示在PDF里面呢?

能否给下完整的代码?

非常感谢!

fireweb

请问楼主的qq是多少？我也正在研究这个问题，咨询一个问题，谢谢哈

fireweb

回复 4# 610302547



我的qq68154866
我的脚本报这个错误，请问楼主这个是什么问题?
[root@localhost script]# python get_screen.py
mysql error: (1130, "Host '192.168.160.199' is not allowed to connect to this MySQL server"
Traceback (most recent call last):
  File "get_screen.py", line 115, in ?
    get_graph(zabbix_host,username,password,screen,width,height,period,save_graph_path)
  File "get_screen.py", line 64, in get_graph
    for i in mysql_query("select screenid from screens where name='%s'"%(screen)):
TypeError: iteration over non-sequence

   

610302547

回复 6# fireweb


   
hi,

已经加你QQ. 谢谢!