爬取淘宝美食信息

用selenium+pyquery爬取淘宝美食的搜索信息，并用mongodb存储！

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
from pyquery import PyQuery as pq
from config import *
import pymongo
client = pymongo.MongoClient(MONGO_URL)
db = client[MONGO_DB]

import re
browser = webdriver.Chrome()
wait =WebDriverWait(browser,10)
def search():
    try:
        browser.get('https://www.taobao.com')
        input =  wait.until(
            EC.presence_of_element_located((By.ID, 'q'))
        )
        input.send_keys('美食')
        submit = wait.until(EC.presence_of_element_located((By.CSS_SELECTOR,'#J_TSearchForm > div.search-button > button')))
        submit.click()
        result = wait.until(EC.presence_of_element_located((By.CSS_SELECTOR,'#mainsrp-pager > div > div > div > div.total')))
        number = re.search('(\d+)',result.text,re.S);
        get_products()
        return int(number.group(1))
    except TimeoutException:
        print('------调用游览器超时------')
        return None

def next_page(page_number):
    print(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.presence_of_element_located((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 TimeoutError:
        print("------翻页失败------")
        next_page(page_number)
def get_products():
    try:
        wait.until(EC.presence_of_element_located((By.CSS_SELECTOR,'#mainsrp-itemlist > div > div')))
        source = browser.page_source
        result = pq(source)
        items = result('#mainsrp-itemlist .items .item').items()
        for item in items:
            product={
                'image':item.find('.pic .img:first').attr('src'),
                'name':item.find('.title').text(),
                'price':item.find('.price').text(),
                'people':item.find('.deal-cnt').text()[:-3],
                'location':item.find('.location').text()
            }
            save_to_mongo(product)
    except TimeoutError:
        print('------获取商品超时------')
def save_to_mongo(result):
    try:
        if db[MONGO_TB].insert(result):
            print('存储到MONGODB成功',result)
    except Exception:
        print('存储到MONGODB失败',result)
def main():
    number = search()

    # for i in range(number-2):
    #     next_page(i+2)

if __name__ == '__main__':
    main()