##code

# -*- coding: utf-8 -*-

#!/usr/bin/python

'''

name: get_pay_info

function: 自动获取流量信息

lib: requests, lxml, tesseract

parameters:

BASE_URL---基本的url，在此网址获取用户名，密码和验证码图片

FORM_URL---登陆的url

PAY_INFO_URL---获取信息的url

USERNAME(username)---用户名

PASSOWORD(passoword)---密码

checkcode---验证码

'''

import re

import time

import os

import requests

from lxml import html

USERNAME = "*********"

PASSOWORD = "*******"

BASE_URL = "http://zyzfw.xidian.edu.cn:8800/"

FORM_URL = "http://zyzfw.xidian.edu.cn:8800/index.php?action=login"

PAY_INFO_URL = "http://zyzfw.xidian.edu.cn:8800/index.php"

TMP_DIR = os.path.expanduser("~/.xidian/")

IMG_PATH = os.path.join(TMP_DIR, "img.jpg")

TEXT_PATH = os.path.join(TMP_DIR, "result.txt")

'''

name: make_data_and_cookies

function: 获取登陆所需的数据和cookies

return: data,cookies

'''

def make_data_and_cookies():

"""make the post data(including vcode) and get cookies"""

vcode = ''

while len(vcode) is not 4:

r = requests.get(BASE_URL)

doc = html.document_fromstring(r.text)

vcode_link = doc.cssselect('form img')[3].get('src')

img_url = BASE_URL + vcode_link

img = requests.get(img_url)

# write to the image file

with open(IMG_PATH, 'w') as f:

f.write(img.content)

# using tesseract to get the vcode img value

try:

os.popen('tesseract %s %s' % (IMG_PATH, TEXT_PATH[:-4]))

except:

print "open tesseract error"

with open(TEXT_PATH) as f:

vcode = f.read().strip('\n')

data = {

"username": USERNAME,

"password": PASSOWORD,

"checkcode": vcode,

"ts": "login"

}

return data, r.cookies

'''

name: submit_form

parameters: data,cookies(其意义与上函数类似)

function: 模仿form进行登陆

return: None

'''

def submit_form(data, cookies):

"""submit the login form so you're logined in"""

form_action_url = FORM_URL

r = requests.post(form_action_url, data=data, cookies=cookies)

'''

name: get_info

parameters: cookies

function: 获取信息,并打印出来

return: None

'''

def get_info(cookies):

"""retrieve the data using the cookies"""

info_url = PAY_INFO_URL

r = requests.get(info_url, cookies=cookies)

doc = html.document_fromstring(r.text)

#items = re.findall('

(.*?)',r.text, re.S)

messageList = doc.cssselect('div table tbody tr td')[41].text_content()

lMsg = messageList.strip().split("\n")

for i in lMsg:

print i.strip()

if __name__ == '__main__':

if not os.path.exists(TMP_DIR):

os.mkdir(TMP_DIR)

#循环，直至成功才跳出循环

while True:

data, cookies = make_data_and_cookies()

submit_form(data, cookies)

time.sleep(1)

try:

get_info(cookies)

break

except:

time.sleep(1)