再战健康打卡|Datetime库及try:except:的使用

重构一时爽, 构错火葬场 重构的起因是因为有一天任务计划程序显示0X1但并没有收到邮件, 很显然, 由于忽忽悠的采用了比肩国际一流游戏厂商的极为先进的生物(土豆)供电服务器, 导致经常出现难以预料的情况(喂, 明明是你的脚本鲁棒性不够好嘛).

开搞之前

首先分析需求.

之前的脚本发送的邮件中并没有最后打卡时间, 又由于鲁棒性的不足, 导致并不能完全通过收到邮件来判断是否成功打卡.

同样, 之前的脚本并没有对异常情况的处理机制(没错, 我就是能卡你半个小时), 导致出现异常时不能收到邮件反馈.

时间获取

历史填报会打开新的标签页, 所以就需要切换标签页.

# 获取当前浏览器的所有标签页
tabs = driver.window_handles
# 定位到第二个标签页
driver.switch_to.window(tabs[1])

接下来就是老套路, 定位元素, 获取值, Xpath走起.

table = WebDriverWait(driver, 180).until(EC.presence_of_element_located((By.XPATH, '//*[@id="countTableBody"]/tr[1]/td[1]')))

那么如何获取当前时间呢? 我们可以使用datetime库.

today = datetime.datetime.now()

时间处理

datetime库中提供了一些函数来对时间进行处理, 比如从字符串中提取时间, 将时间转换为字符串.

today = datetime.datetime.now()
today_date = today.strftime("%Y-%m-%d")
newest_time = datetime.datetime.strptime(table.text, "%Y-%m-%d %H:%M:%S")
newest_date = newest_time.strftime("%Y-%m-%d")

异常处理

通过try-except语句可以实现对异常状况的处理, 首先执行try下的语句块, 如果出现异常就跳转执行except下的语句块.

既然做了异常处理, 就没有必要设置1800s的延迟了, 调成180s以示对极为先进的生物供电服务器的尊敬.

完整代码

这里对敏感信息做了模糊处理, 请根据实际情况自行替换。

点我查看完整代码

#源自CSDN：https://blog.csdn.net/wjl_zyl_1314/article/details/107036245

#自动填报部分
from selenium import webdriver
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.by import By
import time
import datetime
import smtplib
from email.mime.text import MIMEText
from email.header import Header

head = '今日疫情填报情况'
mail_text = '无法获取历史填报信息'

try:
    driver = webdriver.Firefox()
    url = 'http://ids.hhu.edu.cn/amserver/UI/Login?goto=http://form.hhu.edu.cn:80/pdc/form/list'
    driver.get(url)
    driver.maximize_window()
    username = driver.find_element_by_id('IDToken1')
    stu_number = '***'
    username.send_keys(stu_number)
    stu_password = '***'
    password = driver.find_element_by_id('IDToken2')
    password.send_keys(stu_password)
    sign_inBtn = WebDriverWait(driver, 180).until(EC.presence_of_element_located((By.XPATH, "//*[contains(@onclick, 'defaultSubmit()')]")))
    #sign_inBtn.click()
    driver.execute_script('defaultSubmit()')
    goto = WebDriverWait(driver, 180).until(EC.presence_of_element_located((By.CLASS_NAME, "datav-flex-box")))
    goto.click()
    submit = WebDriverWait(driver, 180).until(EC.presence_of_element_located((By.ID, "saveBtn")))
    #submit.click()
    time.sleep(5)
    driver.execute_script('save()')
    #driver.close()

    # 拖时间, 避免没打完卡就返回上一页
    WebDriverWait(driver, 180).until(EC.presence_of_element_located((By.CLASS_NAME, "panel-body")))
    driver.back()
    driver.back()
    driver.back()
    driver.back()
    driver.back()
    time.sleep(5)
    username = driver.find_element_by_id('IDToken1')
    #username.send_keys(stu_number)
    password = driver.find_element_by_id('IDToken2')
    password.send_keys(stu_password)
    sign_inBtn = WebDriverWait(driver, 180).until(EC.presence_of_element_located((By.XPATH, "//*[contains(@onclick, 'defaultSubmit()')]")))
    #sign_inBtn.click()
    driver.execute_script('defaultSubmit()')
    goto = WebDriverWait(driver, 180).until(EC.presence_of_element_located((By.CLASS_NAME, "datav-flex-box")))
    goto.click()
    # 获取最近一次填报信息
    history = WebDriverWait(driver, 180).until(EC.presence_of_element_located((By.ID, "showHistoryBtn")))
    driver.execute_script('showFormFilled()')
    # 获取当前浏览器的所有标签页
    tabs = driver.window_handles
    # 定位到第二个标签页
    time.sleep(5)
    driver.switch_to.window(tabs[1])
    table = WebDriverWait(driver, 180).until(EC.presence_of_element_located((By.XPATH, '//*[@id="countTableBody"]/tr[1]/td[1]')))
    time.sleep(5)
    # 获取今日日期和最新填报日期
    today = datetime.datetime.now()
    today_date = today.strftime("%Y-%m-%d")
    newest_time = datetime.datetime.strptime(table.text, "%Y-%m-%d %H:%M:%S")
    newest_date = newest_time.strftime("%Y-%m-%d")
    if today_date == newest_date:
        mail_text = '今日疫情填报已完成, '
    else:
        mail_text = '今日疫情填报未完成, '
    mail_text += '最后填报时间: ' + table.text
