mirror of https://github.com/yingoja/DemoUI.git
129 lines
3.9 KiB
Python
129 lines
3.9 KiB
Python
#!/usr/bin/env python
|
|
# _*_ coding:utf-8 _*_
|
|
__author__ = 'YinJia'
|
|
|
|
import os,sys
|
|
sys.path.append(os.path.dirname(os.path.dirname(os.path.dirname(__file__))))
|
|
from config import setting
|
|
from selenium.webdriver.support.wait import WebDriverWait
|
|
from selenium.webdriver.support import expected_conditions as EC
|
|
from selenium.common.exceptions import NoSuchFrameException,NoSuchWindowException,NoAlertPresentException,NoSuchElementException
|
|
import configparser
|
|
from public.models.log import Log
|
|
|
|
con = configparser.ConfigParser()
|
|
con.read(setting.CONFIG_DIR,encoding='utf-8')
|
|
# --------- 读取config.ini配置文件 ---------------
|
|
login_url = con.get("WebURL","URL")
|
|
log = Log()
|
|
|
|
class Page(object):
|
|
"""
|
|
基础类，用于页面对象类的继承
|
|
"""
|
|
def __init__(self,selenium_driver,base_url=login_url,parent=None):
|
|
self.base_url = base_url
|
|
self.driver = selenium_driver
|
|
self.parent = parent
|
|
self.timeout = 10
|
|
|
|
def on_page(self):
|
|
"""
|
|
URL地址断言
|
|
:return: URL地址
|
|
"""
|
|
return self.driver.current_url == (self.base_url + self.url)
|
|
|
|
def _open(self,url):
|
|
"""
|
|
打开浏览器URL访问
|
|
:param url: URL地址
|
|
:return:
|
|
"""
|
|
url = self.base_url + url
|
|
self.driver.get(url)
|
|
assert self.on_page(),'Did not land on %s' % url
|
|
|
|
def open(self):
|
|
"""
|
|
内部调用_open私有函数
|
|
:return:
|
|
"""
|
|
self._open(self.url)
|
|
|
|
def find_element(self,*loc):
|
|
"""
|
|
单个元素定位
|
|
:param loc: 传入元素属性
|
|
:return: 定位到的元素
|
|
"""
|
|
try:
|
|
WebDriverWait(self.driver, 10).until(EC.visibility_of_element_located(loc))
|
|
return self.driver.find_element(*loc)
|
|
except:
|
|
log.error("{0}页面中未能找到{1}元素".format(self,loc))
|
|
|
|
def find_elements(self,*loc):
|
|
"""
|
|
多个元素定位
|
|
:param loc: 传入元素属性
|
|
:return: 定位到的元素
|
|
"""
|
|
try:
|
|
WebDriverWait(self.driver, 10).until(EC.visibility_of_element_located(loc))
|
|
return self.driver.find_elements(*loc)
|
|
except:
|
|
log.error("{0}页面中未能找到{1}元素".format(self,loc))
|
|
|
|
def script(self,src):
|
|
"""
|
|
提供调用JavaScript方法
|
|
:param src: 脚本文件
|
|
:return: JavaScript脚本
|
|
"""
|
|
return self.driver.execute_script(src)
|
|
|
|
# 重写定义send_keys方法
|
|
def send_key(self, loc, vaule, clear_first=True, click_first=True):
|
|
try:
|
|
loc = getattr(self, "_%s" % loc) # getattr相当于实现self.loc
|
|
if click_first:
|
|
self.find_element(*loc).click()
|
|
if clear_first:
|
|
self.find_element(*loc).clear()
|
|
self.find_element(*loc).send_keys(vaule)
|
|
except AttributeError:
|
|
log.error("%s 页面中未能找到 %s 元素" % (self, loc))
|
|
|
|
def switch_frame(self, loc):
|
|
"""
|
|
多表单嵌套切换
|
|
:param loc: 传元素的属性值
|
|
:return: 定位到的元素
|
|
"""
|
|
try:
|
|
return self.driver.switch_to_frame(loc)
|
|
except NoSuchFrameException as msg:
|
|
log.error("查找iframe异常-> {0}".format(msg))
|
|
|
|
def switch_windows(self,loc):
|
|
"""
|
|
多窗口切换
|
|
:param loc:
|
|
:return:
|
|
"""
|
|
try:
|
|
return self.driver.switch_to_window(loc)
|
|
except NoSuchWindowException as msg:
|
|
log.error("查找窗口句柄handle异常-> {0}".format(msg))
|
|
|
|
def switch_alert(self):
|
|
"""
|
|
警告框处理
|
|
:return:
|
|
"""
|
|
try:
|
|
return self.driver.switch_to_alert()
|
|
except NoAlertPresentException as msg:
|
|
log.error("查找alert弹出框异常-> {0}".format(msg))
|