利用Python實(shí)現(xiàn)圖書(shū)超期提醒

更新時(shí)間：2016年08月02日 15:49:06 投稿：daisy

很多人喜歡逛圖書(shū)館，時(shí)不時(shí)去借本書(shū)，但每本書(shū)可能只可以借兩個(gè)月，一旦超期不還就會(huì)進(jìn)行相應(yīng)的處罰！為什么不寫(xiě)個(gè)腳本來(lái)通知自己圖書(shū)超期呢？說(shuō)了這么多廢話，我們就進(jìn)入主題吧?。?！

一、模擬登錄圖書(shū)館管理系統(tǒng)

我們可以先看一下登錄頁(yè)面（很多學(xué)校這些管理系統(tǒng)頁(yè)面就是很low）：

兩種方式去模擬登錄圖書(shū)館：

1. 構(gòu)造登錄表單進(jìn)行模擬登錄

這種方式模擬登錄似乎是很可靠的，但有時(shí)候就是在驗(yàn)證碼獲取上很困難，如果簡(jiǎn)單的網(wǎng)站，有的會(huì)利用當(dāng)前時(shí)間戳來(lái)構(gòu)造驗(yàn)證碼，這種就很容易從網(wǎng)頁(yè)上觀察出來(lái)，但比如我們這次要模擬登錄的網(wǎng)站似乎是不能這樣做，因?yàn)樗鞘褂肑avaScript標(biāo)準(zhǔn)庫(kù)里的Math函數(shù)直接隨機(jī)生成的驗(yàn)證碼鏈接，可以從下面圖片上觀察驗(yàn)證碼處的代碼：

它使用Math.random()函數(shù)返回 [0-1) 的浮點(diǎn)值偽隨機(jī)數(shù)（大于等于0，小于1）
好吧！我們換用一種比這個(gè)更簡(jiǎn)單的方式模擬登錄吧！

2. 通過(guò)Cookie登錄圖書(shū)館

Cookie，指某些網(wǎng)站為了辨別用戶身份、進(jìn)行session跟蹤而儲(chǔ)存在用戶本地終端上的數(shù)據(jù)（通常經(jīng)過(guò)加密）。

這里我們使用Requests庫(kù)來(lái)進(jìn)行模擬登錄過(guò)程，在這之前我們還有個(gè)問(wèn)題，怎么獲取Cookie呢？？
如果你使用的是谷歌瀏覽器，那你可以通過(guò)按F12就可以看到下圖里面有個(gè)Cookie的內(nèi)容，這就是你要的東西：

再上個(gè)圖分析一下，希望大家能有耐心讀下去：

通過(guò)圖片我們知道可以獲取借閱日期和應(yīng)還日期，獲取日期后根據(jù)應(yīng)還日期和當(dāng)前日期比較，就可以得出是否超期的結(jié)果。不多說(shuō)，先貼代碼再說(shuō)：

import requests
session = requests.Session()  # 會(huì)話對(duì)象讓你能夠跨請(qǐng)求保持某些參數(shù),它也會(huì)在同一個(gè)Session實(shí)例發(fā)出的所有請(qǐng)求之間保持cookie
session.headers = {
  'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/51.0.2704.84 Safari/537.36',
  'Cookie': 'ASP.NET_SessionId=1qri0rmoylpyrs45rurzme55; Hm_lvt_ed06d5e5f94d85932b82e4aac94d0c68=1467535679,1469713840; Hm_lpvt_ed06d5e5f94d85932b82e4aac94d0c68=1469713840; PHPSESSID=ev339udv0rrhqg6tfdvfukqos1'
}

上述代碼使用了requests的會(huì)話對(duì)象來(lái)保存Cookie, 如果我們需要跳轉(zhuǎn)到其它頁(yè)面，我們不用每次都模擬登錄，因?yàn)閏ookie已經(jīng)保存了我們的登錄狀態(tài)。

會(huì)不會(huì)有人疑問(wèn)，不是要說(shuō)模擬登錄的嗎？？怎么沒(méi)有這過(guò)程呢？？

其實(shí)我們上面代碼中的Cookie已經(jīng)保存了我們的登錄狀態(tài)，相當(dāng)于我們已經(jīng)模擬登錄過(guò)了，這樣子模擬登錄是不是簡(jiǎn)單多了，但缺點(diǎn)是我們需要手動(dòng)在登錄頁(yè)面輸入一遍，然后再?gòu)牡卿涰?yè)面找到cookie粘貼到代碼中來(lái)

