python 之post、get与cookie实战

　　项目名称：登陆考勤管理系统爬取个人考勤信息并写入excel表格

编写目的：

　　公司经常要统计员工的考勤信息，而员工每次都要登陆考勤系统，再复制相关信息出来，贴到EXCEL，再转给统计人员，统计人员再挨个核对，麻烦无比，实在是看不下去了。我擦。。。。。

　　由于登陆的是内网的考勤系统，出了公司就登陆不了，所以本篇文章仅做参考，来体验一下cookie、post与get登陆使用。

先说用用到哪些知识吧：

　　1、tkinter Gui编程，写爬虫没有GUI怎么能行

　　2、cookie

　　3、post,get

其实不多是吧，简单。

要爬取一个网站，总得要矢爬的网站登陆地址，及登陆信息吧。分析一下看看。。。



好的，我们获取了如下信息：

　　登陆地址：LoginUrl = 'http://17.xx.xx.xx/j_acegi_security_check'

　　post提交：post_data = {'j_username':'your_username','j_password':'your_password'}

那就开工 ，写一下登陆代码吧

#!/usr/bin/env python
# _*_ coding:utf-8 _*_

import urllib
import urllib2
import cookielib

LoginUrl = 'http://17.xx.xx.xx/j_acegi_security_check'
post_data = {'j_username':'your_username','j_password':'your_password'}

# 定义一个类
class kaoqin_login(object):
def __init__(self):
self.cj = cookielib.CookieJar() # 初始化一个cookie补全，用于存储cookie
self.handler = urllib2.HTTPCookieProcessor(self.cj) # 创建一个cookie处理器
self.opener = urllib2.build_opener(self.handler) # 创建一个功能强大的opener

def login(self,url,post): # post提交
post_data = urllib.urlencode(post) # 编码post数据为标准格式
req = urllib2.Request(url,post_data) # 做为data参数传给Request对象,此处也可以写成data=post_data
response = self.opener.open(req) # 登陆网站，获取返回结果
print response.url

if __name__ == '__main__':
cls = kaoqin_login()
cls.login(LoginUrl,post_data)

返回信息如下： http://17.xx.xx.xx/;jsessionid=7AC236C4B9C5E653208D59ECCE55E8EF[/code] 
　　

好，登陆成功了，之后怎么办呢？嗯，对了，找一下考勤页面在哪里，分析一下吧



瞧，真实页面的信息就这些了。自己把url和post信息整理出来吧，注意，提交表单信息要写成字典，url呢，问号之前的都是了。

示例代码

#!/usr/bin/env python
# _*_ coding:utf-8 _*_

import urllib
import urllib2
import cookielib
import json

LoginUrl = 'http://17.xx.xx.xx/j_acegi_security_check'
post_data = {'j_username':'your_username','j_password':'your_password'}

# 定义一个类
class kaoqin_login(object):
def __init__(self):
self.cj = cookielib.CookieJar() # 初始化一个cookie补全，用于存储cookie
self.handler = urllib2.HTTPCookieProcessor(self.cj) # 创建一个cookie处理器
self.opener = urllib2.build_opener(self.handler) # 创建一个功能强大的opener

def login(self,url,post): # post提交
post_data = urllib.urlencode(post) # 编码post数据为标准html格式
req = urllib2.Request(url,post_data) # 做为data参数传给Request对象,由request做post数据隐式提交,此处也可以写成data=post_data
response = self.opener.open(req) # 登陆网站，获取返回结果
print response.url

def get_response(self,url,post):  # get方式提交数据
get = urllib.urlencode(post)  # 将提交的数据编码成html标准格式
response = self.opener.open(url,get)  # 将标准的编码数据放到url后面，变成真正的url地址
try:
JsData = json.loads(response.read())
return JsData['datas']
except Exception as e:
return False

