class BaseFun:
def __init__(self, driver):
self.driver = driver
self.log = Log()
print(id(self.driver))
def find_element(self, loc, timeout=60):
try:
ele = WebDriverWait(self.driver, timeout).until(
EC.presence_of_element_located(loc))
return ele
except Exception as e:
self.log.error("获取元素时失败,失败原因__{}__".format(e))
def clear(self, loc):
self.find_element(loc).clear()
def send_keys(self, loc, text):
self.find_element(loc).send_keys(text)
def click(self, loc):
self.find_element(loc).click()
def get_text(self, loc):
return self.find_element(loc).text
def maximizewindow(self):
self.driver.maximize_window()
def get_url(self, url):
self.driver.get(url)
def wait(self, sec):
self.driver.implicitly_wait(sec)
def quit(self):
self.driver.quit()
class ReadFile:
# 初始化yaml地址
def __init__(self, file_path=None):
if file_path:
self.yaml_path = file_path
else:
self.yaml_path = "../yamlFile/desired_caps.yaml"
# self.mysql_yaml_path = os.path.join(os.path.dirname(os.path.dirname(__file__)), "config/mysql.yaml")
self.log = Log()
# 读取yaml文件
def read_yaml(self):
try:
file_path = self.yaml_path
# if path_type == "yaml_path":
# file_path = self.yaml_path
# elif path_type == "mysql_path":
# file_path = self.mysql_yaml_path
# else:
# print("文件类型输入错误")
"""
open与with open区别在于 open不关闭 需要手动f.close()
with open会自动关闭
"""
# f = open("file","r",encoding="utf-8")
# print(yaml.load(f.read()))
with open(file_path, "r", encoding="utf-8") as f:
return yaml.load(f.read(), Loader=yaml.FullLoader)
except Exception as e:
print("读取文件报错{}".format(e))
self.log.error("读取文件报错{}".format(e))
# 读取excel文件
def read_excel(self, sheet_name, function, casename=None, excel_path=None):
if excel_path:
excel_path = excel_path
else:
excel_path = self.excel_path
"""
读取excel
:param sheet_name:
:param function:
:param casename:
:return:
"""
try:
book = xlrd.open_workbook(excel_path)
data = book.sheet_by_name(sheet_name)
param = []
# print("列数:%s" % data.nrows)
for i in range(0, data.nrows):
get_func_name = data.row_values(i)[0]
get_run = data.row_values(i)[3]
get_cas_name = data.row_values(i)[1]
if casename is None:
if get_func_name == function and get_run == 1:
param.append(data.row_values(i))
else:
if get_func_name == function and get_cas_name == casename and get_run == 1:
param.append(data.row_values(i))
return param
except Exception as e:
print("读取excel报错{}".format(e))
