基于python实现的抓取腾讯视频所有电影的爬虫

我搜集了国内10几个电影网站的数据，里面近几十W条记录，用文本没法存，mongodb学习成本非常低，安装、下载、运行起来不会花你5分钟时间。

# -*- coding: utf-8 -*-
# by awakenjoys. my site: www.dianying.at
import re
import urllib2
from bs4 import BeautifulSoup
import string, time
import pymongo

NUM  = 0   #全局变量,电影数量
m_type = u''  #全局变量,电影类型
m_site = u'qq' #全局变量,电影网站

#根据指定的URL获取网页内容
def gethtml(url):
req = urllib2.Request(url)
response = urllib2.urlopen(req)
html = response.read()
return html

#从电影分类列表页面获取电影分类
def gettags(html):
global m_type
soup = BeautifulSoup(html)  #过滤出分类内容
#print soup
#<ul class="clearfix _group" gname="mi_type" gtype="1">
tags_all = soup.find_all('ul', {'class' : 'clearfix _group' , 'gname' : 'mi_type'})
#print len(tags_all), tags_all
#print str(tags_all[1]).replace('\n', '')

#<a _hot="tag.sub" class="_gtag _hotkey" href="http://v.qq.com/list/1_0_-1_-1_1_0_0_20_0_-1_0.html" title="动作" tvalue="0">动作</a>
re_tags = r'<a _hot=\"tag\.sub\" class=\"_gtag _hotkey\" href=\"(.+?)\" title=\"(.+?)\" tvalue=\"(.+?)\">.+?</a>'
p = re.compile(re_tags, re.DOTALL)

tags = p.findall(str(tags_all[0]))
if tags:
 tags_url = {}
 #print tags
 for tag in tags:
  tag_url = tag[0].decode('utf-8')
  #print tag_url
  m_type = tag[1].decode('utf-8')
  tags_url[m_type] = tag_url

else:
  print "Not Find"
return tags_url

#获取每个分类的页数
def get_pages(tag_url):
tag_html = gethtml(tag_url)
#div class="paginator
soup = BeautifulSoup(tag_html)  #过滤出标记页面的html
#print soup
#<div class="mod_pagenav" id="pager">
div_page = soup.find_all('div', {'class' : 'mod_pagenav', 'id' : 'pager'})
#print div_page #len(div_page), div_page[0]

#<a class="c_txt6" href="http://v.qq.com/list/1_2_-1_-1_1_0_24_20_0_-1_0.html" title="25"><span>25</span></a>
re_pages = r'<a class=.+?><span>(.+?)</span></a>'
p = re.compile(re_pages, re.DOTALL)
pages = p.findall(str(div_page[0]))
#print pages
if len(pages) > 1:
 return pages[-2]
else:
 return 1

def getmovielist(html):
soup = BeautifulSoup(html)

#<ul class="mod_list_pic_130">
divs = soup.find_all('ul', {'class' : 'mod_list_pic_130'})
#print divs
for div_html in divs:
 div_html = str(div_html).replace('\n', '')
 #print div_html
 getmovie(div_html)

def getmovie(html):
global NUM
global m_type
global m_site

#<h6 class="caption"> <a href="http://www.tudou.com/albumcover/Z7eF_40EL4I.html" target="_blank" title="徒步旅行队">徒步旅行队</a> </h6> <ul class="info"> <li class="desc">法国卖座喜剧片</li> <li class="cast"> </li> </ul> </div> <div class="ext ext_last"> <div class="ext_txt"> <h3 class="ext_title">徒步旅行队</h3> <div class="ext_info"> <span class="ext_area">地区: 法国</span> <span class="ext_cast">导演: </span> <span class="ext_date">年代: 2009</span> <span class="ext_type">类型: 喜剧</span> </div> <p class="ext_intro">理查德·达奇拥有一家小的旅游公司，主要经营 * 客到非洲大草原的旅游服务。六个 * 客决定参加理查德·达奇组织的到非洲的一...</p>

re_movie = r'<li><a class=\"mod_poster_130\" href=\"(.+?)\" target=\"_blank\" title=\"(.+?)\"><img.+?</li>'
p = re.compile(re_movie, re.DOTALL)
movies = p.findall(html)
if movies:
 conn = pymongo.Connection('localhost', 27017)
 movie_db = conn.dianying
 playlinks = movie_db.playlinks
 #print movies
 for movie in movies:
  #print movie
  NUM += 1
  print "％s : ％d" ％ ("=" * 70, NUM)
  values = dict(
   movie_title = movie[1],
   movie_url = movie[0],
   movie_site  = m_site,
   movie_type  = m_type
   )
  print values
  playlinks.insert(values)
  print "_" * 70
  NUM += 1
  print "％s : ％d" ％ ("=" * 70, NUM)

#else:
# print "Not Find"

def getmovieinfo(url):
html = gethtml(url)
soup = BeautifulSoup(html)

#pack pack_album album_cover
divs = soup.find_all('div', {'class' : 'pack pack_album album_cover'})
#print divs[0]

#<a href="http://www.tudou.com/albumplay/9NyofXc_lHI/32JqhiKJykI.html" target="new" title="《血滴子》独家纪录片" wl="1"> </a>
re_info = r'<a href=\"(.+?)\" target=\"new\" title=\"(.+?)\" wl=\".+?\"> </a>'
p_info = re.compile(re_info, re.DOTALL)
m_info = p_info.findall(str(divs[0]))
if m_info:
 return m_info
else:
 print "Not find movie info"

return m_info

def insertdb(movieinfo):
global conn
movie_db = conn.dianying_at
movies = movie_db.movies
movies.insert(movieinfo)

if __name__ == "__main__":
global conn

tags_url = "http://v.qq.com/list/1_-1_-1_-1_1_0_0_20_0_-1_0.html"
#print tags_url
tags_html = gethtml(tags_url)
#print tags_html
tag_urls = gettags(tags_html)
#print tag_urls

for url in tag_urls.items():
 print str(url[1]).encode('utf-8') #,url[0]
 maxpage = int(get_pages(str(url[1]).encode('utf-8')))
 print maxpage

for x in range(0, maxpage):
  #http://v.qq.com/list/1_0_-1_-1_1_0_0_20_0_-1_0.html
  m_url = str(url[1]).replace('0_20_0_-1_0.html', '')
  movie_url = "％s％d_20_0_-1_0.html" ％ (m_url, x)
  print movie_url
  movie_html = gethtml(movie_url.encode('utf-8'))
  #print movie_html
  getmovielist(movie_html)
  time.sleep(0.1)