基于YAML配置的Python动态页面对象生成机制
在使用Selenium进行Web自动化测试时,通常需要为每个页面创建对应的PageObject类。传统做法如下所示:
class BaiduPageObject(object):
def __init__(self, driver):
self.driver = driver
# 元素定义
search_ipt_loc = ('id', 'kw')
search_btn_loc = ('id', 'su')
# 操作方法
def input_search_ipt(self, keyword):
search_ipt = self.driver.find_element(*search_ipt_loc)
search_ipt.clear()
search_ipt.send_keys(keyword)
def click_search_btn(self):
self.driver.find_element(*search_btn_loc).click()
def perform_search(self, keyword):
self.input_search_ipt(keyword)
self.click_search_btn()
对于大型项目而言,维护大量结构类似的PageObject类会带来重复性工作。为了提升效率,可以通过配置驱动的方式来自动生成这些页面对象。
设计方案
采用YAML格式描述页面信息,并通过工厂模式动态构建页面对象。配置文件示例如下:
# baidu.yaml
url: https://www.baidu.com
title: 百度一下,你就知道
elements:
search_input: [id, kw]
submit_button: [id, su]
actions:
execute_search:
input: [search_input, placeholder_text]
click: submit_button
核心实现
# page_factory.py
from types import MethodType
from collections import OrderedDict
import yaml
from selenium import webdriver
class DynamicPage(object):
def __init__(self, driver, config_file=None, **kwargs):
self.driver = driver
self.page_url = kwargs.get('url', None)
self.page_title = kwargs.get('title', None)
self.locators = kwargs.get('elements', {})
self.behaviors = kwargs.get('actions', OrderedDict())
if config_file:
self.load_config(config_file)
def load_config(self, file_path):
with open(file_path, 'r', encoding='utf-8') as f:
config_data = yaml.safe_load(f)
self.page_url = config_data.get('url')
self.page_title = config_data.get('title')
self.locators = config_data.get('elements', {})
self.behaviors = OrderedDict(config_data.get('actions', {}))
def navigate(self):
if self.page_url:
self.driver.get(self.page_url)
def verify_page(self):
if self.page_title:
return self.page_title in self.driver.title
return True
def enter_text(self, element_key, content):
target_element = getattr(self, element_key)
target_element.clear()
target_element.send_keys(content)
def trigger_click(self, element_key):
target_element = getattr(self, element_key)
target_element.click()
def process_locator(self, key):
locator_info = self.locators.get(key)
if locator_info:
found_element = self.driver.find_element(*locator_info)
setattr(DynamicPage, key, found_element)
return found_element
def process_behavior(self, action_name):
behavior_steps = self.behaviors.get(action_name)
if behavior_steps:
def dynamic_method(self, input_value=None):
for operation, parameters in behavior_steps.items():
if operation == 'input':
if input_value and isinstance(parameters, list) and len(parameters) > 1:
parameters[1] = input_value
self.enter_text(*parameters)
elif operation == 'click':
self.trigger_click(parameters)
return MethodType(dynamic_method, self)
def __getattr__(self, attribute_name):
if not hasattr(DynamicPage, attribute_name):
return self.process_locator(attribute_name) or self.process_behavior(attribute_name)
# 使用示例
if __name__ == '__main__':
browser = webdriver.Chrome()
baidu_page = DynamicPage(browser, 'baidu.yaml')
baidu_page.navigate()
baidu_page.execute_search('搜索关键词')
这种方式虽然没有完全消除页面描述的工作量,但将原本的编码任务转换为配置管理,具备以下优势:
- 实现了界面元素与业务逻辑的解耦
- 支持多种数据源(文件、数据库、接口)驱动页面生成
- 便于与其他系统集成,如可视化配置平台或自动抓取工具