使用Selenium模拟登陆新浪微博

发表于:2018-6-14 08:11  作者:温良spider   来源:博客园

字体: | 上一篇 | 下一篇 |我要投稿 | 推荐标签: 软件测试工具 Selenium 自动化测试工具

  一.selenium基本使用
  1.selenium安装及基本操作
  selenium是一个自动化测试工具,它支持各种浏览器,包括Chrome,Safari,Firefox等主流界面浏览器驱动,也包括Phantomjs的无界面浏览器。
  通过selenium+phantomjs可以直接渲染js
  首先,需要下载不同浏览器的驱动,直接百度搜索就可以,例如搜过Chrome浏览器驱动,下载到本地
  然后,将下载好的驱动放入到安装python的根目录下,例如C:/pyhton3/usr/bin,就是要把驱动和python.exe放在同一个文件夹。(虚拟环境也是一样,放入对应的虚拟环境下的python的根目录)
  最后,通过pip安装selenium模块。
# coding = utf-8
import time
from selenium import webdriver  # 导入模块
driver = webdriver.Chrome()  # 1.声明浏览器对象
driver.get('https://www.baidu.com')  # 2. 访问地址
# dirver.get('https://www.weibo.com')
try:
driver.save_screenshot('test.jpg')  # 3.操作。截图
finally:
time.sleep(3)
driver.close()  # 4.关闭浏览器
  这里模拟一个浏览器,访问百度,并将浏览器打开后的页面进行截图。
  2.元素选取
  要用代码操作浏览器,访问网页,做一些操作必须要通过定位,获得要操作元素的位置,例如,要在百度输入框搜索python关键词,要经过如下操作:
  首先,通过代码定位到输入框所在位置;
  然后,通过代码在输入框中输入关键字
  最后,通过代码实现点击搜索。
import time
from selenium import webdriver  # 导入模块
driver = webdriver.Chrome()  # 1.声明浏览器对象
driver.get('https://www.baidu.com')  # 2. 访问地址
try:
# 通过审查元素,发现百度输入框的id为kw,
# 通过find_element_by_id("kw")定位到该输入框
# 通过send_keys("Python") 实现在输入框中输入关键字
driver.find_element_by_id('kw').send_keys('python')  # 输入数据
# 同样的,通过find_element_by_id定位“百度一下”按钮
# 通过click()实现点击
driver.find_element_by_id('su').click()
finally:
time.sleep(30)
driver.close()
当然,还有很多其他方式来定位:
# 通过文本,定位贴吧的位置,并点击
driver.find_element_by_link_text('贴吧').click()  # 通过文本链接点击进入
# 通过xpath路径定位元素
driver.find_element_by_xpath('//*[@id="1"]/h3/a[1]').click()
# 通过元素的name来定位,当然,后可以by_class_name()
driver.find_element_by_name('loginname').send_keys('13212702940')
还有一些常用操作
print(driver.page_source)  # 获取网页经过js渲染的源代码,相当于审查元素
# driver.find_elements_by_id()  # 获取多个元素
# driver.find_element_by_css_selector
  3.页面交互
  1)窗口切换
  使用selenium操作浏览器的时候肯定会有很多窗口,selenium默认不切换窗口,所以需要有方法实现窗口的切换。
  切换窗口有如下方法:
  通过driver.window_handlers获取每个窗口的操作对象,然后for循环,实现切换窗口的目的。
  for handler in driver.window_handlers:
  driver.switch_to_window(handler)
  2)cookies处理
  在selenium中,cookies的用法一般是通过模拟登陆之后,获取到cookies值,再通过requests等库实现请求,
  增加效率。
  cookie = {}
  for item in driver.get_cookies():
  cookie[item.get('name')] = item.get('value')
  3)元素拖动
  在selenium中,除了简单的点击动作外,还有一些稍微复杂的动作,例如元素的拖动。
  在实现元素拖动中时,需要另一个模块。ActionChains模块满足对鼠标操作的要求。
  ActionChains的执行原理就是当调用ActionChains方法时,不会立即执行,而是将所有操作暂时放在一个队列中,
  当调用perform()方法时,队列会按照放入的先后顺序,依次执行。
  例如验证码的滑动,其实就是将一个元素拖动到某一个位置,然后松开,因此,元素的拖动经过下面几个步骤:
  将鼠标移上去,点击鼠标左键不放,拖拽到目的点,松开鼠标
  ActionChains中提供的方法:
