使用Selenium驱动chrome页面,获得淘宝信息并用BeautifulSoup分析得到结果。
使用Selenium时注意页面的加载判断,以及加载超时的异常处理。
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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
import json import re from bs4 import BeautifulSoup from selenium import webdriver from selenium.common.exceptions import TimeoutException from selenium.webdriver.common.by import By from selenium.webdriver.support.ui import WebDriverWait from selenium.webdriver.support import expected_conditions as EC browser = webdriver.Chrome() # 浏览器需要多次使用,所以单独拿出来。设置一个最长的等待时间,等待目标加载完成 wait = WebDriverWait(browser, 10 ) def search(keyword): # wait容易出现加载时间长的问题,因此用try来捕捉异常 try : browser.get( 'https://www.taobao.com' ) # 加载需要一定时间的,设置了等待时间,等待加载 # 输入按钮的加载等待 input = wait.until( # 设置加载目标,它是一个选择器,参数是需要选择方式和等待加载的内容 EC.presence_of_element_located((By.CSS_SELECTOR, "#q" )) # 选择CSS选择器和选择内容 ) # 提交按钮 submit = wait.until( # EC后面是选择条件,按钮的加载条件最好的是element_to_be_clickable,意思为元素可以点击的 EC.element_to_be_clickable((By.CSS_SELECTOR, "#J_TSearchForm > div.search-button > button" )) ) input .send_keys(keyword) # send_keys对输入框输入内容 submit.click() # 提交搜索内容,进入下一个页面 # 等待页码元素加载完成,并返回最大页码数 total = wait.until( EC.presence_of_element_located((By.CSS_SELECTOR, "#mainsrp-pager > div > div > div > div.total" )) ) # 等待加载完成后获取信息 get_products() return total.text except TimeoutException: # 超时后重新请求,因此递归调用 return search() def next_page(page_number): try : # 页码输入框和翻页按钮 input = wait.until( EC.presence_of_element_located((By.CSS_SELECTOR, "#mainsrp-pager > div > div > div > div.form > input" )) ) # 提交按钮 submit = wait.until( EC.element_to_be_clickable( (By.CSS_SELECTOR, "#mainsrp-pager > div > div > div > div.form > span.btn.J_Submit" )) ) input .clear() input .send_keys(page_number) submit.click() # 判断翻页成功 wait.until( EC.text_to_be_present_in_element((By.CSS_SELECTOR, '#mainsrp-pager > div > div > div > ul > li.item.active > span' ), str (page_number))) get_products() except TimeoutException: return next_page(page_number) def get_products(): # 判断单个页面是否被加载出来 wait.until(EC.presence_of_element_located((By.CSS_SELECTOR, '#mainsrp-itemlist .items .item' ))) html = browser.page_source # 获取页面源代码,所有的 # 使用BS进行分析 soup = BeautifulSoup(html, 'lxml' ) items = soup.select( '#mainsrp-itemlist .items .item' ) for item in items: image = item.select( '.pic .img' )[ 0 ][ 'data-src' ] price = item.select( '.price strong' )[ 0 ].text deal = item.select( '.deal-cnt' )[ 0 ].text[: - 3 ] title = item.select( '.title' )[ 0 ].text.strip() shop = item.select( '.shop' )[ 0 ].text.strip() location = item.select( '.location' )[ 0 ].text product = { 'image' : image, 'price' : price, 'deal' : deal, 'title' : title, 'shop' : shop, 'location' : location } save_text(product) # 下载内容 def save_text(product): # 保存为txt文件,a追加写模式,编码模式utf-8 with open ( 'text.txt' , 'a' , encoding = 'utf-8' ) as f: # 使用JSON把字典转换为str格式,加换行符 f.write(json.dumps(product, ensure_ascii = False ) + '\n' ) f.close() def main(): # 通过关键字在淘宝进行搜索 total = search( '美食' ) # 用正则提取页码数字 total = int (re. compile ( '(\d+)' ).search(total).group( 1 )) # 翻页 for i in range ( 2 , total + 1 ): # 循环包含前,不包含尾 next_page(i) browser.close() if __name__ = = '__main__' : main() |
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持服务器之家。
原文链接:http://blog.csdn.net/wenboyu/article/details/78176859