#coding:gbkimport urllib2,urllib,cookielib,re''' 通用的登陆DZ论坛 参数说明parms: username:用户名(必填), password :密码(必填), domain:网站域名注意格式必须是：(必填), answer:问题答案, questionid:问题ID, referer:跳转地址 这里使用了可变关键字参数(相关信息可参考手册)'''def login_dz(**parms): #初始化 parms_key = ['domain','answer','password','questionid','referer','username'] arg = {} for key in parms_key: if key in parms:
arg[key] = parms[key] else:
arg[key] = ''
#cookie设置 cookieFile = './kan_cookies.dat' cookie = cookielib.LWPCookieJar() opener = urllib2.build_opener(urllib2.HTTPCookieProcessor(cookie)) #获取formhash pre_login = arg['domain']+'member.phpmod=loggingaction=logininfloat=yeshandlekey=logininajax=1ajaxtarget=fwin_content_login' c = opener.open(pre_login).read() cookie.save(cookieFile) patt = re.compile(r'.*name="formhash".*value="(.*)".*') formhash = patt.search(c) if not formhash: raise Exception('GET formhash Fail!') formhash = formhash.group(1) #登陆 postdata = { 'answer':arg['answer'], 'formhash':formhash, 'password':arg['password'], 'questionid':0 if arg['questionid']=='' else arg['questionid'], 'referer':arg['domain'] if arg['referer']=='' else arg['referer'], 'username':arg['username'], } postdata = urllib.urlencode(postdata) req = urllib2.Request( url= arg['domain']+'member.phpmod=loggingaction=loginloginsubmit=yeshandlekey=loginloginhash=LCaB3inajax=1', data=postdata ) c = opener.open(req).read(300) flag = '登陆失败 %s'%arg['username'] if 'succeedhandle_login' in c: flag = True return flag#使用例子：基本参数登陆user='xxx'pwd='xxx'dom='http://www.discuz.net/' #另一个测试网站：: flag = login_dz(username=user,password=pwd,domain=dom) print(flag)except Exception,e: print('Error:',e)