# click(on_element=None)  # 单击鼠标左键
# click_and_hold(on_element=None)  # 点击鼠标左键,按住不放
# context_click(on_element=None)  # 点击鼠标右键
# double_click(on_element=None)  # 双击鼠标左键
# drag_and_drop(source, target)  # 拖拽到某个元素,然后松开
# drag_and_drop_by_offset(source, xoffset, yoffset)  # 拖拽到某个坐标然后松开
# move_by_offset(xoffset, yoffset)  # 鼠标移动到距离当前位置(x, y)
# move_to_element(to_element)  # 移动到某个元素
#
# # 将鼠标移动至距离某个元素多少距离的位置
# move_to_element_with_offset(to_element, xoffset, yoffset)
#
# release(on_element=None)  # 在某个元素位置松开鼠标左键
# perform()  # 执行链中的所有动作
  来一个简单拖拽例子,测试网址:http://www.treejs.cn/v3/demo/cn/exedit/drag.html
# 实例化一个浏览器
driver = webdriver.Chrome()
try:
# 打开页面(可以自己先打开试一下)
driver.get('http://www.treejs.cn/v3/demo/cn/exedit/drag.html')
time.sleep(2)
# 1.通过id定位要拖动的元素元素(相当于鼠标移上去)
element = driver.find_element_by_id('treeDemo_2_span')
# 2.通过id定位要拖到的位置
target = driver.find_element_by_id('treeDemo_3_span')
# 拖动元素到目标位置
ActionChains(driver).drag_and_drop(element, target).perform()
time.sleep(15)
finally:
driver.quit()
  4)弹窗处理
  很多网页里面可能会出现弹窗,需要对弹窗进行一定的处理(就是js里的alert())。
  使用:
  driver.switch_to_alert()  # 切换到弹窗
  time.sleep(3)
  driver.accept()  #接收弹窗
  对弹窗进行处理。
  4.页面等待
  现在的大多数web应用,越来越多采用了Ajax技术,以及程序不能确定某个元素何时完全加载出来,
  这会让元素定位困难,会提高ElementNotVisibleException的概率,因此出现了等待。
  就是等待页面中,某个元素出现,就执行某些代码。selenium提供两种等待方式。
  1)显式等待
  显式等待是指定某个条件,然后设置最长等待时间,如果在这个时间里面还没有找到元素,那么就抛出异常。
  显式等待主要使用了WebDriverWait类和excepted_conditions模块。
  WebDriverWait是显式等待类,主要方法有:
  __init__中:
  driver:传入的WebDriver实例
  timeout:超时时间,等待的最长时间(同时要考虑隐性等待时间)
  poll_frequency:调用until中的方法的间隔时间,默认0.5s
  ignored_exception:忽略的异常,如果在调用until过程中,抛出这个异常,则不中断代码,继续执行
  如果抛出的是这个元组外的异常,则代码中观,抛出异常,默认只有NoSuchElementException
  until中:
  method:在等待期间,每隔一段时间,调用这个传入的方法,直到返回值不是False
  message:如果超时,抛出TimeOutException,将message传入异常。
  expected_conditions是selenium的一个模块,其中包含一系列等待的条件:
