利用selenium 自動搶魚丸

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

Selenium是什么

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

Talk is cheap, Show you my code

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

自動搶魚丸流程

具體步驟如下：

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

# 獲取全頻道播放的火箭信息
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

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

 

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

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:
        # 找到彈幕聊天窗口，輸入聊天內容
        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:
            # 點擊搶魚丸按鈕，在完成之后，按鈕消息，繼續點擊出現異常，自動關閉瀏覽器即可
            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()

老板，來份魚丸

按照上述方法完成編碼之后，我用自己的帳號測試了一下，效果是這樣的：

 

發現斗魚可能是做了限制，每個人大概一天只能搶到2000個左右的魚丸(斗魚：本來為代碼寫測試的，誰讓你沒事干搶我們魚丸的？！，我：生無可戀.jpg)，娛樂而已，不過說句心里話，selenium可真的是個好東西。

原文轉自：http://www.jianshu.com/p/3a085e0f53c9