TA的每日心情 | 无聊 2024-9-19 09:07 |
---|
签到天数: 11 天 连续签到: 2 天 [LV.3]测试连长
|
原本打算使用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上托管,感兴趣的同学可以前去浏览。整个抢鱼丸礼物的过程可以由这个图来说明:
自动抢鱼丸流程
具体步骤如下:
与斗鱼弹幕服务器建立连接,监听当前最热门房间弹幕,由于火箭礼物消息会通过广播向所有直播间发送,所以只需要根据火箭消息的标识符来判定火箭消息即可;
火箭消息接收之后可以将其转存到redis中,利用pub/sub完成火箭消息的广播和订阅;
订阅redis火箭消息频道,获取当前可以抢鱼丸的房间完整url;
利用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可真的是个好东西。
|
|