except:
    head = '疫情填报异常'
    try:
        driver = webdriver.Firefox()
        url = 'http://ids.hhu.edu.cn/amserver/UI/Login?goto=http://form.hhu.edu.cn:80/pdc/form/list'
        driver.get(url)
        driver.maximize_window()
        username = driver.find_element_by_id('IDToken1')
        stu_number = '****'
        username.send_keys(stu_number)
        stu_password = '****'
        password = driver.find_element_by_id('IDToken2')
        password.send_keys(stu_password)
        sign_inBtn = WebDriverWait(driver, 180).until(EC.presence_of_element_located((By.XPATH, "//*[contains(@onclick, 'defaultSubmit()')]")))
        #sign_inBtn.click()
        driver.execute_script('defaultSubmit()')
        goto = WebDriverWait(driver, 180).until(EC.presence_of_element_located((By.CLASS_NAME, "datav-flex-box")))
        goto.click()
        # 获取最近一次填报信息
        history = WebDriverWait(driver, 180).until(EC.presence_of_element_located((By.ID, "showHistoryBtn")))
        driver.execute_script('showFormFilled()')
        #获取当前浏览器的所有标签页
        tabs = driver.window_handles
        # 定位到第二个标签页
        driver.switch_to.window(tabs[1])
        table = WebDriverWait(driver, 180).until(EC.presence_of_element_located((By.XPATH, '//*[@id="countTableBody"]/tr[1]/td[1]')))
        time.sleep(5)
        # 获取今日日期和最新填报日期
        today = datetime.datetime.now()
        today_date = today.strftime("%Y-%m-%d")
        newest_time = datetime.datetime.strptime(table.text, "%Y-%m-%d %H:%M:%S")
        newest_date = newest_time.strftime("%Y-%m-%d")
        if today_date == newest_date:
            mail_text = '今日疫情填报已完成, '
        else:
            mail_text = '今日疫情填报未完成, '
        mail_text += '最后填报时间: ' + table.text
        head += ', 获取填报信息正常'
    except:
        head += ', 获取填报信息异常'

# 发信方邮箱及密码
from_addr = '***@qq.com'
password = '***'

# 收信方邮箱
to_addr = '***@qq.com'

# 发信服务器
smtp_server = 'smtp.qq.com'

# 邮箱正文内容，第一个参数为内容（正文部分），第二个参数为格式(plain 为纯文本)，第三个参数为编码
print(mail_text)
msg = MIMEText(mail_text, 'plain', 'utf-8')

# 邮件头信息
msg['From'] = Header(from_addr)
msg['To'] = Header(to_addr)
msg['Subject'] = Header(head)

server = smtplib.SMTP_SSL(smtp_server)
server.connect(smtp_server, 465)
server.login(from_addr, password)
server.sendmail(from_addr, to_addr, msg.as_string())
server.quit()
driver.quit()