# 使用selenium+phantomJS模拟浏览器爬取淘宝商品信息
# 思路：
# 第一步：利用selenium驱动浏览器，搜索商品信息，得到商品列表
# 第二步：分析商品页数，驱动浏览器翻页，并得到商品信息
# 第三步：爬取商品信息
# 第四步：存储到mongodb

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait from selenium.webdriver.support import expected_conditions as EC from selenium.common.exceptions import TimeoutException from config import * import re

browser = webdriver.PhantomJS(executable_path='/usr/bin/phantomjs',service_args=SERVICE_ARGS)

# 表示给browser浏览器一个10秒的加载时间
wait = WebDriverWait(browser,10)# 使用webdriver打开chrome，打开淘宝页面，搜索美食关键字，返回总页数
def search():print('正在搜索……')try:# 打开淘宝首页browser.get('http://www.taobao.com')# 判断输入框是否已经加载input = wait.until(EC.presence_of_element_located((By.ID,'q')))# < selenium.webdriver.remote.webelement.WebElement(session="d575fc60-91a9-11e8-917b-3dd730d5073d",element=":wdc:1532701944023") ># 判断搜索按钮是否可以进行点击操作submit = wait.until(EC.element_to_be_clickable((By.CSS_SELECTOR,'#J_TSearchForm > div.search-button > button')))# 输入美食input.send_keys(KEYWORD)# 点击搜索按钮submit.click()# 使用css_selector找到显示总页面的元素total = wait.until(EC.presence_of_element_located((By.CSS_SELECTOR,'#mainsrp-pager > div > div > div > div.total')))return total.textexcept TimeoutException:print('超时')return search()def main():total = search()print(total)total = int(re.compile('(\d+)').search(total).group(1))print(total)if __name__ == '__main__':main()

# 引入配置对象DesiredCapabilities
from selenium.webdriver.common.desired_capabilities import DesiredCapabilitiesdcap = dict(DesiredCapabilities.PHANTOMJS)
# 从USER_AGENTS列表中随机选一个浏览器头，伪装浏览器
dcap["phantomjs.page.settings.userAgent"] = USER_AGENTS
# 不载入图片，爬页面速度会快很多
dcap["phantomjs.page.settings.loadImages"] = False
# 设置代理
service_args = ['--disk-cache=true','--load-images=false']
# 打开带配置信息的phantomJS浏览器
browser = webdriver.PhantomJS(executable_path='/usr/bin/phantomjs', desired_capabilities=dcap, service_args=service_args)
# 隐式等待5秒，可以自己调节
browser.implicitly_wait(5)
# 设置10秒页面超时返回，类似于requests.get()的timeout选项，driver.get()没有timeout选项
# 以前遇到过driver.get(url)一直不返回，但也不报错的问题，这时程序会卡住，设置超时选项能解决这个问题。
browser.set_page_load_timeout(10)
# 设置10秒脚本超时时间
browser.set_script_timeout(10)

