Python selenium自动化网页抓取器

（开开心心每一天~ ---虫瘾师）

直接入正题---Python selenium自动控制浏览器对网页的数据进行抓取，其中包含按钮点击、跳转页面、搜索框的输入、页面的价值数据存储、mongodb自动id标识等等等。

1、首先介绍一下 Python selenium ---自动化测试工具，用来控制浏览器来对网页的操作，在爬虫中与BeautifulSoup结合那就是天衣无缝，除去国外的一些变态的验证网页，对于图片验证码我有自己写的破解图片验证码的源代码，成功率在85%。

     详情请咨询QQ群--607021567（这不算广告，群里有好多Python的资源分享，还有大数据的一些知识【hadoop】）

2、beautifulsoup就不需要详细的介绍了，直接上网址-https://www.crummy.com/software/BeautifulSoup/bs4/doc/（BeautifulSoup的官方文档）

3、关于mongodb的自动id的生成。mongodb中所有的存储数据都是有固定的id的，但是mongodb的id对于人类来讲是复杂的，对于机器来讲是小菜一碟的，所以在存入数据的同时，我习惯用新id来对每一条数据的负责！

在Python中使用mongodb的话需要引进模块 from pymongo import MongoClient,ASCENDING, DESCENDING ，该模块就是你的责任！

 

接下来开始讲程序，直接上实例（一步一步来）：

引入模块：

from selenium import webdriver
from bs4 import BeautifulSoup
import requests
from pymongo import MongoClient,ASCENDING, DESCENDING
import time
import re

其中的每一个模块都会说已经解释过了，其中的re、requests都是之前就有提过的，他们都是核心缺一不可！

首先，我举一个小例子，淘宝的自动模拟搜索功能（源码）：

先说一下selenium 的定位方法

• find_element_by_id
• find_element_by_name
• find_element_by_xpath
• find_element_by_link_text
• find_element_by_partial_link_text
• find_element_by_tag_name
• find_element_by_class_name
• find_element_by_css_selector

源码：

from selenium import webdriver
from bs4 import BeautifulSoup
import requests
from pymongo import MongoClient,ASCENDING, DESCENDING
import time
import re

def TaoBao():
    try:
        Taobaourl = 'https://www.taobao.com/'
        driver = webdriver.Chrome()
        driver.get(Taobaourl)
        time.sleep(5)#通常这里需要停顿，不然你的程序很有可能被检测到是Spider
        text='Strong Man'#输入的内容
        driver.find_element_by_xpath('//input[@class="search-combobox-input"]').send_keys(text).click()
        driver.find_element_by_xpath('//button[@class="btn-search tb-bg"]').click()
        driver.quit()
        
    except Exception,e:
        print e

if __name__ == '__main__':
    TaoBao()

效果的实现，你们可以直接复制后直接运行！我只用了xpath的这个方法，因为它最实在！橙色字体（如果我没有色盲的话），就是网页中定位的元素，可以找到的！

接下来就是与BeautifulSoup的结合了，但是我们看到的只是打开了网页，并没有源码，那么就需要 “变量名.page_source”这个方法，他会实现你的梦想，你懂得?

ht =  driver.page_source
#print ht 你可以Print出啦看看
soup = BeautifulSoup(ht,'html.parser')

下面就是BeautifulSoup的一些语法操作了，对于数据的结构还有采集，在上一篇里面有详细的抓取操作！！！

算了！说一个最简单的定位抓取：

soup = BeautifulSoup(ht,'html.parser')
a = soup.find('table',id="ctl00_ContentMain_SearchResultsGrid_grid")
if a:  #必须加判断，不然访问的网页或许没有这一元素，程序就会都停止！

class的标签必须是class_,一定要记住！

哈哈哈！mongodb了昂，细节细节，首先需要用到模块----from pymongo import MongoClient,ASCENDING, DESCENDING

因为在python，mongodb的语法仍然实用，所以需要定义一个库，并且是全局性的，还有链接你计算机的一个全局变量。

 

