Selenium 浏览器查看信息技巧、截屏、切换窗口

WebDriver 对象的一些方法

3.6 获取当前窗口 title

driver.title  

比如打开百度，然后打印他的标题。

from selenium import webdriver

import time

driver = webdriver.Chrome('E:\ChromDriver\chromedriver.exe')

driver.implicitly_wait(10)

driver.get('https://www.baidu.com/')

print(driver.title)

driver.quit()

运行截图:

image.png

比如我百度搜索“宋曲”，他的标题也会跟着变。

image.png

对应的代码：

from selenium import webdriver
import time

driver = webdriver.Chrome('E:\ChromDriver\chromedriver.exe')

driver.implicitly_wait(10)

driver.get('https://www.baidu.com/')
#找到输入框，输入宋曲
driver.find_element_by_id("kw").send_keys('宋曲')
#点击搜索
driver.find_element_by_id('su').click()

time.sleep(1)

print(driver.title)

driver.quit()

运行截图：

image.png

大家注意，我这里为什么加 sleep(1) 因为如果我们的代码执行的很快，窗口还没跳转过来他就打印了窗口的标题,实际上他在打印标题的时候如果没有加 sleep(1) 他打印标题的时候还是百度的页面，有的时候我们打开的不是百度的页面，我们打开某个公司的官网，他的网络服务器可能做的不是那么好，速度时快时慢(比如1~10秒)，那我们这里要怎么硬性的等待10秒吗？这种方法太浪费时间，比如他大部分时间3秒内就响应了，那设置10秒是不是多余，我们可以用这种方法，我们打开一个窗口之后去查找里面一个元素，我们判断能不能找到这个元素，去确定这个窗口被打开了，我们搜索宋曲的时候里面会有一个结果。

image.png

id='1' 的 webelement，我们只要判断这个元素存不存在就判断页面已经加载成功了。

代码是：

from selenium import webdriver

driver = webdriver.Chrome('E:\ChromDriver\chromedriver.exe')

driver.implicitly_wait(10)

driver.get('https://www.baidu.com/')

driver.find_element_by_id("kw").send_keys('宋曲\n')

driver.find_element_by_id('1')

print(driver.title)

driver.quit()

这里就不用 sleep(1) 啦。是不是很完美！！！

3.6.1 获取当前窗口地址栏 url

driver.current_url

打开百度打印它的标题和URL，输入搜索“宋曲”打印当前的URL和标题

from selenium import webdriver

driver = webdriver.Chrome('E:\ChromDriver\chromedriver.exe')
driver.implicitly_wait(10)

driver.get('https://www.baidu.com/')
#打印标题及url
print(driver.title)
print(driver.current_url)

driver.find_element_by_id("kw").send_keys('宋曲\n')
driver.find_element_by_id('1')
#打印标题及url
print(driver.title)
print(driver.current_url)

driver.quit()

输出结果是：

image.png

这个地址我们可以拿他做一个校验，比如你点击链接之后看URL是不是你点链接的地址，方法是不是很简单，接下来说一下截屏。

• 截屏

driver.get_screenshot_as_file(img_path)

我们用代码试一下:

from selenium import webdriver
import time

driver = webdriver.Chrome('E:\ChromDriver\chromedriver.exe')
driver.implicitly_wait(10)

#打开百度，搜索宋曲
driver.get('https://www.baidu.com/')
driver.find_element_by_id("kw").send_keys('宋曲')

driver.find_element_by_id('su').click()
driver.find_element_by_id('1')

#截图
driver.get_screenshot_as_file('shot.png')

input('press any key to quit...')
driver.quit()

它截出来的效果是这样的。

shot.png

注意截图保存的路径我们可以自己设置比如driver.get_screenshot_as_file('d:\shot.png')就会存到D盘里面，如果我们没有设它默认的就是当前工作目录，我们可以点击

image.png

然后点击：

image.png

然后会看到它默认的工作的目录就是当前的项目文件夹。

image.png

有的时候我们不想截整个图片，比如我们要截某一个元素，比如截百度一下的log，需要找到它的 id ，先获取到这个元素，然后在调用 screenshot 方法。对应的代码是：

from selenium import webdriver
import time

driver = webdriver.Chrome('E:\ChromDriver\chromdriver2.43\chromedriver.exe')
driver.implicitly_wait(10)

driver.get('https://www.baidu.com/')
ele = driver.find_element_by_id('lg')
ele.screenshot('baidulog.png')
driver.quit()

截到的效果是：

baidulog.png

判断有没有截到可以通过True或Flast来判断。

3.6.2 切换窗口

比如我百度宋曲，点击第一个搜索结果，他会跳到一个新的窗口，我们前面用Webdriver来操控窗口，那新打开的窗口可不可以用Webdriver来操控呢，看下面的代码点击新窗口之后，点击“帮助”按钮。

image.png

代码如下：

from selenium import webdriver
import time

driver = webdriver.Chrome('E:\ChromDriver\chromedriver.exe')
driver.implicitly_wait(10)

driver.get('https://www.baidu.com/')
print(driver.title)

driver.find_element_by_id('kw').send_keys('宋曲\n')

#点击第一条搜索结果，进入百度百科
ele=driver.find_element_by_xpath('//div[@id="1"]/h3/a').click()
#点击帮助按钮
ele2 = driver.find_element_by_xpath('//button[@id="searchLemma"]/following-sibling::a').click()

title=driver.title

print(title)

driver.quit()

它会报错：

image.png

会报错找不到元素，这里的原因是：driver 对象他在前面的窗口当中去控制浏览器点击之类的，他对浏览器的控制权只停留在第一个上面，他只控制前面自己打开的窗口，如果点了一个超链接打开了一个新的窗口 webdriver 对新的窗口是没有控制权的，他不知道的新打开的连接 HTML 里面有什么内容，没办法控制新打开的窗口，怎么样才能把控制权从前面的窗口跳到新打开的窗口呢，我们可以通过切换窗口来解决。

• 切换窗口：

driver.switch_to.window(handle)

handle 代表窗口的句柄，本质是一个字符串。这个方法通过唯一一个参数，就可以切入新的窗口了。

• 如何获取 handle ？

通过:

driver.window_handles

方法获取当前浏览器所有窗口的 handle，通过循环切换窗口判断标题的方法来确定当前窗口是否为目标窗口，保留当前窗 handle:

driver.current_window_handle

我们在之前的代码上加上 driver.window_handles 方法，看看打印的结果是什么。

from selenium import webdriver
import time

driver = webdriver.Chrome('E:\ChromDriver\chromedriver.exe')
driver.implicitly_wait(10)

driver.get('https://www.baidu.com/')
print(driver.title)

driver.find_element_by_id('kw').send_keys('宋曲\n')

#点击第一条搜索结果，进入百度百科
ele=driver.find_element_by_xpath('//div[@id="1"]/h3/a').click()
# 显示 标题栏 文本，可以发现还是在当前网页
print(driver.title)
print(driver.window_handles)

driver.quit()

运行结果:

image.png

可以看到 driver.window_handles返回的是一个列表对象，这个列表里面有 2 个值分别对应了 2个窗口，如果我们打开的窗口很多，就会有好多字符串，我们不可能猜那个字符串对应那个窗口，那我们怎么知道那个是对应的呢，可以通过我们上面说的 title 和 URL都可以，比如我们这里用 title ，我们获取到 handles 可以去循环遍历一下，拿到 handle 就切进去，然后就判断宋曲的这个字符串是否被标题包含，如果包含就停止循环，我们可以看到其实我们停止循环他就切进来了，切进来之后我们打印下 title 看有没有切入。

from selenium import webdriver

driver = webdriver.Chrome('E:\ChromDriver\chromedriver.exe')
driver.implicitly_wait(10)

driver.get('https://www.baidu.com/')

driver.find_element_by_id('kw').send_keys('宋曲\n')

#点击第一条搜索结果，进入百度百科
ele=driver.find_element_by_xpath('//div[@id="1"]/h3/a').click()
# 显示 标题栏 文本，可以发现还是在当前网页
print(driver.title)
print(driver.window_handles)

for handle in driver.window_handles:
#切换到新窗口
    driver.switch_to_window(handle)
#检查是否是我们要进入的window
    if '宋曲_百度百科' in driver.title:
        break
#点击帮助按钮
ele2 = driver.find_element_by_xpath('//button[@id="searchLemma"]/following-sibling::a').click()

print(driver.title)

看运行截图:

image.png

这样就切入运行起来了， title 就变了，如果我们想判断当前的 handle 到底是那个，我们可以在 for 循环里面打印一下。

for handle in driver.window_handles:
    driver.switch_to_window(handle)
    if '宋曲_百度百科' in driver.title:
        print(handle)
        break

运行截图：

image.png

这样就可以啦！大家注意切入到新窗口的时候就失去了对原来窗口的控制权，如果想点击原来窗口的元素是没有效果的，想回到最初的窗口可以把当前的 handle 字符串保存下来，字符串相当于窗口的唯一 id，拿到 id 才能切换到窗口，保存主窗口 handle ：

driver.current_window_handle

代码：

from selenium import webdriver
import time

driver = webdriver.Chrome('E:\ChromDriver\chromedriver.exe')
driver.implicitly_wait(10)

driver.get('https://www.baidu.com/')

driver.find_element_by_id('kw').send_keys('宋曲\n')

#点击第一条搜索结果，进入百度百科
ele=driver.find_element_by_xpath('//div[@id="1"]/h3/a').click()
# 显示 标题栏 文本，可以发现还是在当前网页
print(driver.title)

#保存主窗口handle
mainWindow = driver.current_window_handle

# 显示 标题栏 文本，可以发现还是在当前网页
print(driver.window_handles)

for handle in driver.window_handles:
#切换至新窗口
    driver.switch_to_window(handle)
#检查是否是我们要进入的window
    if '宋曲_百度百科' in driver.title:
        print(handle)
        break
#点击帮助按钮
ele2 = driver.find_element_by_xpath('//button[@id="searchLemma"]/following-sibling::a').click()

#切换到主窗口

driver.switch_to_window(mainWindow)

driver.find_element_by_id('kw').send_keys('111')

time.sleep(3)

driver.quit()

我们打开多个窗口怎么关闭呢? 用 close() 方法：

driver.close()  #用于关闭当前窗口

代码：

from selenium import webdriver
import time

driver = webdriver.Chrome('E:\ChromDriver\chromedriver.exe')
driver.implicitly_wait(10)

driver.get('https://www.baidu.com/')

driver.find_element_by_id('kw').send_keys('宋曲\n')

#点击第一条搜索结果，进入百度百科
ele=driver.find_element_by_xpath('//div[@id="1"]/h3/a').click()

# 显示 标题栏 文本，可以发现还是在当前网页
print(driver.title)

#保存主窗口handle
mainWindow = driver.current_window_handle

# 显示 标题栏 文本，可以发现还是在当前网页
print(driver.window_handles)

for handle in driver.window_handles:
#切换到新窗口
    driver.switch_to.window(handle)
#检查是否是我们要进入的window
    if '宋曲_百度百科' in driver.title:
        print(handle)
        break
#点击帮助按钮
ele2 = driver.find_element_by_xpath('//button[@id="searchLemma"]/following-sibling::a').click()

#关闭窗口
driver.close()
time.sleep(5)

#切换到主窗口
driver.switch_to.window(mainWindow)

driver.find_element_by_id('kw').send_keys('111')

input('press any key to quit...')
driver.quit()

可以自己运行看下效果。