使用python selenium webdriver模拟浏览器

　　selenium是进行web自动化测试的一个工具，支持C，C++，Python，Java等语言，他能够实现模拟手工操作浏览器，进行自动化，通过webdriver驱动浏览器操作，我使用的是chrome浏览器，下载chrome  webdriver 放到python的安装目录。

参考连接：

https://pypi.python.org/pypi/selenium

http://selenium-python.readthedocs.io/api.html

http://www.cnblogs.com/fnng/p/3160606.html

from selenium import webdriver
import time
import string
import datetime


def usage():
    print("*********************************************************************")
    print("欢迎使用Amazone差评神器,Enover保留版权，作者：Anker 日期：2016-12-18")
    print("*********************************************************************")

def genSearchDate():
    now = datetime.datetime.now()
    print("当前的日期是：%s/%s/%s" % (now.day, now.month, now.year%2000))

    #计算当前月的的日期范围  
    dayarr = []
    if now.day <= 10 :
        dayarr = [10,1]
    elif now.day/10 <= 2:
        dayarr = [now.day,10,1]
    else:
        dayarr = [now.day,20,10,1]

    #判断是否闰年
    day2 = 0
    if (now.year%4 == 0 and now.year%100 != 0) or now.year%400 == 0:
        day2 = 29
    else:
        day2 = 28
          
    
    months=[[0,0],[31,20,10,1,],[day2,20,10,1],[31,20,10,1],[30,20,10,1],[31,20,10,1],[30,20,10,1],[31,20,10,1],[31,20,10,1],[30,20,10,1],[31,20,10,1],[30,20,10,1],[31,20,10,1]]

    mon=now.month
    searchDate=[]
    while (mon > 0):
        if (mon == now.month):
            tmp = dayarr
        else:
            tmp = months[mon]
        for d in range(0,len(tmp)-1):
            if d==0:
                enddate='%s/%s/%s' % (mon, tmp[d], now.year%2000)
            else:
                enddate='%s/%s/%s' % (mon, tmp[d]-1, now.year%2000)
            begdate='%s/%s/%s' % (mon, tmp[d+1], now.year%2000)
            val=[begdate,enddate]
            searchDate.append(val)
        mon=mon-1
    #print(searchDate)
    return searchDate


#登陆亚马逊
def loginAmazone(driver):
    driver.get("https://sellercentral.amazon.com")
    driver.find_element_by_id('ap_email').send_keys('xxxxx')
    driver.find_element_by_id('ap_password').send_keys('xxxxx')
    driver.find_element_by_name('signIn').submit()


#设置查询条件 ASIN 和 时间
def searchProcess(driver, asin, begdate,enddate):
    driver.get("https://sellercentral.amazon.com/gp/orders-v2/search/ref=ag_myosearch_apsearch_myo")
    driver.find_element_by_id('_myoSO_searchTypeSelect').send_keys('ASIN')
    driver.find_element_by_id('_myoSO_searchKeyword').send_keys(asin)

    driver.find_element_by_id('_myoSO_SearchOption_exactDates').click()
    driver.find_element_by_id('exactDateBegin').clear()
    driver.find_element_by_id('exactDateBegin').send_keys(begdate)
    driver.find_element_by_id('exactDateEnd').clear()
    driver.find_element_by_id('exactDateEnd').send_keys(enddate)

    driver.find_element_by_id('_myoSO_SearchButton').click()
    time.sleep(2)


#设置每页显示50个
def setpage50(driver):
    driver.find_element_by_xpath('//option [@value="50"]').click()  # click
    driver.find_element_by_xpath('//form [@onsubmit="return MYO.LO.DoAjaxSearchCall( this );"]').submit()
    time.sleep(2)
    driver.find_element_by_id('_myoLO_saveDefaultSearchCheckBox').click()

#计算记录个数
def countPage(source):
    pattern='</strong> of <strong>'
    pos1=source.find(pattern)
    beg=pos1+len(pattern)
    pos2=source.find('</strong>',pos1+len(pattern))
    total=int(source[beg:pos2])

    page=total%50
    if page==0:
        page=total/50
    else:
        page=int(total/50)+1
    print("订单总数为：%s,共计%s页" % (total, page))
    return page

#翻页 jump to page
def jumppage(driver, page, custid):
    rc=False
    for index in range(1,page):
        print("正在查找第%s页" % index)
        elements = driver.find_elements_by_xpath('//input [@maxlength="7"]')
        elements[1].find_element_by_xpath('//input [@name="currentPage"]').send_keys(str(index))
        driver.find_element_by_id('_myoSO_GoToPageForm_1').submit()
        time.sleep(4)
        source=driver.page_source
        pos=source.find(custid)
        if pos != -1:
            print('终于找到了，查找记录如下：')
            print(source[pos-270:pos+24])
            rc=True
            break
    return rc

def searchBadReview(driver, asin, custid, searchDate):
    for i in range(0, len(searchDate)):
        tmpDate=searchDate[i]
        begdate=tmpDate[0]
        enddate=tmpDate[1]
        print('==============================================')
        print("开始找%s到%s的订单" %(begdate, enddate))
        searchProcess(driver, asin, begdate, enddate)
        setpage50(driver)
        source=driver.page_source
        page=countPage(source)
        rc = jumppage(driver, page, custid)
        if  rc == True:
            break
    

#主函数
def main():
    usage()

    #输入参数
    asin = input("请输入ASIN：")
    print("你输入的ASIN是: ", asin)
    custid = input("请输入Customer profile id：")
    print("你输入的内容是: ", custid)
    searchDate=genSearchDate()
    #print("查找时间范围如下：")
    #print(searchDate)

    #默认浏览器行为
    print('==============================================')
    print("开始打开浏览器，并登陆Amazone seller center")
    driver = webdriver.Chrome()
    loginAmazone(driver)
    time.sleep(1)
    searchBadReview(driver, asin, custid, searchDate)
    driver.quit()
    time.sleep(60)

if __name__ == "__main__":
    main()