python模拟登录知乎

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

import json
import urllib
import urllib2
import cookielib

class Zhihu_login(object):
"""登陆知乎的一个演示程序"""

def __init__(self):
self.test_url = "http://www.zhihu.com/topic"
self.login_url = "http://www.zhihu.com/login/email"
self.checkcode_url = "http://www.zhihu.com/captcha.gif"
"""创建一个带cookie的opener"""
cj = cookielib.CookieJar()
self.opener = urllib2.build_opener(urllib2.HTTPCookieProcessor(cj));

def getCheckCode(self):
"""获取验证码, 保存在当前目录下"""
pic = open("checkcode.gif", "w")
"""
验证码一定要用self.opener打开,
因为服务器需要检验cookie是否与当前提交的验证码匹配,
opener请求的时候可以将cookie一起发过去。
"""
resp = self.opener.open(self.checkcode_url)
print >> pic, resp.read()
pic.close()

def login(self):
post_data = {}
post_data["email"] = raw_input("输入用户名：\n")
post_data["password"] = raw_input("输入密码：\n")
self.getCheckCode()
info = raw_input("输入验证码，看不清输入'?'再换一张\n")
while(info == '?'):
self.getCheckCode()
info = raw_input("输入验证码，看不清输入'?'再换一张\n")
post_data["captcha"] = info
post_data = urllib.urlencode(post_data)
resp = self.opener.open(self.login_url, post_data)
status = json.loads(resp.read())
if status["r"] == 0:
print "登陆成功!"
print json.dumps(status, ensure_ascii=False).encode("utf-8", "ignore")
#print "开始测试。。。"
#self.test()
else:
print "登录失败!"
print "失败信息如下："
print json.dumps(status, ensure_ascii=False).encode("utf-8", "ignore")

def test(self):
"""看看是否能在源代码里面查找到你的知乎名字，如果能说明确实成功了。"""
resp = self.opener.open(self.test_url)
print resp.read()

if __name__ == "__main__":
Zhihu_login().login()