【Python 】selenium 简介

从源码中可以找到selenium 2.48.0支持的浏览器如下：

　　[python] view plain copy

1. Firefox
2. Chrome
3. ChromeOptions
4. Ie
5. Edge
6. Opera
7. Safari
8. BlackBerry
9. PhantomJS
10. Android
11. Remote
12. DesiredCapabilities
13. ActionChains
14. TouchActions

一、安装与配置

 

以笔者用的Chrome为例：

1.安装selenium

命令行运行：

1. pip install selenium

2.安装ChromeDriver 2.19
官网：https://sites.google.com/a/chromium.org/chromedriver/

下载：http://chromedriver.storage.googleapis.com/index.html （可能无法打开，请自备梯子，笔者正在使用这一款：ShadowSocks）

若使用其他浏览器测试须另行下载对应的Driver。

下载后放到Chrome安装目录下，并把该目录添加到path环境变量。

也可以直接把chromedriver放到Python的根目录。总之它要放在Path变量的文件夹中。

二、基本用法

1. from selenium import webdriver
2. from selenium.webdriver.common.action_chains import ActionChains #ActionChains鼠标操作类
3. from selenium.webdriver.common.keys import Keys #keys类操作
4. # 打开浏览器：
5. browser = webdriver.Chrome()
6. # # 若未配置环境变量：
7. # path = r'C:\Program Files\Google\Chrome\Application\chromedriver.exe'
8. # browser = webdriver.Chrome(executable_path = path)
9. # 在指定时间范围等待：
10. browser.implicitly_wait(30)
11. # 设置超时
12. browser.set_page_load_timeout(30)
13. browser.set_script_timeout(30)
14. # 打开url:
15. browser.get(r"http://www.你的网站.com")
16. # 前进/后退
17. browser.forward()
18. browser.back()
19. # 刷新
20. browser.refresh()
21. # 将浏览器最大化
22. browser.maximize_window()
23. # 将设置浏览器为宽480，高800显示'
24. browser.set_window_size(480, 800)
25. # 关闭浏览器
26. browser.quit()
27. # 关闭标签
28. browser.close()
29. '''''
30. 获取浏览器窗口相关信息
31. browser.title # 当前页面标题
32. browser.name # 浏览器名
33. browser.current_url # 返回当前页面url
34. browser.window_handles # 返回当前浏览器的所有窗口
35. browser.current_window_handle # 返回当前浏览器的窗口句柄
36. '''
37. # 跳转到其他标签页
38. window = browser.window_handles
39. browser.switch_to_window(window[1])
40. # 选择窗口
41. browser.switch_to_window('window_name')
42. # 截取当前显示的页面并保存
43. browser.get_screenshot_as_file(r'd:\backup\140591\桌面\首页图片\test.png')
44. # 定位元素
45. # 需要先分析源码
46. text_username = browser.find_element_by_name('username')
47. text_password = browser.find_element_by_name('password')
48. form_login = browser.find_element_by_name('formlogin')
49. # 也可以使用其他方法如xpath，注意转义符
50. text_username = browser.find_element_by_xpath('//input[@name = \'username\']')
51. '''''
52. 所有相关方法：
53. find_element(by='id', value=None)
54. find_element_by_class_name(name)
55. find_element_by_css_selector(css_selector)
56. find_element_by_id(id)
57. find_element_by_link_text(link_text)
58. find_element_by_name(name)
59. find_element_by_partial_link_text(link_text)
60. find_element_by_tag_name(name)
61. find_element_by_xpath(xpath)
62. element后加s则返回一组对象
63. '''
64. #-----------------------WebElement相关-----------------------
65. # 输入值
66. # send_keys也可以用于上传文件：send_keys('d:/abc.txt')
67. text_username.send_keys("你的用户名_错误")
68. text_password.send_keys("你的密码")
69. # 清空输入框，换成正确的用户名
70. text_username.clear()
71. text_username.send_keys("你的用户名_正确")
72. # 提交表单
73. form_login.submit()
74. '''''
75. 相关方法
76. clear()  # 清除元素的内容
77. send_keys()  # 在元素上模拟按键输入
78. click()  # 单击元素
79. submit()  # 提交表单
80. size()  # 返回元素的尺寸
81. text()  # 获取元素的文本
82. get_attribute(name)  # 获得属性值
83. is_displayed()  # 设置该元素是否用户可见
84. '''
85. #-------------------alert/confirm/prompt对话框处理-------------------
86. '''''
87. switch_to_alert()   # 用于获取网页上的警告信息。
88. text   # 返回 alert/confirm/prompt 中的文字信息。
89. accept()   # 点击确认按钮。
90. dismiss()   # 点击取消按钮，如果有的话。
91. send_keys()   # 输入值，这个alert\confirm没有对话框就不能用了，不然会报错。
92. '''
93. #-----------------------下拉框处理-----------------------
94. # 二次定位
95. # driver.find_element_by_xx('xx').find_element_by_xx('xx').click()
96. #先定位到下拉框
97. m=driver.find_element_by_id("ShippingMethod")
98. #再点击下拉框下的选项
99. m.find_element_by_xpath("//option[@value='10.69']").click()
100. #-----------------------cookie处理-----------------------
101. '''''
102. get_cookies()   # 获得所有cookie信息
103. get_cookie(name)   # 返回特定name 有cookie信息
104. add_cookie(cookie_dict)   # 添加cookie，必须有name 和value 值
105. delete_cookie(name)   # 删除特定(部分)的cookie信息
106. delete_all_cookies()   # 删除所有cookie信息
107. '''
108. #-----------------------文件上传-----------------------
109. # 定位上传按钮，添加本地文件
110. # driver.find_element_by_xx('xx').send_keys('d:/abc.txt')
111. driver.find_element_by_name("file").send_keys('D:\\selenium_use_case\upload_file.txt')
112. #-----------------------文件下载-----------------------
113. fp = webdriver.FirefoxProfile()
114. fp.set_preference("browser.download.folderList",2)
115. fp.set_preference("browser.download.manager.showWhenStarting",False)
116. fp.set_preference("browser.download.dir", os.getcwd())
117. fp.set_preference("browser.helperApps.neverAsk.saveToDisk", "application/octet-stream")
118. browser = webdriver.Firefox(firefox_profile=fp)
119. browser.get("http://pypi.python.org/pypi/selenium")
120. browser.find_element_by_partial_link_text("selenium-2").click()
121. #-----------------------Keys类键盘操作-----------------------
122. from selenium.webdriver.common.keys import Keys
123. # element.send_keys(...)，下同
124. '''''
125. send_keys(Keys.BACK_SPACE) # 返回键
126. send_keys(Keys.BACK_SPACE) 删除键（BackSpace）
127. send_keys(Keys.SPACE)  空格键(Space)
128. send_keys(Keys.TAB)  制表键(Tab)
129. send_keys(Keys.ESCAPE)  回退键（Esc）
130. send_keys(Keys.ENTER) 回车键（Enter）
131. send_keys(Keys.CONTROL,'a') 全选（Ctrl+A）
132. send_keys(Keys.CONTROL,'c') 复制（Ctrl+C）
133. send_keys(Keys.CONTROL,'x') 剪切（Ctrl+X）
134. send_keys(Keys.CONTROL,'v') 粘贴（Ctrl+V）
135. '''
136. # ----------------ActionChains类鼠标操作--------------------
137. from selenium.webdriver.common.action_chains import ActionChains
138. # 右键单击
139. ActionChains(browser).context_click(text_username).perform()
140. # 双击
141. ActionChains(browser).double_click(text_username).perform()
142. #执行元素的移动操作
143. ActionChains(browser).drag_and_drop(text_username, text_password).perform()
144. '''''
145. ActionChains 类鼠标操作的常用方法：
146. context_click()   # 右击
147. double_click()   # 双击
148. drag_and_drop()   # 拖动
149. move_to_element()   # 鼠标悬停在一个元素上
150. click_and_hold()   # 按下鼠标左键在一个元素上
151. '''
152. # -----------执行JavaScript脚本 --------------
153. js="var q=document.getElementById(\"user_name\");q.style.border=\"1px solid red\";"
154. #调用js
155. browser.execute_script(js)
156. # eg:
157. # 将页面滚动条拖到底部
158. js="var q=document.documentElement.scrollTop=10000"
159. browser.execute_script(js)
160. # 隐藏文字信息
161. driver.execute_script('$("#tooltip").fadeOut();')
162. # 隐藏按钮：
163. button = driver.find_element_by_class_name('btn')
164. driver.execute_script('$(arguments[0]).fadeOut()',button)

