原本打算使用selenium給這個(gè)小項(xiàng)目做自動(dòng)化測(cè)試的，沒成想selenium功能如此強(qiáng)大本鸣，那就順勢(shì)做個(gè)自動(dòng)搶魚丸的小工具吧。
Selenium是什么
Selenium是一款由ThoughtWorks公司開發(fā)的web自動(dòng)化測(cè)試工具，它可以讓測(cè)試人員編寫的測(cè)試代碼直接在瀏覽器中進(jìn)行測(cè)試，并且它提供了當(dāng)前絕大多數(shù)現(xiàn)代瀏覽器的驅(qū)動(dòng)稿茉。同時(shí)Selenium也擁有很多開發(fā)語言版本的驅(qū)動(dòng)。我使用的就是python版本的驅(qū)動(dòng)芥炭。具體安裝步驟可以參考這里漓库。
具體來說，Selenium可以模擬大多數(shù)的頁面操作：填寫表單园蝠、拖拽頁面渺蒿、頁面切換、返回上一頁面彪薛、前進(jìn)到下一頁面等茂装。
告訴了Selenium網(wǎng)址之后，它可以自動(dòng)打開指定的瀏覽器善延，并且待頁面完全加載之后分別通過頁面元素的id少态、name、XPath易遣、Link Text彼妻、tag、class或者是css選擇器完成元素的選擇。
選定了頁面元素之后可以按照預(yù)期完成一些列動(dòng)作侨歉。
Talk is cheap, Show you my code
整個(gè)項(xiàng)目的代碼在github上托管屋摇，感興趣的同學(xué)可以前去瀏覽。整個(gè)搶魚丸禮物的過程可以由這個(gè)圖來說明：
具體步驟如下：
- 與斗魚彈幕服務(wù)器建立連接幽邓，監(jiān)聽當(dāng)前最熱門房間彈幕炮温，由于火箭禮物消息會(huì)通過廣播向所有直播間發(fā)送，所以只需要根據(jù)火箭消息的標(biāo)識(shí)符來判定火箭消息即可牵舵；
- 火箭消息接收之后可以將其轉(zhuǎn)存到redis中柒啤，利用pub/sub完成火箭消息的廣播和訂閱；
- 訂閱redis火箭消息頻道棋枕，獲取當(dāng)前可以搶魚丸的房間完整url白修；
- 利用selenium打開瀏覽器并訪問指定直播間妒峦，自動(dòng)發(fā)送彈幕消息重斑，并且點(diǎn)擊搶魚丸的按鈕，最終完成自動(dòng)搶魚丸禮物的整個(gè)流程肯骇。
# 獲取全頻道播放的火箭信息
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, "送給房間號(hào)為:", recver_room, "的", recver_id, "一個(gè)",\\\\
gift, "<", datetime.now(), ">"
# 在此處完成自動(dòng)搶魚丸流程
getGift(str(recver_room))
except Exception, e:
print "error occur:", repr(data)
finally:
pass
在此過程中窥浪，起到關(guān)鍵作用的selenium主要扮演如下圖所示的角色：
- 獲取瀏覽器cookie，這樣就免去了破解斗魚帳號(hào)自動(dòng)登錄的麻煩笛丙；
- 獲取指定元素漾脂，比如找到彈幕發(fā)送窗口，輸入彈幕信息胚鸯，發(fā)送彈幕骨稿，點(diǎn)擊搶魚丸按鈕；
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:
# 點(diǎn)擊搶魚丸按鈕坦冠，在完成之后，按鈕消息哥桥，繼續(xù)點(diǎn)擊出現(xiàn)異常辙浑，自動(dòng)關(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()
老板，來份魚丸
按照上述方法完成編碼之后拟糕，我用自己的帳號(hào)測(cè)試了一下判呕，效果是這樣的：
發(fā)現(xiàn)斗魚可能是做了限制，每個(gè)人大概一天只能搶到2000個(gè)左右的魚丸(斗魚：本來為代碼寫測(cè)試的送滞，誰讓你沒事干搶我們魚丸的侠草？！犁嗅，我：生無可戀.jpg)边涕，娛樂而已，不過說句心里話愧哟，selenium可真的是個(gè)好東西奥吩。
