import os
import time
from PIL import Image
from selenium import webdriver
from appium import webdriver as app
from selenium.common.exceptions import *
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.common.action_chains import ActionChains
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.support.select import Select
from selenium.webdriver.support.wait import WebDriverWait
from common.logger import Log
from common import read_config def open_browser(browser='chrome'):
"""打开浏览器函数。"firefox"、"chrome"、"ie",'phantomjs'""" # 驱动路径
driver_path = os.path.abspath(os.path.dirname(os.path.dirname(__file__)))
try:
if browser == 'firefox':
executable_path = os.path.join(driver_path, 'driver\\geckodriver.exe')
# executable_path = os.path.join(driver_path, 'driver/geckodriver')
driver = webdriver.Firefox(executable_path=executable_path)
return driver
elif browser == 'chrome':
# 加启动配置,忽略 Chrome正在受到自动软件的控制 提示
option = webdriver.ChromeOptions()
option.add_argument('disable-infobars')
# chrome启动静默模式;默认显示浏览器界面
if read_config.chrome_interface != 'True':
option.add_argument('headless')
executable_path = os.path.join(driver_path, 'driver\\chromedriver.exe')
# executable_path = os.path.join(driver_path, 'driver/chromedriver')
driver = webdriver.Chrome(chrome_options=option, executable_path=executable_path)
return driver
elif browser == 'ie':
driver = webdriver.Ie()
return driver
elif browser == 'js':
driver = webdriver.PhantomJS()
return driver
else:
Log().warning('额，暂不支持此浏览器诶。先试试firefox、chrome、ie、phantomJS浏览器吧。')
return
except Exception as msg:
Log().error('浏览器出错了呀！%s' % msg)
return def open_app():
try:
desired_caps = {
'platformName': read_config.platform_name, 'deviceName': read_config.device_name, 'platformVersion': read_config.platform_version, 'appPackage': read_config.app_package, 'appActivity': read_config.app_activity, 'noReset': True, # 隐藏手机默认键盘
'unicodeKeyboard': True, 'resetKeyboard': True
}
# 关联appium
driver = app.Remote('http://127.0.0.1:4723/wd/hub', desired_caps)
return driver
except Exception as e:
raise Exception('连接 Appium 出错：{}'.format(e)) class Crazy:
"""基于原生的selenium框架做二次封装""" def __init__(self, driver):
"""启动浏览器参数化，默认启动chrome"""
self.driver = driver
self.action = ActionChains(self.driver)
self.timeout = 5 # 显示等待超时时间
self.t = 1
self.log = Log() def open(self, url, t=''):
"""get url，最大化浏览器，判断title"""
self.driver.get(url)
self.driver.implicitly_wait(10)
# 是否最大化浏览器
if read_config.maximize != 'True':
self.driver.maximize_window()
try:
WebDriverWait(self.driver, self.timeout, self.t).until(EC.title_contains(t))
self.log.info('打开网页成功！')
except TimeoutException:
self.log.error('打开%s title错误，超时' % url)
except Exception as msg:
self.log.error('打开网页产生的其他错误：%s' % msg) def find_element(self, locator):
"""重写元素定位方法"""
if not isinstance(locator, tuple):
self.log.error('locator参数必须是元组类型，而不是：{}'.format(type(locator)))
return ""
else:
try:
element = WebDriverWait(self.driver, self.timeout, self.t).until(
EC.presence_of_element_located(locator))
if element.is_displayed():
return element
except:
self.log.info('%s页面中未能找到元素%s' % (self, locator))
return "" def find_elements(self, locator):
"""定位一组元素"""
if not isinstance(locator, tuple):
self.log.error('locator参数必须是元组类型，而不是：{}'.format(type(locator)))
return ""
else:
try:
elements = WebDriverWait(self.driver, self.timeout, self.t).until(
EC.presence_of_all_elements_located(locator))
return elements
except:
self.log.info('%s页面中未能找到元素%s' % (self, locator))
return "" def click_coordinate(self, coordinate, timeout=10):
"""点击坐标"""
self.driver.tap(coordinate, timeout) def clicks(self, locator, n):
"""点击一组元组中的一个"""
element = self.find_elements(locator)[n]
element.click() def click(self, locator):
"""点击操作"""
element = self.find_element(locator)
element.click() def double_click(self, locator):
"""双击操作"""
element = self.find_element(locator)
self.action.double_click(element).perform() def send_keys(self, locator, text):
"""发送文本，清空后输入"""
element = self.find_element(locator)
element.clear()
element.send_keys(text) def sends_keys(self, locator, n, text):
"""选中一组元素中的一个，发送文本，清空后输入"""
element = self.find_elements(locator)[n]
element.clear()
element.send_keys(text) def send_keys_enter(self):
"""敲enter"""
self.action.send_keys(Keys.ENTER).perform() def send_keys_down(self):
"""敲向下键"""
self.action.send_keys(Keys.DOWN).perform() def send_keys_arrow_down(self):
self.action.send_keys(Keys.ARROW_DOWN).perform() def send_keys_arrow_right(self):
self.action.send_keys(Keys.ARROW_RIGHT).perform() def is_text_in_element(self, locator, text):
"""判断文本在元素里，没定位到元素返回False，定位到返回判断结果布尔值"""
try:
result = WebDriverWait(self.driver, self.timeout, self.t).until(
EC.text_to_be_present_in_element(locator, text))
self.log.info('is_text_in_element 成功')
return result
except TimeoutException:
self.log.error('%s元素没有定位到' % str(locator))
return False def is_text_in_value(self, locator, value):
"""判断元素的value值，没有定位到返回False，定位到返回判断结果布尔值"""
try:
result = WebDriverWait(self.driver, self.timeout, self.t).until(
EC.text_to_be_present_in_element_value(locator, value))
except TimeoutException:
self.log.error('元素没有定位到：%s' % str(locator))
return False
else:
return result def is_title(self, title):
"""判断title完全等于"""
result = WebDriverWait(self.driver, self.timeout, self.t).until(EC.title_is(title))
return result def is_title_contains(self, title):
"""判断title包含"""
result = WebDriverWait(self.driver, self.timeout, self.t).until(EC.title_contains(title))
return result def is_selected(self, locator):
"""判断元素被选中，返回布尔值， 一般用在下拉框"""
result = WebDriverWait(self.driver, self.timeout, self.t).until(EC.element_located_to_be_selected(locator))
return result def is_selected_be(self, locator, selected=True):
"""判断元素的状态，selected是期望的参数True/False，返回布尔值"""
result = WebDriverWait(self.driver, self.timeout, self.t).until(
EC.element_located_selection_state_to_be(locator, selected))
return result def is_alert_present(self):
"""判断页面是否有alert，有返回alert，没有返回False"""
result = WebDriverWait(self.driver, self.timeout, self.t).until((EC.alert_is_present()))
text = EC.alert_is_present()(self.driver)
if text:
self.log.info('alert弹框显示文本是：%s' % text.text)
else:
self.log.warning('没有发现alert弹框。')
return result def is_visibility(self, locator):
"""元素可见返回本身，不可见返回False"""
result = WebDriverWait(self.driver, self.timeout, self.t).until(EC.visibility_of_element_located(locator))
return result def is_invisibility(self, locator):
"""元素可见返回本身，不可见返回True，没有找到元素也返回True"""
result = WebDriverWait(self.driver, self.timeout, self.t).until(EC.invisibility_of_element_located(locator))
return result def is_clickAble(self, locator):
"""元素可以点击返回本身，不可点击返回False"""
result = WebDriverWait(self.driver, self.timeout, self.t).until(EC.element_to_be_clickable(locator))
return result def is_locator(self, locator):
"""判断元素有没有被定位到（并不意味着可见），定位到返回element，没有定位到返回False"""
result = WebDriverWait(self.driver, self.timeout, self.t).until(EC.presence_of_element_located(locator))
return result def move_to_element(self, locator):
"""鼠标悬停操作"""
element = self.find_element(locator)
ActionChains(self.driver).move_to_element(element).perform() def move(self, locator, locator1):
"""循环调用鼠标事件，死循环"""
self.move_to_element(locator)
time.sleep(2)
element = self.find_element(locator1)
self.move_to_element(locator1)
try:
if element.is_displayed:
self.click(locator1)
else:
self.move(locator, locator1)
except ElementNotVisibleException as e:
self.log.error('鼠标点击事件失败：%s' % e) def drag_and_drop(self, element, element1):
"""拖拽"""
ActionChains(self.driver).drag_and_drop(element, element1).perform()
ActionChains(self.driver).click_and_hold(element).release(element1).perform() def switch_frame(self, frame):
"""切换ifarm"""
try:
self.driver.switch_to_frame(self.find_element(frame))
self.log.info('切换iframe成功！')
except:
self.log.warning('没有发现iframe元素%s' % frame) def current_window_handle(self):
"""浏览器handle"""
return self.driver.current_window_handle def switch_window_handle(self, n):
"""切换handle"""
if not isinstance(n, int):
self.driver.switch_to.window(n)
else:
all_handle = self.driver.window_handles
self.driver.switch_to.window(all_handle[n]) def back(self):
"""返回之前的网页"""
self.driver.back() def forward(self):
"""前往下一个网页"""
self.driver.forward() def close(self):
"""关闭当前网页"""
self.driver.close() def quit(self):
"""关闭所有网页"""
self.driver.quit() def get_title(self):
"""获取title"""
return self.driver.title def get_texts(self, locator, n):
"""获取一组相同元素中的指定文本"""
element = self.find_elements(locator)[n]
if element:
return element.text
else:
return None def get_text(self, locator):
"""获取文本"""
element = self.find_element(locator)
if element:
return element.text
else:
return None def get_attribute(self, locator, name):
"""获取属性"""
element = self.find_element(locator)
if element:
return element.get_attribute(name) def js_execute(self, js):
"""执行js"""
return self.driver.execute_script(js) def js_focus_element(self, locator):
"""聚焦元素"""
target = self.find_element(locator)
self.driver.execute_script("arguments[0].scrollIntoView();", target) def js_scroll_top(self):
"""滚动到顶部"""
js = "var q=document.documentElement.scrollTop=0"
self.driver.execute_script(js) def js_scroll_bottom(self):
"""滚动到底部"""
js = "var q=document.documentElement.scrollTop=10000"
self.driver.execute_script(js) def select_by_index(self, locator, index):
"""通过索引，index是第几个，从0开始, 下拉框"""
element = self.find_element(locator)
Select(element).select_by_index(index) def select_by_value(self, locator, value):
"""通过value属性"""
element = self.find_element(locator)
Select(element).select_by_value(value) def select_by_text(self, locator, text):
"""通过text属性"""
element = self.find_element(locator)
Select(element).select_by_visible_text(text) def save_screenshot(self, img_path):
"""获取电脑屏幕截屏"""
self.driver.save_screenshot(img_path) def save_report_html(self):
"""可以在html报告中使用的截图"""
self.driver.get_screenshot_as_base64() def save_element_img(self, locator, img_path):
"""获取元素截图"""
self.driver.save_screenshot(img_path)
element = self.find_element(locator)
left = element.location['x']
top = element.location['y']
right = element.location['x'] + element.size['width']
bottom = element.location['y'] + element.size['height']
im = Image.open(img_path)
im = im.crop((left, top, right, bottom))
im.save(img_path) def get_cookies(self):
"""获取cookies"""
return self.driver.get_cookies() def swipeDown(self, t=500, n=1):
'''向下滑动屏幕'''
l = self.driver.get_window_size()
x1 = l['width'] * 0.5 # x坐标
y1 = l['height'] * 0.25 # 起始y坐标
y2 = l['height'] * 0.75 # 终点y坐标
for i in range(n):
self.driver.swipe(x1, y1, x1, y2, t) def swipeUp(self, t=500, n=1):
'''向上滑动屏幕'''
l = self.driver.get_window_size()
x1 = l['width'] * 0.5 # x坐标
y1 = l['height'] * 0.65 # 起始y坐标
y2 = l['height'] * 0.25 # 终点y坐标
for i in range(n):
self.driver.swipe(x1, y1, x1, y2, t) if __name__ == '__main__':
driver = open_browser()
driver.get('file:///D:/UIAutomation/report/2019-01-15%2017-40-10report.html')