python发送grafana视图

2019独角兽企业重金招聘Python工程师标准>>>

cat common.py(该文件为一个公共库文件；实现了两个功能：视图截取、邮件发送)

#! /usr/bin/env python
#coding=utf-8
import time,os
import urllib
import urllib2
import cookielib
import MySQLdb
import smtplib
from email.mime.text import MIMEText
from email.mime.image import MIMEImage
from email.mime.multipart  import MIMEMultipart
from email.header import Header
from selenium import webdriver
from selenium.webdriver.common.desired_capabilities import DesiredCapabilities
from selenium.webdriver.common.proxy import ProxyType


def take_screenshot(url, save_fn):
    headers = {'Authorization':'Bearer eyJrIjoiOWNCZW9lRnpBRGZzenhSSlA3QzJXU3hTU0lWZERsMmYiLCJuIjoiZ3JhZmFuYSByZXBvcnQiLCJpZCI6MX0='}
    desire = DesiredCapabilities.PHANTOMJS.copy()
    for key, value in headers.iteritems():
            desire['phantomjs.page.customHeaders.{}'.format(key)] = value
    driver = webdriver.PhantomJS(executable_path='/usr/local/bin/phantomjs',desired_capabilities=desire, service_args=['--load-images=yes'])
    #phantomjs截取整张网页
    # driver = webdriver.PhantomJS()
    driver.set_window_size(1920, 2000)
    driver.get(url) # Load page
    #将页面的滚动条拖到最下方，然后再拖回顶部
    driver.execute_script("""
        (function () {
            var y = 0;
            var step = 100;
            window.scroll(0, 0);

            function f() {
                if (y < document.body.scrollHeight) {
                    y += step;
                    window.scroll(0, y);
                    setTimeout(f, 100);
                } else {
                    window.scroll(0, 0);
                    document.title += "scroll-done";
                }
            }

            setTimeout(f, 1000);
        })();
    """)

    for i in xrange(30):
        if "scroll-done" in driver.title:
            break
        time.sleep(10)

    driver.save_screenshot(save_fn)
    driver.close()


def send_mail(username,password,smtp_server,smtp_port,to_list,sub,image):

    msgRoot = MIMEMultipart('related')

    me = "ops"+"<"+username+">"
    msgRoot["Subject"] = sub
    msgRoot["From"] = Header("ops", 'utf-8')
    msgRoot["To"] = ";".join(to_list)

    msgAlternative = MIMEMultipart('alternative')
    msgRoot.attach(msgAlternative)

    mail_msg = """
    <p><img src="cid:image1"></p>
    """
    msgAlternative.attach(MIMEText(mail_msg, 'html', 'utf-8'))

    print time.strftime('%Y-%m-%d %H:%M',time.localtime(time.time()))
    print to_list

    fp = open(image,'rb')
    msgImage = MIMEImage(fp.read())
    fp.close()
    # 定义图片ID
    msgImage.add_header('Content-ID', '<image1>')
    msgRoot.attach(msgImage)

    print smtp_server
    try:
        server = smtplib.SMTP()
        server.connect(smtp_server,smtp_port)
        #server.ehlo()
        #server.starttls()
        server.login(username,password
        server.sendmail(me,to_list,msgRoot.as_string())
        server.close()
        print "邮件发送成功"
    except smtplib.SMTPException,e:
        print "Error: 无法发送邮件"
        print e

具体报表文件

#!/usr/bin/python
# -*- coding: utf-8 -*-

import os
import time
import common

# mail
sub = "ELK平台技术运营日报"
smtp_server = "smtp.web.com"
smtp_port = "443"
mail_user = "user"
mail_pass = "pass"
domain = ''

BOSS_team = ["leeypp@foxmail.com"]
NORMAL_team = ["leeypp@foxmail.com"]
to_list = BOSS_team + NORMAL_team

#png
timestr = time.strftime("%Y%m%d", time.localtime())
png_file = "ELK_Daily"+timestr+".png"

if __name__ == "__main__":
    common.take_screenshot("http://ip:3000/d/0ZdLbJliz/es-ls?orgId=1&from=now-1d%2Fd&to=now-1d%2Fd&kiosk=tv",png_file)
    common.send_mail(mail_user,mail_pass,smtp_server,smtp_port,to_list,sub,png_file)
    os.popen("mv *png pngback ; cat ghostdriver.log >> logs/ghostdriver.log ; rm ghostdriver.log")

转载于:https://my.oschina.net/leeypp1/blog/3011906