|
| 1 | +#!/usr/bin/env python |
| 2 | +# encoding=utf-8 |
| 3 | +import requests,re |
| 4 | +import codecs |
| 5 | +from bs4 import BeautifulSoup |
| 6 | +from openpyxl import Workbook |
| 7 | +wb = Workbook() |
| 8 | +dest_filename = '电影.xlsx' |
| 9 | +ws1 = wb.active |
| 10 | +ws1.title = "电影top250" |
| 11 | + |
| 12 | +DOWNLOAD_URL = 'http://movie.douban.com/top250/' |
| 13 | + |
| 14 | + |
| 15 | +def download_page(url): |
| 16 | + """获取url地址页面内容""" |
| 17 | + headers = { |
| 18 | + 'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_11_2) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/47.0.2526.80 Safari/537.36' |
| 19 | + } |
| 20 | + data = requests.get(url, headers=headers).content |
| 21 | + return data |
| 22 | + |
| 23 | + |
| 24 | +def get_li(doc): |
| 25 | + soup = BeautifulSoup(doc, 'html.parser') |
| 26 | + ol = soup.find('ol', class_='grid_view') |
| 27 | + name = [] #名字 |
| 28 | + star_con = [] #评价人数 |
| 29 | + score = [] #评分 |
| 30 | + info_list = [] #短评 |
| 31 | + for i in ol.find_all('li'): |
| 32 | + detail = i.find('div', attrs={'class': 'hd'}) |
| 33 | + movie_name = detail.find('span', attrs={'class': 'title'}).get_text() #电影名字 |
| 34 | + level_star = i.find('span',attrs={'class':'rating_num'}).get_text() #评分 |
| 35 | + star = i.find('div',attrs={'class':'star'}) |
| 36 | + star_num = star.find(text=re.compile('评价')) #评价 |
| 37 | + |
| 38 | + info = i.find('span',attrs={'class':'inq'}) #短评 |
| 39 | + if info: #判断是否有短评 |
| 40 | + info_list.append(info.get_text()) |
| 41 | + else: |
| 42 | + info_list.append('无') |
| 43 | + score.append(level_star) |
| 44 | + |
| 45 | + |
| 46 | + name.append(movie_name) |
| 47 | + star_con.append(star_num) |
| 48 | + page = soup.find('span', attrs={'class': 'next'}).find('a') #获取下一页 |
| 49 | + if page: |
| 50 | + return name,star_con,score,info_list,DOWNLOAD_URL + page['href'] |
| 51 | + return name,star_con,score,info_list,None |
| 52 | + |
| 53 | + |
| 54 | +def main(): |
| 55 | + url = DOWNLOAD_URL |
| 56 | + name = [] |
| 57 | + star_con=[] |
| 58 | + score = [] |
| 59 | + info = [] |
| 60 | + while url: |
| 61 | + doc = download_page(url) |
| 62 | + movie,star,level_num,info_list,url = get_li(doc) |
| 63 | + name = name + movie |
| 64 | + star_con = star_con + star |
| 65 | + score = score+level_num |
| 66 | + info = info+ info_list |
| 67 | + for (i,m,o,p) in zip(name,star_con,score,info): |
| 68 | + col_A = 'A%s'%(name.index(i)+1) |
| 69 | + col_B = 'B%s'%(name.index(i)+1) |
| 70 | + col_C = 'C%s'%(name.index(i)+1) |
| 71 | + col_D = 'D%s'%(name.index(i)+1) |
| 72 | + ws1[col_A]=i |
| 73 | + ws1[col_B] = m |
| 74 | + ws1[col_C] = o |
| 75 | + ws1[col_D] = p |
| 76 | + wb.save(filename=dest_filename) |
| 77 | + |
| 78 | +if __name__ == '__main__': |
| 79 | + main() |