三、Demo

1. 用Chrome打开百度搜索关键字：Python

先分析一下百度首页的源码，Chrome中按F12打开控制台，用控制台左上角的放大镜图标找出输入框和提交按钮对应的代码。我们可以用id属性来获取这两个控件，如下图：

我们可以根据name、class、id等获取该控件。

完整代码如下：

1. from selenium import webdriver
2. from selenium.common.exceptions import NoSuchElementException
3. import time
4. # 打开浏览器：
5. browser = webdriver.Chrome()
6. browser.set_page_load_timeout(30)
7. browser.set_script_timeout(30)
8. # 在此时间范围内智能等待操作完成
9. browser.implicitly_wait(30)
10. # 每次点击链接后等待的秒数，用于观察操作效果
11. wait_time = 5
12. # 打开url:
13. browser.get(r"http://www.baidu.com")
14. time.sleep(wait_time)
15. # 获取输入框和提交按钮
16. search_input = browser.find_element_by_id("kw")
17. search_submit = browser.find_element_by_id("su")
18. # 输入Python
19. search_input.send_keys("Python")
20. # 点击「百度一下」按钮
21. search_submit.click()
22. time.sleep(wait_time)
23. # 获取百度百科的链接
24. target = browser.find_element_by_partial_link_text("Python_百度百科")
25. target.click()
26. # 关闭
27. browser.quit()

 

