Python爬虫(二)--糗事百科

  1. #coding=utf-8
  2. import urllib
  3. import urllib2
  4. import re
  5. import thread
  6. import time
  7. class QSBK:
  8. # 初始化方法,定义变量
  9. def __init__(self):
  10. self.pageIndex = 1
  11. self.user_agent = 'Mozilla/5.0 (Windows NT 10.0; Win64; x64)'
  12. # 初始化headers
  13. self.headers = {'User-Agent':self.user_agent}
  14. # 存放段子的变量,每一个元素是每一页的段子
  15. self.stories = []
  16. # 存放程序是否继续运行的变量
  17. self.enable = False
  18. def getPage(self,pageIndex):
  19. try:
  20. url = 'http://www.qiushibaike.com/hot/page/' + str(pageIndex)
  21. # 构建请求的request
  22. request = urllib2.Request(url,headers = self.headers)
  23. # 利用urlopen获取页面代码
  24. response = urllib2.urlopen(request)
  25. # 将页面转化为UTF-8编码
  26. pageCode = response.read().decode('utf-8')
  27. return pageCode
  28. except urllib2.URLError,e:
  29. # if hasattr(e, "code"):
  30. # print e.code
  31. if hasattr(e, "reason"):
  32. print u"连接糗事百科失败,错误原因",e.reason
  33. return None
  34. def getPageIntems(self,pageIndex):
  35. pageCode = self.getPage(pageIndex)
  36. if not pageCode:
  37. print "页面加载失败..."
  38. return None
  39. pattern = re.compile('h2>(.*?)</h2.*?<img(.*?)alt.*?content">(.*?)</.*?number">(.*?)</i', re.S)
  40. items = re.findall(pattern,pageCode)
  41. # 用来存储每页的段子
  42. pageStories = []
  43. # 遍历正则表达式匹配的信息
  44. for item in items:
  45. replaceBR = re.compile('<br/>')
  46. text = re.sub(replaceBR,"\n",item[2])
  47. # item[0]--段子发布者,item[2]--段子内容,item[3]--段子点赞数
  48. pageStories.append([item[0].strip(),text.strip(),item[3].strip()])
  49. return pageStories
  50. # 加载并提取页面内容,加入到列表中
  51. def loadPage(self):
  52. # 如果当前未看的页数少于2页,则加载新一页
  53. if self.enable == True:
  54. if len(self.stories) < 2:
  55. # 获取新一页
  56. pageStories = self.getPageIntems(self.pageIndex)
  57. # 将该页的段子存到全局变量列表list中
  58. if pageStories:
  59. self.stories.append(pageStories)
  60. # 获取完之后页码索引+1,表示下次读取下一页
  61. self.pageIndex += 1
  62. # 调用该方法,每次敲回车打印输出一个段子
  63. def getOneStory(self,pageStories,page):
  64. # 遍历当前页的每一个段子
  65. for story in pageStories:
  66. # 等待用户输入
  67. input = raw_input()
  68. # 每当输入回车一次,判断一下是否要加载新页面
  69. self.loadPage()
  70. if input == "Q":
  71. # 输入Q则程序结束
  72. self.enable = False
  73. return
  74. print u"第%d页\t发布人:%s\t赞:%s\n%s"%(page,story[0],story[2],story[1])
  75. # 开始方法
  76. def start(self):
  77. print u"正在读取糗事百科,按回车查看新段子。按Q退出"
  78. # 变量为True,程序开始运行
  79. self.enable = True
  80. # 先加载第一页内容
  81. self.loadPage()
  82. # 局部变量,控制当前页数
  83. nowPage = 0
  84. while self.enable:
  85. if len(self.stories)>0:
  86. # 从全局List中获取当前页的段子
  87. pageStories = self.stories[0]
  88. # 当前读到的页数+1
  89. nowPage += 1
  90. # 将全局List中的第一个元素删掉
  91. del self.stories[0]
  92. # 输出该页的段子
  93. self.getOneStory(pageStories,nowPage)
  94. spider = QSBK()
  95. spider.start()




posted @ 2016-10-17 20:43  ldbmcs  阅读(277)  评论(0编辑  收藏  举报