if __name__ == '__main__':  

  global db#全局变量                   
  global table#全局数据库
  table = 'mouser_product'
  mconn=MongoClient("mongodb://localhost")#地址
  db=mconn.test
  db.authenticate('test','test')#用户名和密码
  Taobao()

 

定义这些后，需要我们的新id来对数据的跟踪加定义：

db.sn.find_and_modify({"_id": table}, update={ "$inc": {'currentIdValue': 1}},upsert=True)
dic = db.ids.find({"_id":table}).limit(1)
return dic[0].get("currentIdValue") 

这个方法是通用的，所以只要记住其中的mongodb的语法就可以了！因为这里是有返回值的，所以这个是个方法体，这里不需要太过于纠结是怎么实现的，理解就好，中心还是在存数据的过程中

count = db[table].find({'数据':数据}).count() #是检索数据库中的数据
if count <= 0:                               #判断是否有
    ids= getNewsn()                          #ids就是我们新定义的id，这里的id是1开始的增长型id
    db[table].insert({"ids":ids,"数据":数据})            

这样我们的数据就直接存入到mongodb的数据库中了，这里解释一下为什么在大数据中这么喜欢mongodb，因为它小巧，速度佳！

最后来一个实例源码：

from selenium import webdriver
from bs4 import BeautifulSoup
import requests
from pymongo import MongoClient,ASCENDING, DESCENDING
import time
import re
def parser():
    try:
        f = open('sitemap.txt','r')
        for i in  f.readlines():
            sorturl=i.strip()
            driver = webdriver.Firefox()
            driver.get(sorturl)
            time.sleep(50)
            ht =  driver.page_source
            #pageurl(ht)
            soup = BeautifulSoup(ht,'html.parser')
            a = soup.find('a',class_="first-last")
            if a:
                pagenum = int(a.get_text().strip())
                print pagenum
                for i in xrange(1,pagenum):
                    element = driver.find_element_by_xpath('//a[@id="ctl00_ContentMain_PagerTop_%s"]' %i)
                    element.click()
                    html =  element.page_source
                    pageurl(html)
                    time.sleep(50)
                    driver.quit()
    except Exception,e:
        print e
def pageurl(ht):
    try:
        soup = BeautifulSoup(ht,'html.parser')
        a = soup.find('table',id="ctl00_ContentMain_SearchResultsGrid_grid")
        if a:
            tr = a.find_all('tr',class_="SearchResultsRowOdd")
            if tr:
                    for i in tr:
                        td = i.find_all('td')
                        if td:
                            url = td[2].find('a')
                            if url:
                                producturl = '网址'+url['href']
                                print producturl
                                count = db[table].find({"url":producturl}).count()
                                if count<=0:
                                    sn = getNewsn()
                                    db[table].insert({"sn":sn,"url":producturl})
                                    print str(sn) + ' inserted successfully'
                                    time.sleep(3)
                                else:
                                    print 'exists url'
            tr1 = a.find_all('tr',class_="SearchResultsRowEven")
            if tr1:
                    for i in tr1:
                        td = i.find_all('td')
                        if td:
                            url = td[2].find('a')
                            if url:
                                producturl = '网址'+url['href']
                                print producturl
                                count = db[table].find({"url":producturl}).count()
                                if count<=0:
                                    sn = getNewsn()
                                    db[table].insert({"sn":sn,"url":producturl})
                                    print str(sn) + ' inserted successfully'
                                    time.sleep(3)
                                else:
                                    print 'exists url'
                                #time.sleep(5)

    except Exception,e:
        print e
def getNewsn(): 
    db.sn.find_and_modify({"_id": table}, update={ "$inc"{'currentIdValue': 1}},upsert=True)
    dic = db.sn.find({"_id":table}).limit(1)
    return dic[0].get("currentIdValue")

if __name__ == '__main__':  

  global db                    
  global table
  table = 'mous_product'
  mconn=MongoClient("mongodb://localhost")
  db=mconn.test
  db.authenticate('test','test')
  parser()

这一串代码是破解一个老外的无聊验证码界面结缘的，我真的对他很无语了！破解方法还是实践中！这是完整的源码，无删改的哦！纯手工！