3 脚本解析

# coding = utf-8

可加可不加，开发人员喜欢加一下，防止乱码嘛。

from selenium import webdriver

要想使用 selenium 的 webdriver 里的函数，首先把包导进来嘛

browser = webdriver.Firefox()

我们需要操控哪个浏览器呢？Firefox ，当然也可以换成 Ie 或 Chrome 。browser 可以

随便取，但后面要用它操纵各种函数执行。

browser.find_element_by_id("kw").send_keys("selenium")

一个控件有若干属性 id 、name、（也可以用其它方式定位），百度输入框的 id 叫 kw ，

我要在输入框里输入 selenium 。多自然语言呀！

browser.find_element_by_id("su").click()

搜索的按钮的 id 叫 su ，我需要点一下按钮（ click() ）。

browser.quit()

退出并关闭窗口的每一个相关的驱动程序，有洁癖用这个。

browser.close()

关闭当前窗口 ，用哪个看你的需求了。

 

#通过 id 方式定位

browser.find_element_by_id("kw").send_keys("selenium")

#通过 name 方式定位

browser.find_element_by_name("wd").send_keys("selenium")

#通过 tag name 方式定位

browser.find_element_by_tag_name("input").send_keys("selenium")

#通过 class name 方式定位

browser.find_element_by_class_name("s_ipt").send_keys("selenium")

 

#通过 CSS 方式定位

browser.find_element_by_css_selector("#kw").send_keys("selenium")

#通过 xphan 方式定位

browser.find_element_by_xpath("//input[@id='kw']").send_keys("selenium")

############################################

browser.find_element_by_id("su").click()

time.sleep(3)

browser.quit()