Files
news-search-engine/code/spider.py
T
2016-01-07 12:31:13 +08:00

76 lines
2.5 KiB
Python

# -*- coding: utf-8 -*-
"""
Created on Sat Dec 19 11:57:01 2015
@author: bitjoy.net
"""
from bs4 import BeautifulSoup
import urllib.request
import xml.etree.ElementTree as ET
import configparser
def get_news_pool(root, start, end):
news_pool = []
for i in range(start,end,-1):
page_url = ''
if i != start:
page_url = root +'_%d.shtml'%(i)
else:
page_url = root + '.shtml'
try:
response = urllib.request.urlopen(page_url)
except Exception as e:
print("-----%s: %s-----"%(type(e), page_url))
continue
html = response.read()
soup = BeautifulSoup(html)
td = soup.find('td', class_ = "newsblue1")
a = td.find_all('a')
span = td.find_all('span')
for i in range(len(a)):
date_time = span[i].string
url = a[i].get('href')
title = a[i].string
news_info = ['2016-'+date_time[1:3]+'-'+date_time[4:-1]+':00',url,title]
news_pool.append(news_info)
return(news_pool)
def crawl_news(news_pool, min_body_len, doc_dir_path, doc_encoding):
i = 1
for news in news_pool:
try:
response = urllib.request.urlopen(news[1])
except Exception as e:
print("-----%s: %s-----"%(type(e), news[1]))
continue
html = response.read()
soup = BeautifulSoup(html)
try:
body = soup.find('div', class_ = "text clear").find('div').get_text()
except Exception as e:
print("-----%s: %s-----"%(type(e), news[1]))
continue
if '//' in body:
body = body[:body.index('//')]
body = body.replace(" ", "")
if len(body) <= min_body_len:
continue
doc = ET.Element("doc")
ET.SubElement(doc, "id").text = "%d"%(i)
ET.SubElement(doc, "url").text = news[1]
ET.SubElement(doc, "title").text = news[2]
ET.SubElement(doc, "datetime").text = news[0]
ET.SubElement(doc, "body").text = body
tree = ET.ElementTree(doc)
tree.write(doc_dir_path + "%d.xml"%(i), encoding = doc_encoding, xml_declaration = True)
i += 1
if __name__ == '__main__':
config = configparser.ConfigParser()
config.read('../config.ini', 'utf-8')
root = 'http://news.sohu.com/1/0903/61/subject212846158'
news_pool = get_news_pool(root, 854, 849)
crawl_news(news_pool, 140, config['DEFAULT']['doc_dir_path'], config['DEFAULT']['doc_encoding'])
print('done!')