学于:http://cuiqingcai.com/993.html
#-*-coding?utf-8 -*-
import urllib.request
import urllib.parse
import re
class Tool:
#去除img标签,7位长空格
removeImg = re.compile('<img.*?>| {7}|')
#删除超链接标签
removeAddr = re.compile('<a.*?>|</a>')
#把换行的标签换为\n
replaceLine = re.compile('<tr>|<div>|</div>|</p>')
#将表格制表<td>替换为\t
replaceTD= re.compile('<td>')
#把段落开头换为\n加空两格
replacePara = re.compile('<p.*?>')
#将换行符或双换行符替换为\n
replaceBR = re.compile('<br><br>|<br>')
#将其余标签剔除
removeExtraTag = re.compile('<.*?>')
def replace(self,x):
x = re.sub(self.removeImg,"",x)
x = re.sub(self.removeAddr,"",x)
x = re.sub(self.replaceLine,"\n",x)
x = re.sub(self.replaceTD,"\t",x)
x = re.sub(self.replacePara,"\n ",x)
x = re.sub(self.replaceBR,"\n",x)
x = re.sub(self.removeExtraTag,"",x)
#strip()将前后多余内容删除
return x.strip()
class BDTB:
def __init__(self,baseUrl,seeLz,floorTag):
self.baseURL = baseUrl
self.seeLz = '?see_lz='+str(seeLz)
self.tool = Tool()
self.file = None
self.floor = 1
self.defaultTitle = u"百度贴吧"
self.floorTag = floorTag
def getPage(self,pageNum):
try:
url = self.baseURL+self.seeLz+'&pn='+str(pageNum)
request = urllib.request.Request(url)
response = urllib.request.urlopen(request)
#print(response.read())
return response.read().decode('utf-8')
except urllib.request.URLError as e:
if hasattr(e,'reason'):
print('link BDTB error')
return None
def getTitle(self,page):
pattern = re.compile('<h3 class="core_title_txt.*?title="(.*?)" style',re.S)
result = re.search(pattern,page)
if result:
print(result.group(1))
return result.group(1).strip()
else:
return None
def getNum(self,page):
pattern = re.compile('<li class="l_reply_num".*?margin-right:3px">.*?</span>.*?<span class="red">(.*?)</span>',re.S)
result = re.search(pattern , page)
if result:
print(result.group(1))
return result.group(1).strip()
else:
return None
def getContent(self,page):
pattern = re.compile('<cc>.*?<div id=.*?j_d_post_content ">(.*?)</div>',re.S)
result = re.findall(pattern,page)
contents = []
if result:
for line in result:
temp = "\n" + self.tool.replace(line.strip()) + "\n"
print(self.tool.replace(line.strip()))
contents.append(temp)
return contents
else:
return None
def setFileTitle(self,title):
if title is not None:
self.file = open(title + ".txt" , "w+")
else:
self.file = open(self.defaultTitle+".txt", "w+")
def writeData(self,contents):
for item in contents:
if self.floorTag == '1':
floorLine = "\n" + str(self.floor) + u"-----------------------------------------------------------------------------------------\n"
self.file.write(floorLine)
self.file.write(item)
self.floor += 1
def start(self):
indexPage = self.getPage(1)
pageNum =self.getNum(indexPage)
title = self.getTitle(indexPage)
self.setFileTitle(title)
if pageNum == None:
print("找不到")
return
try:
print("该帖子共有"+str(pageNum)+"页")
for i in range(1,int(pageNum)+1):
print("正在写入第" + str(i) + "页数据")
page = self.getPage(i)
contents =self.getContent(page)
self.writeData(contents)
except IOError as e:
print("异常")
finally:
print("下载结束")
print("请输入帖子代号:")
baseURL = 'http://tieba.baidu.com/p/'+str(input(u'http://tieba.baidu.com/p/'))
seeLz = input("是否只获取楼主发言,是输入1,否输入0\n")
floorTag = input("是否写入楼层信息,是输入1,否输入0\n")
dbtb = BDTB(baseURL,seeLz,floorTag)
dbtb.start()