一、實現(xiàn)目的
本來就很喜歡逛圖書館烙肺，時不時去借本書（注：借的都沒看過），但我這個學(xué)期突然發(fā)現(xiàn)了問題氧卧，每本書都可以借兩個月桃笙，但不幸的是我最近一學(xué)期借的書全部超期，一天一毛錢沙绝，我心疼這錢安鳌！Ｉ撩省星著！靈機一動，為什么不寫個腳本來通知自己圖書超期呢粗悯？說了這么多廢話虚循，我們就進入主題吧！！横缔！
二铺遂、模擬登錄圖書館管理系統(tǒng)
我們可以先看一下登錄頁面（很多學(xué)校這些管理系統(tǒng)頁面就是很low）：
兩種方式去模擬登錄圖書館：
1. 構(gòu)造登錄表單進行模擬登錄
這種方式模擬登錄似乎是很可靠的，但有時候就是在驗證碼獲取上很困難剪廉，如果簡單的網(wǎng)站娃循，有的會利用當前時間戳來構(gòu)造驗證碼，這種就很容易從網(wǎng)頁上觀察出來斗蒋，但比如我們這次要模擬登錄的網(wǎng)站似乎是不能這樣做捌斧，因為它是使用JavaScript標準庫里的Math函數(shù)直接隨機生成的驗證碼鏈接，可以從下面圖片上觀察驗證碼處的代碼：
日了個狗泉沾，它使用Math.random()函數(shù)返回 [0-1) 的浮點值偽隨機數(shù)（大于等于0捞蚂，小于1），剛開始我以為python的math.random()函數(shù)生成的隨機數(shù)和JavaScript的有區(qū)別跷究，后來試了一下姓迅，呵呵，原來兩個函數(shù)生成的隨機數(shù)都是[0-1)而且都是16位小數(shù)點的俊马。那樣子我們就可以模擬登錄了丁存。
首先，在模擬登錄先柴我，我們應(yīng)該在瀏覽器上模擬登錄一次解寝，觀察頁面變化情況,剛開始時頁面只有l(wèi)ogin.php頁面的：
然后我們輸入驗證碼后再觀察一下，頁面立刻被轉(zhuǎn)向redr_info.php,同時還有redr_verify.php頁面出現(xiàn)
然后看看我們的redr_info.php里面的東西艘儒，唉聋伦，怎么這個頁面是GET請求呢？界睁？
那驗證登錄請求的POST頁面在哪里去了呢觉增？？帶著疑問看看redr_verify.php,光是看這個頁面的命名就覺得這是個驗證登錄的頁面：
果然翻斟，POST請求在這里逾礁，那我們就可以構(gòu)造登錄表單通過這個頁面來模擬登錄了。
前期工作準備得差不多了访惜，開始找這個redr_verify.php的post提交部分的內(nèi)容了嘹履，我們從登錄頁面應(yīng)該也可以知道我們需要提交學(xué)號、密碼疾牲、驗證碼這三個。我們可以去redr_verify.php下看看我們POST表單提交的數(shù)據(jù)
我們只需要填寫前面四項就可以了衙解，第四項是什么呢阳柔，我回到登錄頁面看了一下，就是下面圖片的選擇蚓峦，
然后貼代碼吧
import subprocess
import sys
import os
session = requests.Session()
session.headers = {
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/52.0.2743.82 Safari/537.36'
}
def login(name, password):
random_num = random.random() # 生成隨機數(shù)舌剂，構(gòu)造獲取驗證碼的鏈接
url = 'http://210.38.102.131:86/reader/captcha.php?' + str(random_num)
get_captcha = session.get(url).content
with open('captcha.png', 'wb') as f:
f.write(get_captcha)
f.close()
'''
這段代碼是為了方便我們打開圖片济锄，它可以直接打開圖片
我們就不用去文件夾里去找,里面是判斷使用什么系統(tǒng)，
不同系統(tǒng)打開方式有點差異霍转，可以找python文檔了解這部分內(nèi)容
'''
if sys.platform.find('darwin') >= 0:
subprocess.call(['open', 'captcha.png'])
elif sys.platform.find('linux') >= 0:
subprocess.call(['xdg-open', 'captcha.png'])
else:
os.startfile('captcha.png')
input_captcha = input('請輸入驗證碼：')
input_captcha = str(input_captcha)
# 構(gòu)造登錄表單荐绝，里面就是我們上面提及的四項
post_data = {
'number': name,
'passwd': password,
'captcha': input_captcha,
'select': 'cert_no'
}
login_url = 'http://210.38.102.131:86/reader/redr_verify.php'
html = session.post(login_url, data=post_data).content
book_hist_url = 'http://210.38.102.131:86/reader/book_hist.php'
content = session.get(book_hist_url).content.decode('utf-8')
print(content)
這就模擬登錄成功了，
好吧避消！我們換用一種比這個更簡單的方式模擬登錄吧低滩！
2. 通過Cookie登錄圖書館
Cookie，指某些網(wǎng)站為了辨別用戶身份岩喷、進行session跟蹤而儲存在用戶本地終端上的數(shù)據(jù)（通常經(jīng)過加密）恕沫。
這里我們使用Requests庫來進行模擬登錄過程，在這之前我們還有個問題纱意，怎么獲取Cookie呢婶溯？？
如果你使用的是谷歌瀏覽器偷霉，那你可以通過按F12就可以看到下圖里面有個Cookie的內(nèi)容迄委，這就是你要的東西：
再上個圖分析一下，希望大家能有耐心讀下去：
通過圖片我們知道可以獲取借閱日期和應(yīng)還日期类少，獲取日期后根據(jù)應(yīng)還日期和當前日期比較叙身，就可以得出是否超期的結(jié)果。不多說瞒滴，先貼代碼再說：
import requests
session = requests.Session() # 會話對象讓你能夠跨請求保持某些參數(shù),它也會在同一個Session實例發(fā)出的所有請求之間保持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的會話對象來保存Cookie, 如果我們需要跳轉(zhuǎn)到其它頁面曲梗，我們不用每次都模擬登錄，因為cookie已經(jīng)保存了我們的登錄狀態(tài)妓忍。
會不會有人疑問虏两，不是要說模擬登錄的嗎？世剖？怎么沒有這過程呢定罢？？
其實我們上面代碼中的Cookie已經(jīng)保存了我們的登錄狀態(tài)旁瘫，相當于我們已經(jīng)模擬登錄過了祖凫，這樣子模擬登錄是不是簡單多了，但缺點是我們需要手動在登錄頁面輸入一遍酬凳，然后再從登錄頁面找到cookie粘貼到代碼中來
三惠况、獲取所借書籍信息
通過分析頁面，我們可以使用BeautifulSoup來提取我們需要的內(nèi)容宁仔，我們需要的是書籍的條形碼稠屠、題名和作者、借閱日期、應(yīng)還日期权埠，其實我們只需要應(yīng)還日期就行榨了，但為了以后需要，先獲取書籍的所有信息并保存進數(shù)據(jù)庫里面：
定義了一個數(shù)據(jù)庫操作的函數(shù)攘蔽，方便以后調(diào)用
def get_mysql():
conn = pymysql.connect(host = 'localhost', user = 'root', passwd = '2014081029', db = 'mysql', charset = 'utf8') # user為數(shù)據(jù)庫的名字龙屉，passwd為數(shù)據(jù)庫的密碼，一般把要把字符集定義為utf8满俗，不然存入數(shù)據(jù)庫容易遇到編碼問題
cur = conn.cursor() # 獲取操作游標
cur.execute('use book') # 使用book這個數(shù)據(jù)庫
return (cur, conn)
定義一個函數(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ù)據(jù)庫的書籍是否已經(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 = [] # 這個是我測試時使用的转捕，作用是把每本書籍的信息列表放在這個列表中
book_every = [] # 一本書籍的所有信息列表
for book_time in soup.find_all('td', class_="whitetext"):
print(book_time.get_text().strip()) # 移除字符串頭尾指定的字符（默認為空格）
pattern = re.compile(r'\s')
content = re.sub(pattern, r'', book_time.get_text()) # 目的也是匹配任何空白符并去除，貌似對空行去除沒影響
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)
接下來我們分析一下上面代碼中沒有注釋的代碼漫雷，首先我們先把處理后的信息加入book_every列表中瓜富，然后從頁面源代碼（tp9.png）中我們可以知道，一本書信息中只需要前面7項內(nèi)容降盹，因此我們使用一個判斷語句：
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ù)庫失敗与柑，執(zhí)行回滾操作
book_every = []
也就是說，如果判斷出book_every已經(jīng)達到7項內(nèi)容蓄坏，就執(zhí)行存入數(shù)據(jù)庫的操作价捧，然后在把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()) # 獲取當前時間
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('還沒有超期的書籍')
print(pattern[2])
time.sleep(3600 * 24)
方案一
我們來分析代碼吧涡戳，首先我們判斷是否超期是根據(jù)當前時間和應(yīng)還日期的相加減得到的结蟋，所以我們考慮到：
- 如果應(yīng)還日期是上個月，這里我們就要進行月份的相加減渔彰，因為閏年和平年的月份不一樣嵌屎，所以我們定義了day_num和day_num1兩個列表來表示閏年和平年的月份天數(shù)。
- 然后我們使用月份當做判斷條件來比較超期天數(shù)
月份判斷,如果當前月份等于應(yīng)還月份恍涂，就執(zhí)行下面操作宝惰，注意里面已經(jīng)包含發(fā)送郵件函數(shù)，下面會貼出發(fā)送郵件函數(shù)再沧，大家也許會想尼夺，為什么沒有判斷年份，因為我一般借書不會超期這么久炒瘸，所以沒有加上這個判斷
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])
然后是當前月份大于應(yīng)還月份時淤堵，這時候就有閏年和平年的判斷了
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 ,郭偉匡, 告訴你一個不好的消息,趕緊帶上你的書,去圖書館交錢吧！你有一本叫《%s》的書籍超期了' \
',而且已經(jīng)超期了%d天了,總共有%d書超期了Ｇ昀９招啊！' % (title, day, number)
msg = MIMEText(text, 'plain', 'utf-8')
msg['From'] = format_addr('圖書館的通知<%s>' % from_addr)
msg['To'] = format_addr('管理員<%s>' % to_addr)
msg['Subject'] = Header('來著郭偉匡的問候......', '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()
方案二
其實我們還有一種最簡單的判斷相差日期的方法隘截，
那就是使用python提供的datetime模塊扎阶，就利用方案一里面的東西來說吧
local_time = time.strftime("%Y-%m-%d", time.localtime()) # 獲取當前時間
local_time = str(local_time)
times = re.split(r'-', local_time)
我們通過split分離出年月日后事富，就可以很簡單得使用datetime進行日期相加減了，我們datetime相加減日期的用法如下：
d1 = datetime.datetime(2016, 4, 3)
d2 = datetime.datetime(2016, 6, 23)
print((d2 - d1).days)
打印結(jié)果就是相差天數(shù)乘陪，這樣在判斷日期方面就變得十分簡單了，所以方案二顯然比方案一好得多了
關(guān)于發(fā)送郵件的知識雕擂。啡邑。。我靠井赌，0:22了谤逼，還沒洗澡呢，下次有空再補上這部分知識仇穗，還是貼出廖雪峰網(wǎng)站關(guān)于這方面的知識吧 廖雪峰網(wǎng)站關(guān)于SMTP發(fā)送郵件流部。
差點忘了把發(fā)送郵件的截圖發(fā)出來：