"""
title_is
title_contains
#这两个条件类验证title,验证传入的参数title是否等于或包含于driver.title
presence_of_element_located
presence_of_all_elements_located
这两个人条件验证元素是否出现,传入的参数都是元组类型的locator,如(By.ID, 'kw')
顾名思义,一个只要一个符合条件的元素加载出来就通过;另一个必须所有符合条件的元素都加载出来才行
visibility_of_element_located
invisibility_of_element_located
visibility_of
这三个条件验证元素是否可见,前两个传入参数是元组类型的locator,第三个传入WebElement
第一个和第三个其实质是一样的
text_to_be_present_in_element
text_to_be_present_in_element_value
这两个人条件判断某段文本是否出现在某元素中,一个判断元素的text,一个判断元素的value
frame_to_be_available_and_switch_to_it  # 有些页面有frame标签,该标签出现后才能定位,例如QQ空间
这个条件判断frame是否可切入,可传入locator元组或者直接传入定位方式:id、name、index或WebElement
alert_is_present
这个条件判断是否有alert出现
element_to_be_clickable
这个条件判断元素是否可点击,传入locator
"""
简单例子:访问百度首页,等待“贴吧”那个地方出现,出现后就拿它的链接
from selenium import webdriver
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.by import By
driver = webdriver.Chrome()
driver.get('https://baidu.com')
locator = (By.LINK_TEXT, '贴吧')
try:
# 等待,直到locator出现为止,driver在20s内,每0.5s检查一次该元素是否出现
# 出现,执行后面内容;没出现,继续等待(20s内,超过就抛异常)
WebDriverWait(driver, 20, 0.5).until(EC.presence_of_element_located(locator))
# 出现后直接获取贴吧的链接
print(driver.find_element_by_link_text('贴吧').get_attribute('href'))
finally:
driver.close()
  2)隐式等待
  隐性等待implicitly_wait(xx),隐形等待是设置了一个最长等待时间,
  如果在规定时间内网页加载完成,则执行下一步,
  否则一直等到时间截止,然后执行下一步。
  弊端就是程序会一直等待整个页面加载完成,就算你需要的元素加载出来了还是需要等待。
  from selenium import webdriver
  driver = webdriver.Chrome()
  driver.implicitly_wait(30)
  driver.get('https://www.baidu.com')
  2.Phantomjs的使用
  PhantomJS是一个无界面的,可脚本编程的WebKit浏览器引擎。
  它原生支持多种web 标准:DOM 操作,CSS选择器,JSON,Canvas 以及SVG。
  phantomjs常用配置:
# 引入配置对象DesiredCapabilities
from selenium.webdriver.common.desired_capabilities import DesiredCapabilities
dcap = dict(DesiredCapabilities.PHANTOMJS)
#从USER_AGENTS列表中随机选一个浏览器头,伪装浏览器
dcap["phantomjs.page.settings.userAgent"] = (random.choice(USER_AGENTS))
# 不载入图片,爬页面速度会快很多
dcap["phantomjs.page.settings.loadImages"] = False
# 设置代理
service_args = ['--proxy=127.0.0.1:9999','--proxy-type=socks5']
#打开带配置信息的phantomJS浏览器
driver = webdriver.PhantomJS(phantomjs_driver_path, desired_capabilities=dcap, service_args=service_args)
# 如果你的phantomjs配置了环境变量,就不需要填写路径(phantomjs_driver_path),否则需要添加你的phantomjs所在的路径
  3.模拟登陆
  利用selenium登录新浪微博,首先访问新浪页面,等待输入账号和密码的输入框出现,然后依次利用selenium输入账号和密码,最后点击登录,并获取登陆后的源码。
# coding = utf-8
"""
用selenium登录新浪微博
"""
import time
from selenium import webdriver
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.by import By
# 1,实例化浏览器驱动
driver = webdriver.Chrome()
url = 'https://weibo.com/'
# 2,给定访问目标
driver.get(url)
# 定位账号密码输入框并登录
username = '184419810@qq.com'
password = '*********'
try:
# 创建等待事件
waiter = WebDriverWait(driver, 20, 0.1)
event = (By.XPATH, '//*[@id="pl_login_form"]')
# # 等待登录框的出现
waiter.until(EC.presence_of_element_located(event))
# 账号输入
driver.find_element_by_xpath('//*[@id="loginname"]').send_keys(username)
time.sleep(1)
# 密码输入
driver.find_element_by_xpath('//*[@id="pl_login_form"]/div/div[3]/div[2]/div/input').send_keys(password)
time.sleep(1)
# 点击登录
driver.find_element_by_xpath('//*[@id="pl_login_form"]/div/div[3]/div[6]/a').click()
# 获取登录后的源码
print(driver.page_source)
except Exception as e:
print(e)
finally:
time.sleep(10)
driver.close()
上文内容不用于商业目的,如涉及知识产权问题,请权利人联系博为峰小编(021-64471599-8017),我们将立即处理。

Python+Selenium大型电商项目(京东商城)实战直播,优惠名额抢占中>>

免费找bug

学软件测试

评 论

论坛新帖

顶部 底部


建议使用IE 6.0以上浏览器,800×600以上分辨率,法律顾问:上海瀛东律师事务所 张楠律师
版权所有 上海博为峰软件技术股份有限公司 Copyright©51testing.com 2003-2018, 沪ICP备05003035号
投诉及意见反馈:webmaster@51testing.com; 业务联系:service@51testing.com 021-64471599-8017

沪公网安备 31010102002173号

51Testing官方微信

51Testing官方微博

扫一扫 测试知识全知道