zabbix图形截图

# -*- coding: utf-8 -*-
# @Time    : 2018/11/15 14:44
# @Author  : xinjie


from selenium import webdriver
import time


def capture(url, save_fn="capture.png"):
    # browser = webdriver.Firefox()  # Get local session of firefox
    browser = webdriver.Chrome()
    browser.set_window_size(1200, 900)
    browser.get(url)  # Load page
    # 输入账号和密码
    browser.find_element_by_name("name").send_keys(u"用户名")
    browser.find_element_by_name("password").send_keys("密码")
    # time.sleep(2)
    # 点击按钮提交登录表单
    browser.find_element_by_xpath("//*[@id='enter']").click()
    # time.sleep(5)
    browser.get("你的截图地址")
    time.sleep(5)

    # browser.find_element_by_xpath("/html/body/div[2]/div[4]/div[11]").screenshot("b.png")
    browser.find_element_by_css_selector(
        "body > div.article > div.dashbrd-grid-widget-container > div:nth-child(3)").screenshot("a.png")
    browser.find_element_by_css_selector(
        "body > div.article > div.dashbrd-grid-widget-container > div:nth-child(4)").screenshot("b.png")
    browser.find_element_by_css_selector(
        "body > div.article > div.dashbrd-grid-widget-container > div:nth-child(5)").screenshot("c.png")
    browser.find_element_by_css_selector(" body > div.article > div.dashbrd-grid-widget-container > div:nth-child(6)").screenshot("d.png")
    #  print(browser.title)
    # 滚屏js
    # browser.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, 50);
    #     } else {
    #       window.scroll(0, 0);
    #       document.title += "scroll-done";
    #     }
    #   }
    #
    #   setTimeout(f, 1000);
    # })();
    # """)
    #
    # for i in range(30):
    #     print(browser.title)
    #     if "scroll-done" in browser.title:
    #         break
    #     time.sleep(1)

    # browser.save_screenshot(save_fn)
    browser.close()


if __name__ == "__main__":
    capture("zabbix地址")