-
Notifications
You must be signed in to change notification settings - Fork 16
/
html_parser.py
29 lines (26 loc) · 1.1 KB
/
html_parser.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
from bs4 import BeautifulSoup
import urlparse,re
class HtmlParser(object):
def parse(self, page_url, content):
soup = BeautifulSoup(content, 'html.parser', from_encoding='utf-8')
urls = self._get_new_urls(page_url, soup)
data = self._get_new_data(page_url, soup)
return urls, data
def _get_new_urls(self, page_url, soup):
new_urls = []
links = soup.find_all('a', href=re.compile("view/\d+\.(html|htm)"))
for link in links:
url = link['href']
url = urlparse.urljoin(page_url, url)
new_urls.append(url)
return new_urls
def _get_new_data(self, page_url, soup):
#<dd class="lemmaWgt-lemmaTitle-title"><h1>Python</h1>
data = {}
title_node = soup.find('dd', class_="lemmaWgt-lemmaTitle-title").find('h1')
data['title'] = title_node.get_text()
#<div class="lemma-summary" label-module="lemmaSummary">
summary_node = soup.find('div', class_="lemma-summary")
data['summary'] = summary_node.get_text()
data['url'] = page_url
return data