利用selenium 自動搶魚丸

imelee 2017-06-29

展開全文

原本打算使用selenium給這個小項目做自動化測試的，沒成想selenium功能如此強(qiáng)大，那就順勢做個自動搶魚丸的小工具吧。

Selenium是什么

Selenium是一款由ThoughtWorks公司開發(fā)的web自動化測試工具，它可以讓測試人員編寫的測試代碼直接在瀏覽器中進(jìn)行測試，并且它提供了當(dāng)前絕大多數(shù)現(xiàn)代瀏覽器的驅(qū)動。同時Selenium也擁有很多開發(fā)語言版本的驅(qū)動。我使用的就是python版本的驅(qū)動。具體安裝步驟可以參考這里。
具體來說，Selenium可以模擬大多數(shù)的頁面操作：填寫表單、拖拽頁面、頁面切換、返回上一頁面、前進(jìn)到下一頁面等。
告訴了Selenium網(wǎng)址之后，它可以自動打開指定的瀏覽器，并且待頁面完全加載之后分別通過頁面元素的id、name、XPath、Link Text、tag、class或者是css選擇器完成元素的選擇。
選定了頁面元素之后可以按照預(yù)期完成一些列動作。

Talk is cheap, Show you my code

整個項目的代碼在github上托管，感興趣的同學(xué)可以前去瀏覽。整個搶魚丸禮物的過程可以由這個圖來說明：

自動搶魚丸流程

具體步驟如下：

與斗魚彈幕服務(wù)器建立連接，監(jiān)聽當(dāng)前最熱門房間彈幕，由于火箭禮物消息會通過廣播向所有直播間發(fā)送，所以只需要根據(jù)火箭消息的標(biāo)識符來判定火箭消息即可；
火箭消息接收之后可以將其轉(zhuǎn)存到redis中，利用pub/sub完成火箭消息的廣播和訂閱；
訂閱redis火箭消息頻道，獲取當(dāng)前可以搶魚丸的房間完整url；
利用selenium打開瀏覽器并訪問指定直播間，自動發(fā)送彈幕消息，并且點擊搶魚丸的按鈕，最終完成自動搶魚丸禮物的整個流程。

# 獲取全頻道播放的火箭信息
def get_rocket(data):
    try:
        sender_id = re.search('\\\\/sn@=(.+?)\\\\/', data).group(1)
        recver_id = re.search('\\\\/dn@=(.+?)\\\\/', data).group(1)
        recver_room = re.search('\\\\/drid@=(.+?)\\\\/', data).group(1)
        gift = re.search('\\\\/gn@=(.+?)\\\\/', data).group(1)
        rocketmsg = {}
        rocketmsg["sender_id"] = sender_id
        rocketmsg["recver_id"] = recver_id
        rocketmsg["recver_room"] = recver_room
        rocketmsg["gift"] = gift
        rocketmsg["date"] = datetime.now()
        col.insert_one(rocketmsg, bypass_document_validation=False)
        publishvalue = {}
        publishvalue["sender_id"] = sender_id
        publishvalue["recver_id"] = recver_id
        publishvalue["recver_room"] = recver_room
        publishvalue["gift"] = gift
        castRocket(publishvalue)
        print sender_id, "送給房間號為:", recver_room, "的", recver_id, "一個",\\\            gift, "<", datetime.now(), ">"
        # 在此處完成自動搶魚丸流程
        getGift(str(recver_room))
    except Exception, e:
        print "error occur:", repr(data)
    finally:
        pass

在此過程中，起到關(guān)鍵作用的selenium主要扮演如下圖所示的角色：

獲取瀏覽器cookie，這樣就免去了破解斗魚賬號自動登錄的麻煩；
獲取指定元素，比如找到彈幕發(fā)送窗口，輸入彈幕信息，發(fā)送彈幕，點擊搶魚丸按鈕；

from selenium import webdriver
from selenium.webdriver.common.action_chains import ActionChains
import time


def getGift(roomid):
    # 這里是指定firefox的cookie位置
    fp = webdriver.FirefoxProfile(
        r'/Users/eclipse/Library/Application Support/Firefox/Profiles/tmsbsjpg.default')
    browser = webdriver.Firefox(fp)
    browser.implicitly_wait(15)  # seconds
    browser.get("http://www.douyu.com/" + roomid)
    try:
        # 找到彈幕聊天窗口，輸入聊天內(nèi)容
        indexvideo = browser.find_element_by_class_name('cs-textarea')
        print type(indexvideo)
        indexvideo.send_keys('2333333333333')
        print indexvideo
        time.sleep(7)
        sendbut = browser.find_element_by_class_name('b-btn')
        ActionChains(browser).move_to_element(
            indexvideo).click(sendbut).perform()
        gift = browser.find_element_by_class_name('peck-cdn')
    except Exception, e:
        print str(e)
        browser.quit()

    times = 0
    while True:
        try:
            # 點擊搶魚丸按鈕，在完成之后，按鈕消息，繼續(xù)點擊出現(xiàn)異常，自動關(guān)閉瀏覽器即可
            ActionChains(browser).move_to_element(gift).click(gift).perform()
            time.sleep(2)
            print times
            times += 1
        except Exception, e:
            print 'completed by an error'
            browser.quit()