如何在Pytest+Selenium的POM框架中存储大量元素定位器？

针对你用POM模式管理200+元素定位器的场景，以下是几种实用且可维护的方案：

1. 页面对象类属性存储（最常用） #

将定位器作为页面对象类的类属性，集中定义在类的顶部，配合BasePage中的通用元素操作方法使用，避免每个定位器写单独的获取方法。

示例代码： #

BasePage（封装通用操作）

from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

class BasePage:
    def __init__(self, driver):
        self.driver = driver
        self.wait = WebDriverWait(driver, 10)

    def find_element(self, locator):
        return self.wait.until(EC.presence_of_element_located(locator))

    def send_keys(self, locator, text):
        self.find_element(locator).send_keys(text)

    def click(self, locator):
        self.find_element(locator).click()

具体页面对象（比如LoginPage）

from selenium.webdriver.common.by import By
from base_page import BasePage

class LoginPage(BasePage):
    # 所有定位器集中定义在类顶部，一目了然
    CURRENT_EMAIL_INPUT = (By.CSS_SELECTOR, "input[id='current-email']")
    PASSWORD_INPUT = (By.ID, "password")
    SUBMIT_BUTTON = (By.XPATH, "//button[@type='submit']")

    def __init__(self, driver):
        super().__init__(driver)

    def enter_current_email(self, email):
        self.send_keys(self.CURRENT_EMAIL_INPUT, email)

    def click_submit(self):
        self.click(self.SUBMIT_BUTTON)

优点：定位器与页面逻辑绑定紧密，修改时无需跨文件查找；代码复用率高，通用操作方法一次定义多次使用。

2. 独立配置文件+字典存储（适合超大量定位器） #

如果定位器数量极多，可将其按页面/模块分组，放在单独的配置文件（如locators.py）中用字典存储，实现定位器与页面对象逻辑分离。

示例代码： #

locators.py（单独存放所有定位器）

from selenium.webdriver.common.by import By

# 按页面分组
LOGIN_PAGE = {
    "current_email": (By.CSS_SELECTOR, "input[id='current-email']"),
    "password": (By.ID, "password"),
    "submit_btn": (By.XPATH, "//button[@type='submit']")
}

HOME_PAGE = {
    "user_menu": (By.CLASS_NAME, "user-dropdown"),
    "logout_link": (By.LINK_TEXT, "Logout")
}

页面对象中使用

from locators import LOGIN_PAGE
from base_page import BasePage

class LoginPage(BasePage):
    def __init__(self, driver):
        super().__init__(driver)
        self.locators = LOGIN_PAGE

    def enter_current_email(self, email):
        self.send_keys(self.locators["current_email"], email)

优点：定位器集中管理，多个页面对象可共享同一配置；按模块分组后，查找和修改定位器更高效。

3. 枚举类存储（类型安全，减少拼写错误） #

利用Python的Enum类定义定位器，借助IDE的类型提示避免字典key拼写错误，适合对代码严谨性要求高的场景。

示例代码： #

locators.py（用枚举类定义）

from enum import Enum
from selenium.webdriver.common.by import By

class LoginPageLocators(Enum):
    CURRENT_EMAIL = (By.CSS_SELECTOR, "input[id='current-email']")
    PASSWORD = (By.ID, "password")
    SUBMIT_BUTTON = (By.XPATH, "//button[@type='submit']")

页面对象中使用

from locators import LoginPageLocators
from base_page import BasePage

class LoginPage(BasePage):
    def enter_current_email(self, email):
        # 通过.value获取定位器元组
        self.send_keys(LoginPageLocators.CURRENT_EMAIL.value, email)

优点：具备类型检查和自动补全，降低人为拼写错误；枚举类本身是结构化的，定位器分类更清晰。

方案选择建议 #

• 定位器数量适中（几十到上百）：优先用类属性存储，简单直接易维护；
• 定位器数量超200+或需跨页面共享：选独立字典配置文件；
• 追求代码严谨性、减少错误：用枚举类存储。

内容的提问来源于stack exchange，提问作者Hubert Łuczak