二、獲取所借書(shū)籍信息

通過(guò)分析頁(yè)面，我們可以使用BeautifulSoup來(lái)提取我們需要的內(nèi)容，我們需要的是書(shū)籍的條形碼、題名和作者、借閱日期、應(yīng)還日期，其實(shí)我們只需要應(yīng)還日期就行，但為了以后需要，先獲取書(shū)籍的所有信息并保存進(jìn)數(shù)據(jù)庫(kù)里面：

定義了一個(gè)數(shù)據(jù)庫(kù)操作的函數(shù)，方便以后調(diào)用

def get_mysql():
  conn = pymysql.connect(host = 'localhost', user = 'root', passwd = '2014081029', db = 'mysql', charset = 'utf8')  # user為數(shù)據(jù)庫(kù)的名字，passwd為數(shù)據(jù)庫(kù)的密碼，一般把要把字符集定義為utf8，不然存入數(shù)據(jù)庫(kù)容易遇到編碼問(wèn)題
  cur = conn.cursor()  # 獲取操作游標(biāo)
  cur.execute('use book')  # 使用book這個(gè)數(shù)據(jù)庫(kù)
  return (cur, conn)

定義一個(gè)函數(shù)來(lái)獲取圖書(shū)信息并保存：

def get_book_name(book_url):
  html = session.get(book_url, cookies = cookie, headers = headers).content.decode('utf-8')
  soup = BeautifulSoup(html, 'lxml')
  book_bar = []  # 書(shū)籍的條形碼列表,用來(lái)判斷要存入數(shù)據(jù)庫(kù)的書(shū)籍是否已經(jīng)存在

  cur, conn = get_mysql()
  sql = 'select * from book_list;'
  cur.execute(sql)
  rows = cur.fetchall()
  for row in rows:
    book_bar.append(row[1])

  book_list = []  # 這個(gè)是我測(cè)試時(shí)使用的，作用是把每本書(shū)籍的信息列表放在這個(gè)列表中
  book_every = [] # 一本書(shū)籍的所有信息列表
  for book_time in soup.find_all('td', class_="whitetext"):
    print(book_time.get_text().strip()) # 移除字符串頭尾指定的字符（默認(rèn)為空格）
    pattern = re.compile(r'\s')
    content = re.sub(pattern, r'', book_time.get_text()) # 目的也是匹配任何空白符并去除，貌似對(duì)空行去除沒(méi)影響

    if content != '':
      book_every.append(content)
      if len(book_every) == 7:
        book_list.append(book_every)
        if book_every[0] not in book_bar:
          sql = 'insert book_list(條形碼, 題名和作者, 借閱日期, 應(yīng)還日期, 續(xù)借量, 館藏地, 附件) value(' + "\'" \
             + book_every[0] + "\'," + "\'" + book_every[1] + "\'," + "\'" + book_every[2] + "\'," + "\'" \
             + book_every[3] + "\'," + "\'" + book_every[4] + "\'," + "\'" + book_every[5] + "\'," + "\'" \
             + book_every[6] + "\'" + ');'
        try:

          cur.execute(sql)
          conn.commit()
        except:
          conn.rollback()
        book_every = []

  print(book_list)

接下來(lái)我們分析一下上面代碼中沒(méi)有注釋的代碼，首先我們先把處理后的信息加入book_every列表中，然后從頁(yè)面源代碼（tp9.png）中我們可以知道，一本書(shū)信息中只需要前面7項(xiàng)內(nèi)容，因此我們使用一個(gè)判斷語(yǔ)句：

if len(book_every) == 7:
  book_list.append(book_every)
  if book_every[0] not in book_title:
    sql = 'insert book_list(條形碼, 題名和作者, 借閱日期, 應(yīng)還日期, 續(xù)借量, 館藏地, 附件) value(' + "\'" \
             + book_every[0] + "\'," + "\'" + book_every[1] + "\'," + "\'" + book_every[2] + "\'," + "\'" \
             + book_every[3] + "\'," + "\'" + book_every[4] + "\'," + "\'" + book_every[5] + "\'," + "\'" \
             + book_every[6] + "\'" + ');'
    try:
      cur.execute(sql)
      conn.commit()    
    except:
      conn.rollback()  # 如果存入數(shù)據(jù)庫(kù)失敗，執(zhí)行回滾操作
  book_every = []

也就是說(shuō)，如果判斷出book_every已經(jīng)達(dá)到7項(xiàng)內(nèi)容，就執(zhí)行存入數(shù)據(jù)庫(kù)的操作，然后在把book_every重置為空列表