# 使用selenium+phantomJS模拟浏览器爬取淘宝商品信息
# 思路：
# 第一步：利用selenium驱动浏览器，搜索商品信息，得到商品列表
# 第二步：分析商品页数，驱动浏览器翻页，并得到商品信息
# 第三步：爬取商品信息
# 第四步：存储到mongodb
from selenium import webdriver
from config import *
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.common.exceptions import TimeoutException
import re
from pyquery import PyQuery as pq
import pymongo# client = pymongo.MongoClient(MONGO_URL)
client = pymongo.MongoClient(host='192.168.33.12', port=27017)
db = client[MONGO_DB_SELENIUM]browser = webdriver.PhantomJS(executable_path=EXECUTABLE_PATH,service_args=SERVICE_ARGS)
browser.set_window_size(1400, 1000)# 表示给browser浏览器一个10秒的加载时间
wait = WebDriverWait(browser,10)# 使用webdriver打开chrome，打开淘宝页面，搜索美食关键字，返回总页数
def search():print('正在搜索……')try:# 打开淘宝首页browser.get('http://www.taobao.com')# 判断输入框是否已经加载input = wait.until(EC.presence_of_element_located((By.CSS_SELECTOR,'#q')))# 判断搜索按钮是否可以进行点击操作submit = wait.until(EC.element_to_be_clickable((By.CSS_SELECTOR,'#J_TSearchForm > div.search-button > button')))# 输入美食input.send_keys(KEYWORD_SELENIUM)# 点击搜索按钮submit.click()# 使用css_selector找到显示总页面的元素total = wait.until(EC.presence_of_element_located((By.CSS_SELECTOR,'#mainsrp-pager > div > div > div > div.total')))# 获取商品信息get_products()return total.textexcept TimeoutException:print('超时')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: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    # 使用PyQuery来解析htmldoc = pq(html)  items = doc("#mainsrp-itemlist .items .item").items()for item in items:product = {# 去掉价格中的换行符"price": item.find(".price").text().replace("\n", ""),"image": item.find(".pic .img").attr("src"),"name": item.find(".title").text(),"location": item.find(".location").text(),"shop": item.find(".shop").text(),}"""
            需要保存的商品信息：（1）商品的图片（2）商品的价格（3）商品的名字（4）商品来源（5）商品店铺"""
        # print(product)# {'price': '¥32.80',#  'image': '//g-search1.alicdn.com/img/bao/uploaded/i4/imgextra/i4/95948676/TB2IdzpcnnI8KJjSszbXXb4KFXa_!!0-saturn_solar.jpg_230x230.jpg',#  'name': '南萃坊流心蛋黄饼20个800克传统糕点办公室网红零食小吃美食整箱', 'location': '浙江 杭州', 'shop': '南萃坊旗舰店'}save_products(product)# print("=" * 30)# == == == == == == == == == == == == == == ==# 将商品的信息存储到MongoDB数据库、txt文件中
def save_products(result):try:# 尝试将结果集插入到数据库中if db[MONGO_TABLE_SELENIUM].insert(result):print("存储到MongoDB数据库成功！", result)# 存储到MongoDB数据库成功！ {'price': '¥33.00',#                   'image': '//g-search1.alicdn.com/img/bao/uploaded/i4/i2/110202222/TB2k.9bhcj_B1NjSZFHXXaDWpXa_!!110202222.jpg_230x230.jpg',#                   'name': '陕西特产红星软香酥小吃美食零食礼包早餐糕点豆沙西安网红千层饼', 'location': '陕西 咸阳', 'shop': '红星软香酥专卖',#                   '_id': ObjectId('5b5dd570e138231772e2ef5d')}# == == == == == == == == == == == == == == ==# {'price': '¥24.90',#  'image': '//g-search1.alicdn.com/img/bao/uploaded/i4/imgextra/i1/13621870/TB2V3LhX56guuRjy1XdXXaAwpXa_!!0-saturn_solar.jpg_230x230.jpg',#  'name': '卜珂椰丝球椰蓉球美食早餐糕点心好吃的点心休闲零食品批发店小吃', 'location': '江苏 苏州', 'shop': '卜珂巧克力旗舰店'}# 存储到MongoDB数据库成功！ {'price': '¥24.90',#                   'image': '//g-search1.alicdn.com/img/bao/uploaded/i4/imgextra/i1/13621870/TB2V3LhX56guuRjy1XdXXaAwpXa_!!0-saturn_solar.jpg_230x230.jpg',#                   'name': '卜珂椰丝球椰蓉球美食早餐糕点心好吃的点心休闲零食品批发店小吃', 'location': '江苏 苏州', 'shop': '卜珂巧克力旗舰店',#                   '_id': ObjectId('5b5dd575e138231772e2ef5e')}# == == == == == == == == == == == == == == ==# 将结果集存储到txt文件中if result:with open("products.txt", "a", encoding="utf-8") as f:f.write(str(result) + "\n")f.close()except Exception:print("存储失败！", result)def main():try:total = search()# print(total)total = int(re.compile('(\d+)').search(total).group(1))# print(total)for i in range(2, total + 1):next_page(i)except Exception:print("出错啦！")finally:browser.close()  # 最后一定都要关闭浏览器if __name__ == '__main__':main()