国家电网瑞星杀毒软件自动查询系统

#========================================
#国家电网瑞星杀毒软件自动查询系统
#开发日期:2018-12-06
#========================================
#coding=utf-8
import time
import winreg
from PIL import Image
from selenium import webdriver

# 创建chrome参数对象
opt = webdriver.ChromeOptions()
#关闭"自动控制"提示
opt.add_argument('disable-infobars')

# 把chrome设置成无界面模式，不论windows还是linux都可以，自动适配对应参数
#opt.set_headless()

# 创建对象
browser = webdriver.Chrome(options=opt)

#窗口最大化
browser.maximize_window()
#browser.set_window_size(1024,768) 自定义窗口大小

browser.implicitly_wait(20)
browser.get('http://10.230.152.35/ESM/Login.aspx?source=%2fESM%2fDefault.aspx')
browser.find_element_by_xpath('//*[@id="txtUserName"]').send_keys("yuanyang")
browser.find_element_by_xpath('//*[@id="txtPassword"]').send_keys("*******")
browser.find_element_by_xpath('//*[@id="btnLogin"]').click()
time.sleep(2)
browser.find_element_by_xpath('//*[@id="header_user_audit"]/a').click()
time.sleep(2)
browser.find_element_by_xpath('//*[@id="topMenu"]/li[2]/a/span').click()
time.sleep(3)
browser.find_element_by_xpath('//*[@id="tableBox1"]/table/thead/tr/th[7]/span').click()
time.sleep(2)

Table_element=browser.find_element_by_xpath('//*[@id="tableBox1"]/table')

#取桌面地址
def get_desktop():
key = winreg.OpenKey(winreg.HKEY_CURRENT_USER,r'Software\Microsoft\Windows\CurrentVersion\Explorer\Shell Folders')#利用系统的链表
return winreg.QueryValueEx(key, "Desktop")[0] #返回的是Unicode类型数据
Desktop_path=str(get_desktop())#Unicode转化为str

txt_now="杀毒查询"+time.strftime('%Y%m%d%H%M%S',time.localtime(time.time()))+'.png'

#截屏
image_name=Desktop_path+'\\'+ txt_now
browser.save_screenshot(image_name)

#裁剪为指定元素
left =Table_element.location['x']
top = Table_element.location['y']
right = Table_element.location['x'] + Table_element.size['width']
bottom = Table_element.location['y'] + Table_element.size['height']
im = Image.open(image_name)
im = im.crop((left, top, right, bottom))
im.save(image_name)

#print('当前页面title',browser.title)
#print('当前页面url',browser.current_url)

#打印网页源码
#print(browser.page_source)

#关闭浏览器
browser.close()

#关闭chreomedriver进程
browser.quit()