三、發(fā)送郵件提醒功能

先貼上代碼：

def send_message():
  day_num = [31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
  day_num1 = [31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
  sql = 'select * from book_list;'
  cur, conn = get_mysql()
  cur.execute(sql)
  rows = cur.fetchall()
  local_time = time.strftime("%Y-%m-%d", time.localtime()) # 獲取當(dāng)前時(shí)間
  local_time = str(local_time)
  times = re.split(r'-', local_time)
  year = times[0]

  number = 0
  while(True):
    for i in rows:
      print(i[4])
      pattern = re.split(r'-', i[4])
      if times[1] == pattern[1]:
        day = int(times[2]) - int(pattern[2])
        if day > 0:
          print('已經(jīng)超期了%d天' % day)
          number += 1
          send_email(day, number, i[2])
      elif times[1] > pattern[1]:
        if (year % 4 == 0 and year % 100 != 0) or year % 400 == 0:
          extend_day = day_num1[int(pattern[1]) - 1] - int(pattern[2]) + times[2]
          print('已經(jīng)超期了%d天' % extend_day)
          number += 1
          send_email(day, number, i[2])
        else:
          extend_day = day_num[int(pattern[1]) - 1] - int(pattern[2]) + times[2]
          print('已經(jīng)超期了%d天' % extend_day)
          number += 1
          send_email(day, number, i[2])

      else:
        print('還沒(méi)有超期的書(shū)籍')

      print(pattern[2])
    time.sleep(3600 * 24)

我們來(lái)分析代碼吧，首先我們判斷是否超期是根據(jù)當(dāng)前時(shí)間和應(yīng)還日期的相加減得到的，所以我們考慮到：

1.如果應(yīng)還日期是上個(gè)月，這里我們就要進(jìn)行月份的相加減，因?yàn)殚c年和平年的月份不一樣，所以我們定義了day_num和day_num1兩個(gè)列表來(lái)表示閏年和平年的月份天數(shù)。

2.然后我們使用月份當(dāng)做判斷條件來(lái)比較超期天數(shù)

月份判斷,如果當(dāng)前月份等于應(yīng)還月份，就執(zhí)行下面操作，注意里面已經(jīng)包含發(fā)送郵件函數(shù)，下面會(huì)貼出發(fā)送郵件函數(shù)，大家也許會(huì)想，為什么沒(méi)有判斷年份，因?yàn)槲乙话憬钑?shū)不會(huì)超期這么久，所以沒(méi)有加上這個(gè)判斷

      if times[1] == pattern[1]:
        day = int(times[2]) - int(pattern[2])
        if day > 0:
          print('已經(jīng)超期了%d天' % day)
          number += 1
          send_email(day, number, i[2])

然后是當(dāng)前月份大于應(yīng)還月份時(shí)，這時(shí)候就有閏年和平年的判斷了

      elif times[1] > pattern[1]:
        if (year % 4 == 0 and year % 100 != 0) or year % 400 == 0:
          extend_day = day_num1[int(pattern[1]) - 1] - int(pattern[2]) + times[2]
          print('已經(jīng)超期了%d天' % extend_day)
          number += 1
          send_email(day, number, i[2])

下面貼出發(fā)送郵件的代碼：

def send_email(day, number, title):
  from_addr = '15602200534@163.com'
  password = '就不告訴你'
  to_addr = '673411814@qq.com'
  smtp_server = 'smtp.163.com'

  text = 'Hello ,郭偉匡, 告訴你一個(gè)不好的消息,趕緊帶上你的書(shū),去圖書(shū)館交錢吧！你有一本叫《%s》的書(shū)籍超期了' \
      ',而且已經(jīng)超期了%d天了,總共有%d書(shū)超期了?。?！' % (title, day, number)
  msg = MIMEText(text, 'plain', 'utf-8')
  msg['From'] = format_addr('圖書(shū)館的通知<%s>' % from_addr)
  msg['To'] = format_addr('管理員<%s>' % to_addr)
  msg['Subject'] = Header('來(lái)著郭偉匡的問(wèn)候......', 'utf-8').encode()

  server = smtplib.SMTP(smtp_server, 25)
  server.set_debuglevel(1)
  server.login(from_addr, password)
  server.sendmail(from_addr, [to_addr], msg.as_string())
  server.quit()

最后把把發(fā)送郵件的截圖發(fā)出來(lái)：