前言. python自动化:Helium 是什么?
Helium 是一款 Web 端自动化开源框架,全称是:Selenium-Python-Helium,从名字上就可以看出,Helium 似乎和 Selenium 息息相关
确实,Helium 针对 Selenium 进行了封装,它屏蔽了 Selenium 很多实现细节,提供了更加简洁直观的 API,更方便我们进行 Web 端的自动化
官方表示,要实现同样的功能,Helium 相比 Selenium 要少 30% - 50% 的代码
目前,Helium 仅支持 Chrome 和 FireFox
优缺点
Helium 主要包含下面 6 个优点:
- Helium 自带 WebDriver,不需要下载、配置浏览器驱动
- 内嵌页面 iframe 页面元素直接操作,不需要使用 switch_to.frame() 切换 iframe
- 窗体管理更方便,可以直接使用窗口标题或部分标题内容来切换窗体
- 隐式等待,针对某个元素执行点击操作,Selenium 如果元素没有出现,脚本会执行失败;而 Helium 默认最多等待 10s,等待元素出现后立马执行点击操作
- 显式等待,Helium 提供更加优雅的 API 来等待页面元素出现
- API 更简洁直观,代码量少
Helium 主要缺点,体现在:
- 由于封装,屏蔽了很多细节,所以它不合适二次开发
- 目前仅支持 Chrome 和 FireFox 浏览器
- 版本更新慢、遗留 Bug 及文档少
安装依赖
pip3 install helium
使用
1. 启动浏览器
def start_firefox(url=None, headless=False, options=None)
- 这三个参数都有默认值,为None,所以都为可选填的
url:要访问的网站网址
headless:是否隐藏浏览器,即执行脚本时,浏览器是否可见
options:浏览器配置,要使用options时需引入配置模块
例子:
start_firefox("google.com")
start_chorme(headless=True)
from selenium.webdriver import ChromeOptions
#firefox引入FirefoxOptions
options = ChromeOptions()
options.add_argument('--start-maximized')
options.add_argument('--proxy-server=1.2.3.4:5678')
start_chrome("www.baidu.com",headless=True,options=options)
def open_url():
# 加启动配置
options = webdriver.ChromeOptions()
# 防止反爬
options.add_experimental_option('useAutomationExtension', False)
options.add_experimental_option('excludeSwitches', ['enable-automation'])
driver = webdriver.Chrome(chrome_options=options)
set_driver(driver)
'''
<script>
if(window.navigator.webdriver == true){
document.write("<span>看到这段就代表你是爬虫</span>")
}else{
document.write("<span>真正的信息在这儿呢</span>")
}
</script>
'''
go_to(r"file:///E:/auto_task/helium_test/test.html")
time.sleep(5)
访问网址
def go_to(url)
写入
def write(text, into=None)
- 顾名思义,将内容写入输入框的功能,其中into为非必填,如果没填的话,默认输入到页面找到的第一个输入框,如果要填的话,对应的是element。
options = ChromeOptions()
options.add_argument("--start-maximized")
start_chrome("https://www.zhihu.com/signin?next=%2F",options=options)
click(Text("密码登录"))
write("python@123.com",into=S("@username"))
write("python",into="密码")
write("royfans",TextField("Username"))
# 大小写不敏感
write("Zlf111",into="Password")
write("test2",into=TextField(to_right_of="username"))
click("Remember password")
click("Log in")
click(TextField("Site"))
按键
def press(key)
- 普通字母数字直接用’key’就可以了,如果是特殊按键,这用对应的按键名即可,组合件的话,用+号连接
press('A')
press('a')
press(ENTER)
press(CONTROL + 'a')
- 特殊按键对应的按键名可以在selenium.webdriver.common.keys.Keys找到。具体如下:
NULL = Keys.NULL
CANCEL = Keys.CANCEL
HELP = Keys.HELP
BACK_SPACE = Keys.BACK_SPACE
TAB = Keys.TAB
CLEAR = Keys.CLEAR
RETURN = Keys.RETURN
ENTER = Keys.ENTER
SHIFT = Keys.SHIFT
LEFT_SHIFT = Keys.LEFT_SHIFT
CONTROL = Keys.CONTROL
LEFT_CONTROL = Keys.LEFT_CONTROL
ALT = Keys.ALT
LEFT_ALT = Keys.LEFT_ALT
PAUSE = Keys.PAUSE
ESCAPE = Keys.ESCAPE
SPACE = Keys.SPACE
PAGE_UP = Keys.PAGE_UP
PAGE_DOWN = Keys.PAGE_DOWN
END = Keys.END
HOME = Keys.HOME
LEFT = Keys.LEFT
ARROW_LEFT = Keys.ARROW_LEFT
UP = Keys.UP
ARROW_UP = Keys.ARROW_UP
RIGHT = Keys.RIGHT
ARROW_RIGHT = Keys.ARROW_RIGHT
DOWN = Keys.DOWN
ARROW_DOWN = Keys.ARROW_DOWN
INSERT = Keys.INSERT
DELETE = Keys.DELETE
SEMICOLON = Keys.SEMICOLON
EQUALS = Keys.EQUALS
NUMPAD0 = Keys.NUMPAD0
NUMPAD1 = Keys.NUMPAD1
NUMPAD2 = Keys.NUMPAD2
NUMPAD3 = Keys.NUMPAD3
NUMPAD4 = Keys.NUMPAD4
NUMPAD5 = Keys.NUMPAD5
NUMPAD6 = Keys.NUMPAD6
NUMPAD7 = Keys.NUMPAD7
NUMPAD8 = Keys.NUMPAD8
NUMPAD9 = Keys.NUMPAD9
MULTIPLY = Keys.MULTIPLY
ADD = Keys.ADD
SEPARATOR = Keys.SEPARATOR
SUBTRACT = Keys.SUBTRACT
DECIMAL = Keys.DECIMAL
DIVIDE = Keys.DIVIDE
F1 = Keys.F1
F2 = Keys.F2
F3 = Keys.F3
F4 = Keys.F4
F5 = Keys.F5
F6 = Keys.F6
F7 = Keys.F7
F8 = Keys.F8
F9 = Keys.F9
F10 = Keys.F10
F11 = Keys.F11
F12 = Keys.F12
META = Keys.META
COMMAND = Keys.COMMAND
点击/双击/右击
def click(element) #单击
def doubleclick(element) #双击
def rightclick(element)
示例:
click("Sign in")
click(Button("OK"))
click(Point(200, 300))
click(ComboBox("File type").top_left + (50, 0))
拖拽
def drag(element, to)
- 将一个元素拖拽到指定位置,这个个人使用时,自己用elment-ui写了个drag功能一直拖拽失败,比较尴尬。
官方示例:
drag("Drag me!", to="Drop here.")
拖拽文件
def drag_file(file_path, to)
将本地的文件拖入到浏览器指定的位置,挺好用的一个功能。
示例:
click("配置管理")
wait_until(Text("文件上传").exists)
click("文件上传")
drag_file(r"D:\\1.jpg",to="将文件拖到此处,或点击上传")
上传文件
def attach_file(file_path, to=None)
- 将文件上传,这个针对的是input type=“file”
示例:
start_chrome("file:///D:/1.html",options=options)
attach_file(r"D:\\2.jpg")
按住/松开左键
def press_mouse_on(element)
def release_mouse_over(element)
鼠标悬浮
def hover(element)
示例:
hover("File size")
hover(Button("OK"))
hover(Link("Download"))
hover(Point(200, 300))
hover(ComboBox("File type").top_left + (50, 0))
hover("任务中心")
click("任务查询")
向上/下/左/右滚动屏幕
def scroll_down(num_pixels=100)
def scroll_up(num_pixels=100)
def scroll_right(num_pixels=100)
def scroll_left(num_pixels=100)
不设置参数时默认为100像素,设置参数的话直接写数字就可以
查找元素组
def find_all(predicate)
这个比较简单,就是找元素组,然后根据index来取元素。示例:
find_all(Button("Open"))
find_all(Window())
find_all(TextField("Address"))
find_all(Text("进入"))
click(find_all(Text("进入"))[0])
# 有相同的placehold,用find_all 查找全部元素
p = find_all(TextField("Please select"))
click(p[0])
下拉框选择
def select(combo_box, value)
从下拉选项中选择一项,示例:
select("Language", "English")
select(ComboBox("Language"), "English")
刷新页面
def refresh()
直接使用即可
等待元素
def wait_until(condition_fn, timeout_secs=10, interval_secs=0.5)
等待某个条件为真才继续往下执行。默认的超时时间为10s,每0.5查询一次,这俩参数选填。可以设置超时时间和轮询间隔。
注意:condition_fn等待条件直接采用函数的话,不能带()或者直接用lambda后带完整函数。
示例:
wait_until(Text("Finished!").exists)
wait_until(lambda:Text("Finished!").exists())
配置文件—隐式等待时间
等待时间:
首先要知道隐式等待时间的含义,是指执行脚本时,如果某个元素元素刚开始没找到,会默认等待一段时间,超时后才会抛出错误。显示等待这是直接在代码里写等待多久,比如time.sleep(xxx)
class Config:
里面只介绍了隐式等待时间的设置。默认值为10s,如果要自己设置,直接调用config即可:
在这里插入图片描述如果要关闭隐式等待时间,则只需要将时间设置为0即可。
onfig.implicit_wait_secs(10)
UI基础类: class GUIElement
最基础的UI类,用的上的只有一个函数
def exists(self)
用于判断某个元素是否存在,返回true或false
元素基础类:class HTMLElement(GUIElement)
界面元素类,继承GUIElement,有六个属性:
width #元素宽度
height #高度
x #元素在界面上的X轴坐标
y #y轴坐标
top_left #元素坐上角的坐标
web_element #转换为web_element,之后可以调用selenium元素的属性和方法
start_chrome("file:///D:/1.html",options=options)
element = Button("submit")
width = element.width
height = element.height
x = element.x
y = element.y
top_left = element.top_left
web_el_text = element.web_element.text
web_el_tag = element.web_element.tag_name
print("元素大小:{width}*{height},元素位置:{x},{y},元素左上角位置:{top_left},元素文本:{web_el_text},元素类别:{web_el_tag}".format(
width = width,
height = height,
x = x,
y = y,
top_left = top_left,
web_el_text = web_el_text,
web_el_tag = web_el_tag
))
结果如下图,可以发现x,y实际就是top_left的具体坐标值,而top_left的返回类型为Point,所以要对top_left做位置变换需要用Point数据:
元素大小:85*26,元素位置:1407,146,元素左上角位置:Point(x=1407, y=146),元素文本:Search,元素类别:button
此外,要关注一下__init__方法,也就是类的初始化:
def __init__(self, below=None, to_right_of=None, above=None, to_left_of=None)
1
这个是说在可以通过相对位置来实现元素及控件的查找
below #在某个元素下方
above #在某个元素上方
to_right_of #在某个元素右侧
to_left_of #在某个元素左侧
所有基于HTMLElement类的元素或者控件都可以通过这4个相对位置来实现元素查找。
例子:
Button("登录",below="password") #位于password下的登录按钮
元素定位类:class S(HTMLElement)
它是基于HTMLElement的,不同定位方式是通过符号来区分的
@ #对应name
. #对应class
# #对应id
// #对应xpath或者css selector
示例:
write("testname",into=S("#username"))
find_all(S("table > tr > td", below="Email"))
write("abc",into=S("[type=text]"))
Helium比较特殊,元素定位类是继承HTMLElement的,而其他控件,比如Button、Link等也全都继承HTMLElement,所以这几个是并列关系,也就是说
控件无法通过定位类S来实现控件查找,只能基于控件文本和HTMLElement中的四个相对位置来实现控件查找
比如Button(S("#submit")是错误的,可以通过Button(“登录”,below=“username”)来实现查找。
控件类
所有控件类都继承HTMLElement类,所以可以试用HTMLElement和GUIElement类的方法与属性。
- 文本控件
用于获取界面文本的控件,拥有value属性来获取控件对应的文本
class Text(HTMLElement)
@property
def value(self)
示例
Text("Do you want to proceed?").exists()
Text(below="Email", to_right_of="John").value
- 链接控件
用于识别界面上的链接,拥有href属性可以获取链接对应的url
class Link(HTMLElement)
@property
def href(self)
示例:
click(Link("Block User", to_right_of="John Doe"))
Link("百度").href
3.序列控件
可以获取界面上的序列
class ListItem(HTMLElement)
click(ListItem("List item 1", below="My first list:"))
assert ListItem("List item 1").exists()
ListItem("Account Resource").exists()
4.按钮控件
可以获取按钮控件,具有is_enabled方法,用于判断按钮是否有被disabled
class Button(HTMLElement)
def is_enabled(self)
示例:
click(Button("OK"))
click(Button("Log In", below=TextField("Password")))
Button("OK").is_enabled()
5.图片控件
识别图片元素,从__init__类的初始化函数可以看到相对其他控件的text=None(,图片控件的是alt=None,也就是说要通过alt而非text来识别图片,至于什么是alt,这个是html基础,不懂得可以百度。
class Image(HTMLElement)
def __init__(self, alt=None, below=None, to_right_of=None, above=None,to_left_of=None)
示例:
click(Image(alt="logo"))
lick(Image("logo", to_left_of=ListItem("Download")))
Image("logo").exists()
6.文本框控件
可识别可以用于输入的文本框控件,它是通过label加相对位置来定位的,这个label既可以采用本文框左侧的文字,也可以采用文本框内部的提示语,也就是html的placeholder里的文本。
value属性,可以获取文本框里的文本;
is_enabled()方法,识别文本框是否被disabled,界面上显示的就是置灰
is_editable()方法,识别文本框是否可编辑,也就是是否有配置readonly,这时控件是不置灰的,但是不可编辑。
class TextField(HTMLElement)
def __init__(self, label=None, below=None, to_right_of=None, above=None,to_left_of=None)
@property
def value(self)
def is_enabled(self)
def is_editable(self)
示例:
html代码如下:
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>Title</title>
</head>
<body>
<h1>hello worlds</h1>
username:<input placeholder="请输入用户名"></input>
</body>
</html>
自动化代码如下:
from helium import *
from selenium.webdriver import ChromeOptions
import time
Config.implicit_wait_secs=30
options = ChromeOptions()
options.add_argument("--start-maximized")
start_chrome("file:///D:/1.html",options=options)
write("test1",into=TextField("username"))
time.sleep(3)
write("test2",into=TextField(to_right_of="username"))
time.sleep(3)
write("test3",into=TextField("请输入用户名"))
element = Button("submit")
上面3种方式都是可以找到input控件的,都可以成功输入。
7.下拉列表框控件
这个相对select来说范围更广,一般来说既可以指的select,也可以指dropdown控件,特别是一些可以通过输入文本来匹配选项的控件,都可以用这个控件。
查看__init__可以发现跟文本框是一样的,都是用label来识别控件,查找方法类似上面文本框控件。它本身经常和select一起使用,用于选中某一选项。
value属性:可以获取已选中项的值
options属性:可以获取所有可以选择的选项内容
is_editable()方法:判断是否可输入文本来匹配选项\
class ComboBox(HTMLElement)
def __init__(self, label=None, below=None, to_right_of=None, above=None,to_left_of=None)
def is_editable(self)
@property
def value(self)
@property
def options(self)
示例:
ComboBox("Language").value
select(ComboBox(to_right_of="John Doe", below="Status"), "Active")
8.复选框控件
顾名思义,识别复选框用的。
is_enabled()方法:判断是否被disabled
is_checked()方法:判断是否被选中\
class CheckBox(HTMLElement)
def __init__(self, label=None, below=None, to_right_of=None, above=None,to_left_of=None)
def is_enabled(self)
def is_checked(self)
示例:
CheckBox("已阅读xx").is_checked()
click(CheckBox("记住登录状态", below=Button("登录")))
9.单选框控件
单选框专用
is_selected()方法:判断是否被选中\
class RadioButton(HTMLElement)
def __init__(self, label=None, below=None, to_right_of=None, above=None,to_left_of=None)
def is_selected(self)
示例:
RadioButton("Windows").is_selected()
click(RadioButton("我同意", below="用户协议"))
弹窗Alert
这个是专门针对alert弹窗的,是继承GUIElement的,所以没法用四个相对位置来查找,不过一般alert都会位于界面最顶层,所以一般也比较容易识别出来。
通过__init__可以知道它可以通过文本来查找弹窗,特别是对于多个弹窗的话,用文本来区分即可,如果只有一个弹窗,可以search_text放空一般也可以找到
text属性:获取弹窗上的文本内容
accept()方法:相当于点击弹窗上的OK / 确认键
dismiss()方法:相当于点击弹窗上的Cancel / 取消按键
class Alert(GUIElement)
def __init__(self, search_text=None)
@property
def text(self)
def accept(self)
def dismiss(self)
示例:
write("12",into=Alert("请输入个数"))
Alert().accept()
Alert().dismiss()
坐标点Point
通过x,y来表示元素在界面上的具体坐标,如果要偏移位置,可以用+或者-来实现。
class Point(namedtuple('Point', ['x', 'y']))
def __init__(self, x=0, y=0)
@property #X轴坐标
def x(self)
@property #Y轴坐标
def y(self)
def __eq__(self, other) # 对比两个Point看是否相等
def __ne__(self, other) # 对比两个Point看是否相等
def __hash__(self) #hash,返回x + 7*y,暂时未知作用
def __add__(self,delta) #相加,delta对应某个point
def __radd__(self,delta) #功能和add一致,为啥叫radd未知
def __sub__(self,delta) #相减
def __rsub__(self,delta) #相减,同sub一样
示例:
print(Point(100,100)+ (Point(1,2)))
print(Point(100,100).__add__(Point(1,2)))
窗口
可以用于或者窗口title也可以配合switch_to来进行窗口的切换
class Window(GUIElement)
def __init__(self, title=None)
@property #窗口的title
def title(self)
@property #返回selenium driver的标识,本身不具备操控界面的能力
def handle(self)
示例:
print(Window().title)
switch_to(Window("标题"))
switch_to(find_all(Window())[0])
窗口切换switch_to
通过窗口的title来切换窗口
def switch_to(window)
示例:
switch_to("Google")
switch_to(Window("Google"))
switch_to(find_all(Window())[0])
关掉浏览器
就是单纯的关掉浏览器,如果不调用该方法,执行完后浏览器会保持打开状态
def kill_browser()
高亮highlight
为了凸显某一元素便于查看或者在操作某一元素前,可以采用highlight来标识出来,然后再采用截图的话,可以明确知道该步骤在页面上对哪个元素进行操作。
def highlight(element)
示例:
highlight(TextField("username"))
与selenium配合(set_driver/get_driver)
set_driver(driver)用于将selenium的driver切换到helium,这样就可以直接使用helium的api了
get_driver()用于获取helium的driver,然后driver后可以使用selenium语句
从selenium到helium:
from selenium import webdriver
from helium import *
driver = webdriver.Chrome()
set_driver(driver)
go_to("http://127.0.0.1:8080/#/login")
从helium到selenium:
from helium import *
start_chrome()
driver = get_driver() #或者直接driver = start_chrome()
element = driver.find_element_by_id('auto_test')
print(element.text)
driver.save_screenshot("D:/1.png")
实战应用
1.时间选择:
click(TextField("added at"))
click("17")
click("17")
click('ok')
2. 搜索下拉选择
ListItem("Account Resource").exists()
参考: