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()
运行截图:
比如我百度搜索“宋曲”，他的标题也会跟着变。
对应的代码：
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()
运行截图：
大家注意，我这里为什么加 sleep(1) 因为如果我们的代码执行的很快，窗口还没跳转过来他就打印了窗口的标题,实际上他在打印标题的时候如果没有加 sleep(1) 他打印标题的时候还是百度的页面，有的时候我们打开的不是百度的页面，我们打开某个公司的官网，他的网络服务器可能做的不是那么好，速度时快时慢(比如1~10秒)，那我们这里要怎么硬性的等待10秒吗？这种方法太浪费时间，比如他大部分时间3秒内就响应了，那设置10秒是不是多余，我们可以用这种方法，我们打开一个窗口之后去查找里面一个元素，我们判断能不能找到这个元素，去确定这个窗口被打开了，我们搜索宋曲的时候里面会有一个结果。
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()
输出结果是：
这个地址我们可以拿他做一个校验，比如你点击链接之后看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()
它截出来的效果是这样的。
注意截图保存的路径我们可以自己设置比如driver.get_screenshot_as_file('d:\shot.png')就会存到D盘里面，如果我们没有设它默认的就是当前工作目录，我们可以点击
然后点击：
然后会看到它默认的工作的目录就是当前的项目文件夹。
有的时候我们不想截整个图片，比如我们要截某一个元素，比如截百度一下的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()
截到的效果是：
判断有没有截到可以通过True或Flast来判断。
3.6.2 切换窗口
比如我百度宋曲，点击第一个搜索结果，他会跳到一个新的窗口，我们前面用Webdriver来操控窗口，那新打开的窗口可不可以用Webdriver来操控呢，看下面的代码点击新窗口之后，点击“帮助”按钮。
代码如下：
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()
它会报错：
会报错找不到元素，这里的原因是：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()
运行结果:
可以看到 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)
看运行截图:
这样就切入运行起来了， title 就变了，如果我们想判断当前的 handle 到底是那个，我们可以在 for 循环里面打印一下。
for handle in driver.window_handles:
driver.switch_to_window(handle)
if '宋曲_百度百科' in driver.title:
print(handle)
break
运行截图：
这样就可以啦！大家注意切入到新窗口的时候就失去了对原来窗口的控制权，如果想点击原来窗口的元素是没有效果的，想回到最初的窗口可以把当前的 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()
可以自己运行看下效果。
