#encode=utf-8
class Practice(object):
print u"这个方法的功能仅限于统计邮件"
def __init__(self,url,driver):
self.driver=driver
self.url=url
def brower_content(self):
self.driver.get(self.url)
mail_content=[]
try:
self.driver.switch_to_frame("login_frame")
self.driver.find_element_by_id("img_out_1300994936").click()
time.sleep(5)
self.driver.switch_to.default_content()
self.driver.find_element_by_id("folder_8").click()
time.sleep(5)
self.driver.switch_to_frame("mainFrame")
self.driver.find_element_by_id("ckb_selectAll").click()
time.sleep(1)
self.driver.find_element_by_id("setAllReaded").click()
time.sleep(1)
result=self.driver.find_elements_by_xpath(".//table[@class='i M']")
except Exception:
self.driver.switch_to_frame("login_frame")
self.driver.find_element_by_id("switcher_plogin").click()
self.driver.find_element_by_id("u").send_keys("1300994936@qq.com")
self.driver.find_element_by_id("").send_keys(password)
self.driver.find_element_by_id("login_button").click()
time.sleep(5)
self.driver.switch_to.default_content()
self.driver.find_element_by_id("folder_8").click()
time.sleep(5)
self.driver.switch_to_frame("mainFrame")
self.driver.find_element_by_id("ckb_selectAll").click()
time.sleep(1)
self.driver.find_element_by_id("setAllReaded").click()
time.sleep(1)
result=self.driver.find_elements_by_xpath(".//table[@class='i M']")
else:
print u"获取邮箱内容失败"
for i in result:
if i.find_element_by_xpath(".//td[@class='tl_g tf']").text.decode("utf-8")[:-1]==u"光荣之路涨薪学习2群":
checkout=i.find_element_by_xpath(".//td[@class='dt']//div").text
print checkout[3:5],i.find_element_by_xpath(".//td[@class='gt tf']//u").text
if re.match(u"今天",checkout) and int(checkout[3:5])<9:
mail_content.append(i.find_element_by_xpath(".//td[@class='gt tf']//u").text)
print i.find_element_by_xpath(".//td[@class='gt tf']//u").text
elif re.match(u"昨天",checkout) and int(checkout[3:5])>=9:
mail_content.append(i.find_element_by_xpath(".//td[@class='gt tf']//u").text)
print i.find_element_by_xpath(".//td[@class='gt tf']//u").text
self.driver.quit()
return mail_content
def Analysis(mail_content,file_name):
print u"现在开始分析有效邮件数据"
try:
book=load_workbook(file_name)
book1=book.get_sheet_by_name("Sheet1")
except Exception,e:
print u"excel对象获取失败，错误是: %s" % e.message
book1.cell(row=1,column=2,value=time.strftime("%Y%m%d",time.localtime(time.time()-24*60*60)))
for j in range(2,book1.max_row+1):
book1.cell(row=j,column=3,value=u"")
for i in mail_content:
for j in range(2,book1.max_row+1):
if book1.cell(row=j,column=1).value in i:
value=int(book1.cell(row=j,column=2).value)
book1.cell(row=j,column=2,value=value+1)
book1.cell(row=j,column=3,value=u"练习已提交")
book.save(file_name)
def brower_object(browser):
if browser.lower()=="ie":
driver = webdriver.Ie()
time.sleep(5)
return driver
elif browser.lower()=="firefox":
driver=webdriver.Firefox()
time.sleep(5)
return driver
else:
print u"您输入的浏览器 %s 暂时还不支持驱动" % self.browser
if __name__=="__main__":
from selenium import webdriver
import time
from openpyxl import load_workbook
import re
import sys
reload(sys)
sys.setdefaultencoding("utf-8")
driver=brower_object("firefox")
practice=Practice("http://www.mail.qq.com",driver)
mail_content=practice.brower_content()
name="1"
file_name=u"c://统计//统计.xlsx"
Analysis(mail_content,file_name)