baiduclient.py

复制代码 代码如下:
import urllib.parse
import gzip
import json
import re
from http.client import HTTPConnection
from htmlutils import TieBaParser
import httputils as utils

# 请求头
headers = dict()
headers[\”Connection\”] = \”keep-alive\”
headers[\”Cache-Control\”] = \”max-age=0\”
headers[\”Accept\”] = \”text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8\”
headers[\”User-Agent\”] = \”Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/32.0.1700.107 Safari/537.36\”
headers[\”Content-Type\”] = \”application/x-www-form-urlencoded\”
headers[\”Accept-Encoding\”] = \”gzip,deflate,sdch\”
headers[\”Accept-Language\”] = \”zh-CN,zh;q=0.8\”
headers[\”Cookie\”] = \”\”

# cookie
cookies = list()

# 个人信息
userInfo = {}

def login(account, password):
    \’\’\’登录\’\’\’
    global cookies
    headers[\”Host\”] = \”wappass.baidu.com\”
    body = \”username={0}&password={1}&submit=%E7%99%BB%E5%BD%95&quick_user=0&isphone=0&sp_login=waprate&uname_login=&loginmerge=1&vcodestr=&u=http%253A%252F%252Fwap.baidu.com%253Fuid%253D1392873796936_247&skin=default_v2&tpl=&ssid=&from=&uid=1392873796936_247&pu=&tn=&bdcm=3f7d51b436d12f2e83389b504fc2d56285356820&type=&bd_page_type=\”
    body = body.format(account, password)
    conn = HTTPConnection(\”wappass.baidu.com\”, 80)
    conn.request(\”POST\”, \”/passport/login\”, body, headers)
    resp = conn.getresponse()
    cookies += utils.getCookiesFromHeaders(resp.getheaders())
    utils.saveCookies(headers, cookies)
    # 登录成功会返回302
    return True if resp.code == 302 else False
   

def getTieBaList():
    \’\’\’获取已关注的贴吧列表\’\’\’
    conn = HTTPConnection(\”tieba.baidu.com\”, 80)
    conn.request(\”GET\”, \”/mo/m?tn=bdFBW&tab=favorite\”, \”\”, headers)
    resp = conn.getresponse()   
    tieBaParser = TieBaParser()
    tieBaParser.feed(resp.read().decode())
    tbList = tieBaParser.getTieBaList()
    return tbList
   

def getSignInfo(tieBaName):
    \’\’\’获取贴吧签到信息\’\’\’
    queryStr = urllib.parse.urlencode({\”kw\”:tieBaName, \”ie\”:\”utf-8\”, \”t\”:0.571444})
    conn = HTTPConnection(\”tieba.baidu.com\”, 80)
    conn.request(\”GET\”, \”/sign/loadmonth?\” + queryStr, \”\”, headers)
    data = gzip.decompress(conn.getresponse().read()).decode(\”GBK\”)
    signInfo = json.loads(data)
    return signInfo

    
tbsPattern = re.compile(\’\”tbs\” value=\”.{20,35}\”\’)

def signIn(tieBaName):
    \’\’\’签到\’\’\’
    # 获取页面中的参数tbs
    conn1 = HTTPConnection(\”tieba.baidu.com\”, 80)
    queryStr1 = urllib.parse.urlencode({\”kw\”: tieBaName})
    conn1.request(\”GET\”, \”/mo/m?\” + queryStr1, \”\”, headers)
    html = conn1.getresponse().read().decode()
    tbs = tbsPattern.search(html).group(0)[13:-1]
    # 签到
    conn2 = HTTPConnection(\”tieba.baidu.com\”, 80)
    body = urllib.parse.urlencode({\”kw\”:tieBaName, \”tbs\”:tbs, \”ie\”:\”utf-8\”})
    conn2.request(\”POST\”, \”/sign/add\” , body , headers)
    resp2 = conn2.getresponse()
    data = json.loads((gzip.decompress(resp2.read())).decode())
    return data
   

def getUserInfo():
    \’\’\’获取个人信息\’\’\’
    headers.pop(\”Host\”)
    conn = HTTPConnection(\”tieba.baidu.com\”, 80)
    conn.request(\”GET\”, \”/f/user/json_userinfo\”, \”\”, headers)
    resp = conn.getresponse()
    data = gzip.decompress(resp.read()).decode(\”GBK\”)
    global userInfo
    userInfo = json.loads(data)

if __name__ == \”__main__\”:
    account = input(\”请输入帐号:\”)
    password = input(\”请输入密码:\”)

    ok = login(account, password)
    if ok:
        getUserInfo()
        print(userInfo[\”data\”][\”user_name_weak\”] + \”~~~登录成功\”, end=\”\\n——\\n\”)
        for tb in getTieBaList():
            print(tb + \”吧:\”)
            signInfo = signIn(tb)
            if signInfo[\”no\”] != 0:
                print(\”签到失败!\”)
                print(signInfo[\”error\”])
            else:
                print(\”签到成功!\”)
                print(\”签到天数:\” + str(signInfo[\”data\”][\”uinfo\”][\”cout_total_sing_num\”]))
                print(\”连续签到天数:\” + str(signInfo[\”data\”][\”uinfo\”][\”cont_sign_num\”]))
            print(\”——\”)
    else:
        print(\”登录失败\”)

htmlutils.py

复制代码 代码如下:
\’\’\’
Created on 2014-2-20

@author: Vincent
\’\’\’

from html.parser import HTMLParser

class TieBaParser(HTMLParser):
    def __init__(self):
        HTMLParser.__init__(self)
        self.tieBaList = list()
        self.flag = False

    def getTieBaList(self):
        return self.tieBaList

    def handle_starttag(self, tag, attrs):
        if tag == \”a\”:
            for name , value in attrs:
                if name == \”href\” and \”m?kw=\” in value:
                    self.flag = True

    def handle_data(self, data):
        if self.flag:
            self.tieBaList.append(data)
            self.flag = False

httputils.py

复制代码 代码如下:
\’\’\’
Created on 2014-2-20

@author: Vincent
\’\’\’
def getCookiesFromHeaders(headers):
    \’\’\’从http响应中获取所有cookie\’\’\’
    cookies = list()
    for header in headers:
        if \”Set-Cookie\” in header:
            cookie = header[1].split(\”;\”)[0]
            cookies.append(cookie)
    return cookies

def saveCookies(headers, cookies):
    \’\’\’保存cookies\’\’\’
    for cookie in cookies:
        headers[\”Cookie\”] += cookie + \”;\”

def getCookieValue(cookies, cookieName):
    \’\’\’从cookies中获取指定cookie的值\’\’\’
    for cookie in cookies:
        if cookieName in cookie:
            index = cookie.index(\”=\”) + 1
            value = cookie[index:]
            return value

def parseQueryString(queryString):
    \’\’\’解析查询串\’\’\’
    result = dict()
    strs = queryString.split(\”&\”)
    for s in strs:
        name = s.split(\”=\”)[0]
        value = s.split(\”=\”)[1]
        result[name] = value
    return result