PYTHON实现DISCUZ论坛的自动POST登录发贴回帖

#-*-coding:utf-8-*-
import urllib2, urllib, COOKIElib
import re
import getpass
import sqlite3
import random
import time
 
class Discuz:
def __init__(self,user,pwd,args):
self.username = user
self.password = pwd
self.args = args
self.regex = {
‘loginreg‘:‘‘,
‘replyreg‘:‘‘,
‘tidreg‘: ‘[\s\S]+?‘
}
self.conn = None
self.cur = None
self.islogin = False
self.login()
self.InitDB()
 
def login(self):
try:
loginPage = urllib2.urlopen(self.args[‘loginurl‘]).read()
formhash = re.search(self.regex[‘loginreg‘], loginPage)
formhash = formhash.group(1)
#print ‘login formhash:‘, formhash
print ‘start login...‘
cj = COOKIElib.COOKIEJar()
opener = urllib2.build_opener(urllib2.HTTPCOOKIEProcessor(cj))
user_agent = ‘Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1; Mozilla/4.0 \
(compatible; MSIE 6.0; Windows NT 5.1; SV1) ; .NET CLR 2.0.507‘
opener.addheaders = [(‘User-agent‘, user_agent)]
urllib2.install_opener(opener)
logindata = urllib.urlencode({
‘COOKIEtime‘: 2592000,
‘formhash‘: formhash,
‘loginfield‘:‘username‘,
‘username‘: self.username,
‘password‘: self.password,
‘questionid‘: 0,
‘referer‘: self.args[‘referer‘]
})
request = urllib2.Request(self.args[‘loginsubmiturl‘],logindata)
respOnse= urllib2.urlopen(request)
self.islogin = True
print ‘login success...‘
except Exception,e:
print ‘loggin error: %s‘ % e
 
def PostReply(self, fid, tid, content):
try:
sql = "select * from post where fid=‘%s‘ and tid=‘%s‘" % (fid,tid)
self.cur.execute(sql)
if self.cur.rowcount == -1:
tidurl = self.args[‘tidurl‘] % tid
replysubmiturl = self.args[‘replysubmiturl‘] % (fid,tid)
tidPage = urllib2.urlopen(tidurl).read()
formhash = re.search(self.regex[‘replyreg‘], tidPage)
formhash = formhash.group(1)
#print ‘reply formhash:‘, formhash
print ‘start reply...‘
replydata = urllib.urlencode({
‘formhash‘: formhash,
‘message‘: content,
‘subject‘: ‘‘,
‘usesig‘:‘1‘
})
request = urllib2.Request(replysubmiturl,replydata)
respOnse= urllib2.urlopen(request)
sql = "insert into post values (‘%s‘, ‘%s‘, ‘%d‘)" % (fid, tid, 1)
self.cur.execute(sql)
self.conn.commit()
print ‘reply success for [%s]‘ % tidurl
else:
print ‘Skip! Thread:%s is already replied...‘ % tid
except Exception, e:
print ‘reply error: %s‘ % e
 
def GetTids(self, fid):
if self.islogin:
fidurl = self.args[‘fidurl‘] % fid
respOnse= urllib2.urlopen(fidurl)
cOntent= response.read()
tids = re.findall(self.regex[‘tidreg‘], content)
return tids
else:
print ‘Error Please Login...‘
 
def InitDB(self):
self.conn = sqlite3.connect(‘data.db‘)
self.cur = self.conn.cursor()
sql = ‘‘‘create table if not exists post (
fid text,
tid text,
replied integer)‘‘‘
self.cur.execute(sql)
self.conn.commit()
 
if __name__ == ‘__main__‘:
username = raw_input(‘username:‘).strip()
password = getpass.getpass(‘password:‘).strip()
args = {
‘loginurl‘: ‘http://www.xxx.com/logging.php?action=login‘,
‘loginsubmiturl‘: ‘http://www.xxx.com/logging.php?action=login&loginsubmit=yes‘,
‘fidurl‘: ‘http://www.xxx.com/forum-%s-1.html‘,
‘tidurl‘: ‘http://www.xxx.com/thread-%s-1-1.html‘,
‘replysubmiturl‘: ‘http://www.xxx.com/post.php?action=reply&replysubmit=yes&infloat=yes&handlekey=fastpost&fid=%s&tid=%s‘,
‘referer‘:‘http://www.xxx.com/index.php‘
}
dz = Discuz(username, password,args)
fid = ‘45‘
tids = dz.GetTids(‘45‘)
replylist = [
u‘不错，支持一下，呵呵‘,
u‘已阅，顶一下‘,
u‘看看，顶你，呵呵‘,
u‘多谢分享，顶一下‘,
u‘说的不错，支持一下‘,
u‘提着水桶到处转，哪里缺水哪里灌！ ‘,
u‘你太油菜了!‘
]
for tid in tids:
cOntent= random.choice(replylist)
cOntent= content.encode(‘gbk‘)
dz.PostReply(‘45‘,tid, content)
time.sleep(20)