使用selenium模拟登陆新浪微博
1.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()