51Testing软件测试论坛

标题: 利用selenium 自动抢鱼丸 [打印本页]
作者: 八戒你干嘛    时间: 2017-6-23 10:11
标题: 利用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上托管，感兴趣的同学可以前去浏览。整个抢鱼丸礼物的过程可以由这个图来说明：

                                         自动抢鱼丸流程
具体步骤如下：

与斗鱼弹幕服务器建立连接，监听当前最热门房间弹幕，由于火箭礼物消息会通过广播向所有直播间发送，所以只需要根据火箭消息的标识符来判定火箭消息即可；
火箭消息接收之后可以将其转存到redis中，利用pub/sub完成火箭消息的广播和订阅；
订阅redis火箭消息频道，获取当前可以抢鱼丸的房间完整url；
利用selenium打开浏览器并访问指定直播间，自动发送弹幕消息，并且点击抢鱼丸的按钮，最终完成自动抢鱼丸礼物的整个流程。
  1. def get_rocket(data):
  2.     try:
  3.         sender_id = re.search('\\\\/sn@=(.+?)\\\\/', data).group(1)
  4.         recver_id = re.search('\\\\/dn@=(.+?)\\\\/', data).group(1)
  5.         recver_room = re.search('\\\\/drid@=(.+?)\\\\/', data).group(1)
  6.         gift = re.search('\\\\/gn@=(.+?)\\\\/', data).group(1)
  7.         rocketmsg = {}
  8.         rocketmsg["sender_id"] = sender_id
  9.         rocketmsg["recver_id"] = recver_id
  10.         rocketmsg["recver_room"] = recver_room
  11.         rocketmsg["gift"] = gift
  12.         rocketmsg["date"] = datetime.now()
  13.         col.insert_one(rocketmsg, bypass_document_validation=False)
  14.         publishvalue = {}
  15.         publishvalue["sender_id"] = sender_id
  16.         publishvalue["recver_id"] = recver_id
  17.         publishvalue["recver_room"] = recver_room
  18.         publishvalue["gift"] = gift
  19.         castRocket(publishvalue)
  20.         print sender_id, "送给房间号为:", recver_room, "的", recver_id, "一个",\\\\
  21.             gift, "<", datetime.now(), ">"
  22.         # 在此处完成自动抢鱼丸流程
  23.         getGift(str(recver_room))
  24.     except Exception, e:
  25.         print "error occur:", repr(data)
  26.     finally:
  27.         pass
复制代码
在此过程中，起到关键作用的selenium主要扮演如下图所示的角色：

1.获取浏览器cookie，这样就免去了破解斗鱼帐号自动登录的麻烦；
2.获取指定元素，比如找到弹幕发送窗口，输入弹幕信息，发送弹幕，点击抢鱼丸按钮；
  1. from selenium import webdriver
  2. from selenium.webdriver.common.action_chains import ActionChains
  3. import time


  6. def getGift(roomid):
  7.     # 这里是指定firefox的cookie位置
  8.     fp = webdriver.FirefoxProfile(
  9.         r'/Users/eclipse/Library/Application Support/Firefox/Profiles/tmsbsjpg.default')
  10.     browser = webdriver.Firefox(fp)
  11.     browser.implicitly_wait(15)  # seconds
  12.     browser.get("http://www.douyu.com/" + roomid)
  13.     try:
  14.         # 找到弹幕聊天窗口，输入聊天内容
  15.         indexvideo = browser.find_element_by_class_name('cs-textarea')
  16.         print type(indexvideo)
  17.         indexvideo.send_keys('2333333333333')
  18.         print indexvideo
  19.         time.sleep(7)
  20.         sendbut = browser.find_element_by_class_name('b-btn')
  21.         ActionChains(browser).move_to_element(
  22.             indexvideo).click(sendbut).perform()
  23.         gift = browser.find_element_by_class_name('peck-cdn')
  24.     except Exception, e:
  25.         print str(e)
  26.         browser.quit()

  28.     times = 0
  29.     while True:
  30.         try:
  31.             # 点击抢鱼丸按钮，在完成之后，按钮消息，继续点击出现异常，自动关闭浏览器即可
  32.             ActionChains(browser).move_to_element(gift).click(gift).perform()
  33.             time.sleep(2)
  34.             print times
  35.             times += 1
  36.         except Exception, e:
  37.             print 'completed by an error'
  38.             browser.quit()
复制代码
老板，来份鱼丸

按照上述方法完成编码之后，我用自己的帐号测试了一下，效果是这样的：

发现斗鱼可能是做了限制，每个人大概一天只能抢到2000个左右的鱼丸(斗鱼：本来为代码写测试的，谁让你没事干抢我们鱼丸的？！，我：生无可恋.jpg)，娱乐而已，不过说句心里话，selenium可真的是个好东西。

作者: 清晨一缕阳光    时间: 2017-6-23 10:34
厉害啊！
作者: 悠悠小仙仙    时间: 2017-6-23 13:39
正研究一下Redis 不知仁兄可有快速入门之法？
作者: 八戒你干嘛    时间: 2017-6-23 13:40
悠悠小仙仙 发表于 2017-6-23 13:39
正研究一下Redis 不知仁兄可有快速入门之法？

redis设计与实现 这本书好好看下 官方文档也很不错 然后多动手尝试 这样能够很快上手
作者: 巴黎的灯光下    时间: 2017-6-23 13:41
前段时间...我有一个做淘宝的朋友..找我写一个自动化脚本...也是叫我模拟整个淘宝购买流程...后来我想了一下...其实是没有问题的...但是有一点不好，如果网络条件不好，某些元素找不到...会导致程序报错..无法执行..而且淘宝对这种刷单的行为对IP监控会比较严厉，然后我又在找动态IP的供应商...然后又发现，如果我买了到底怎么去读取那些IP呢？然后我还得想办法部署到服务器上面....感觉即将要成为一个全栈工程师了...



欢迎光临 51Testing软件测试论坛 (http://bbs.51testing.com/) Powered by Discuz! X3.2