if __name__ == '__main__':
cls = kaoqin_login()
cls.login(LoginUrl,post_data)
Login = 'http://17.xx.xx.xx/km/attendance/km_attendance_main/kmAttendanceMain.do'
data ={
'method':'data'
,'categoryId':''
,'q.s_raq':0.5949037911774411
,'pageno':2
,'rowsize':15
,'orderby':'fdAttendanceDate'
,'ordertype':'down'
,'s_ajax':'true'
}
jsdata = cls.get_response(Login,data)
print jsdata


　　好了，数据取到了，发一下全部的代码吧.

上面的代码是我重点想说的，下面的代码就不是那么重要了，关于excel的读取，博客里有详细的解释，GUI开发也有相关文档。

#!/usr/bin/env python
# _*_ coding:utf-8 _*_
import urllib
import urllib2
import cookielib
import json,re,time

'''
因为脚本只取了前4页数据，所以最好用来爬取本月的考勤信息
如需爬取其它月的信息，需要更改代码
for i in range(1,5)  将5改为更大，最大15

'''

# ==============框口区==================
from Tkinter import *
from ScrolledText import ScrolledText
import tkFileDialog
root = Tk()
root.iconbitmap('../data/spider_128px_1169260_easyicon.net.ico')
root.title('邮储个人考勤记录爬虫')
# 第一行
Label(root,text='考勤日期',fg='#8F8C8F').grid(row=0,column=0)
var_date1 = StringVar()
var_date1.set('2016-01-01')
var_date2 = StringVar()
var_date2.set('2016-01-31')
date1 = Entry(root,fg='red',bg='#FFFACD',textvariable=var_date1)
date1.grid(row=0,column=1)
Label(root,text='至',fg='#8F8C8F').grid(row=0,column=2)
date2 =Entry(root,fg='red',bg='#FFFACD',textvariable=var_date2)
date2.grid(row=0,column=3)
# 第二行
Label(root,text='用户密码文件',fg='#8F8C8F').grid(row=1,column=0)
var = StringVar()
def get_file():
filename = tkFileDialog.askopenfilename(initialdir='c:')
var.set(filename)
def get_file_des():
filename = tkFileDialog.askopenfilename(initialdir='c:')
ent_des.set(filename)
var.set(r'C:\spider_dev\data\kaoqin_user_pwd.txt')
Ent = Entry(root,textvariable=var).grid(row=1,column=1)
Button(root,text='选择文件',command=get_file,fg='#006400').grid(row=1,column=2)
ent_des = StringVar()
timpsta = '%d'%(time.time()*1000)
ent_des.set(r'C:\spider_dev\data\kaoqin_%s.xls'%str(timpsta))
Ent_d = Entry(root,textvariable=ent_des).grid(row=2,column=1,sticky=N)
Button(root,text='目标文件',command=get_file_des,fg='#006400').grid(row=2,column=2,sticky=N)

T_logi_info = ScrolledText(root,width=20)
T_logi_info.grid(row=2,column=0)

# 标签
v_lable = StringVar()
v_lable.set('错误记录...')
Label(root,textvariable=v_lable,fg='red').grid(row=3,column=0)
# 信息框
text = ScrolledText(root,bg='#87CEFA')
text.grid(row=0,column=4,rowspan=3)

# ==============框口区==================

# --------------爬虫区-----------------
class kaoqin_login(object):
def __init__(self):
self.cj = cookielib.CookieJar()
self.handler = urllib2.HTTPCookieProcessor(self.cj)
self.opener = urllib2.build_opener(self.handler)

def login(self,url,post):
post_data = urllib.urlencode(post)
req = urllib2.Request(url,post_data)
response = self.opener.open(req)
if response.code == 200:
return True
else:
return False

def get_response(self,url,post):
get = urllib.urlencode(post)
response = self.opener.open(url,get)
try:
JsData = json.loads(response.read())
return JsData['datas']
except Exception as e:
return False

