python实现的登陆discuz!论坛通用代码分享

代码如下:

#coding:gbk
import urllib2,urllib,cookielib,re
”’
通用的登陆dz论坛
参数说明parms:
username:用户名(必填),
password :密码(必填),
domain:网站域名,注意格式必须是:http://www.xxx.xx/(必填),
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.php?mod=logging&action=login&infloat=yes&handlekey=login&inajax=1&ajaxtarget=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.php?mod=logging&action=login&loginsubmit=yes&handlekey=login&loginhash=lcab3&inajax=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/’ #另一个测试网站:http://bbs.bitscn.com/
try:
flag = login_dz(username=user,password=pwd,domain=dom)
print(flag)
except exception,e:
print(‘error:’,e)

Posted in 未分类