def post_page(self,pages):
get={
'method':'data'
,'categoryId':''
,'q.s_raq':0.9518796035349586
,'pageno':pages
,'rowsize':15
,'orderby':'fdAttendanceDate'
,'ordertype':'down'
,'s_ajax':'true'
}
return get

def get_data_file(self,filename):
with open(filename,'rb') as f:
data = f.read().split('\r\n')
return data

# --------------爬虫区-----------------

# <------------Excel文件数据写入-------->
from openpyxl.workbook import Workbook
from openpyxl.writer.excel import ExcelWriter
from openpyxl.cell import get_column_letter

wb = Workbook()
ew = ExcelWriter(workbook=wb)
ws = wb.worksheets[0]
ws.title = u'第一个sheet'

# 数据开始写入的行、列、无组数据及文件名
def write_2_excel(r,c,t_data,FileName):
i = r
for record in t_data:
for x in range(c,len(record)+c):
col = get_column_letter(x)
ws.cell('%s%s'%(col,i)).value ='%s'%record[x-c]
i+=1
ew.save(filename=FileName)
# <------------Excel文件数据写入-------->

# ************主函数*****************
if __name__ == '__main__':
def Spider():
#初始化实例
cls = kaoqin_login()
if var.get():
List_username = cls.get_data_file(var.get())
else:
v_lable.set('密码文件对吗？？？')
#出勤状态，如正常、迟到
Real_Data = []
#按用户获取数据
for each_user in List_username:
#标题数据，比如excel的头部，日期，星期、上下午
Title_Data = []
#网站登陆用户信息
Pno,Pname,username,password = each_user.split(',')[0],each_user.split(',')[1],each_user.split(',')[2],each_user.split(',')[3]
LoginUrl = 'http://17.xx.xx.xx/j_acegi_security_check'
LoginPost ="{'j_username':'%s','j_password':'%s'}"%(username,password)
LoginPost = eval(LoginPost)
if cls.login(LoginUrl,LoginPost):
#进入考勤页面
Kq_Url = 'http://17.xx.xx.xx/km/attendance/km_attendance_main/kmAttendanceMain.do'
tmp1=[]
tmp2=[]
tmp3=[]
tmp4=[]
tmp4.append(Pno)
tmp4.append(Pname)
#考勤post列表，第一页到第四页数据
for i in range(1,5):
jsret = cls.get_response(Kq_Url,cls.post_page(i))
if jsret:
pass
else:
T_logi_info.insert(END,Pname+'登陆失败'+ '\n')
T_logi_info.see(END)
break
#对考勤数据处理
for each_day in jsret:
v_date=v_week=v_Bc=v_status=v_name=''
start_date = var_date1.get()
end_date = var_date2.get()
if start_date and end_date:
for item_dic in each_day:
col = item_dic['col']
if col == 'fdAttendanceDate':
v_date = item_dic['value'].strip()
if col == 'fdWeek':
v_week = item_dic['value'].strip()
if col == 'fdBc':
v_Bc = item_dic['value'].strip()
if col == 'fdStatus':
v_status = item_dic['value'].strip()
v_status = re.search('>(.*?)<',v_status).group(1)
if col == 'fdPerson.fdName':
v_name = item_dic['value']

if v_date >= start_date and v_date <= end_date:
tmp1.append(v_date)
tmp2.append(v_week)
tmp3.append(v_Bc)
tmp4.append(v_status)
text.insert(END, v_name +',' + v_date+','+v_week+','+v_Bc+','+v_status+'\n')
text.see(END)
text.update()
else:
v_lable.set('时间不对吧...')
pass
Title_Data.append(tmp1)
Title_Data.append(tmp2)
Title_Data.append(tmp3)
Real_Data.append(tmp4)
write_2_excel(1,3,Title_Data,ent_des.get())
write_2_excel(4,1,Real_Data,ent_des.get())

#第三行
Button(root,text='开始爬取',fg='#3CB371',command=Spider).grid(row=3,column=4)
root.mainloop()


　　好了，看看一下实际效果吧