如何在Python中正确处理大量（格式）常量？

Question

我正在 Python 中使用 Selenium 建立一个端到端测试框架。为了单击 DOM 元素，我必须定义许多定位器（如 XPath、CSS 选择器等），这些定位器大多是不变的。

但是，有一些常量在每次运行中都会有所不同，因此需要在运行时格式化某些内容。

目前我有一个基本的 locators.py 文件，它以视图对应的方式组织所有常量，如下所示：

class Locators:

    class LoginPage:
        USERNAME_INPUT = "//input[@id='username']"
        PASSWORD_INPUT = "//input[@id='password']"
        LOGIN_BUTTON = "//button[text()='Login']"
        # ...
    
    class Homepage:
        ACTION_BUTTON = lambda t: f"//buton[text()='{t}']"
        # ...

# ...

然后可以像这样使用：

from locators import Locators

browser.click(Locators.LoginPage.USERNAME_INPUT)
browser.click(Locators.Homepage.ACTION_BUTTON(var))

这里的最后一个示例（ACTION_BUTTON）是我最感兴趣的。由于例如 PyCharm 显示 PEP 8 警告，不鼓励使用 lambda，我想知道是否有更优雅的方法来处理此类“可格式化”常量？

此外，我希望能够将所有按钮概括为外部类，如下所示：

class Locators:

    _BUTTON = lambda s: f"//button[text()='{s}']"

    class LoginPage:
        USERNAME_INPUT = "//input[@id='username']"
        PASSWORD_INPUT = "//input[@id='password']"
        LOGIN_BUTTON = _BUTTON("Login")

    class Homepage:
        ACTION_BUTTON = _BUTTON

但这会产生语法错误。

Answer 1

我想这取决于个人风格。

就我个人而言，我会在此用例中使用常规格式字符串。

class Locators:

    _BUTTON = "//button[text()='{action}']"

    class LoginPage:
        USERNAME_INPUT = "//input[@id='username']"
        PASSWORD_INPUT = "//input[@id='password']"
        LOGIN_BUTTON = _BUTTON.format(action="Login")

    class Homepage:
        ACTION_BUTTON = _BUTTON

browser.click(Locators.LoginPage.USERNAME_INPUT)
browser.click(Locators.Homepage.ACTION_BUTTON.format(action=var))

Answer 2

我没有发现将辅助函数放在 Locators 命名空间之外有任何问题。毕竟，在执行之前它们不是定位器。

_BUTTON = lambda s: f"//button[text()='{s}']"

class Locators:

    class LoginPage:
        USERNAME_INPUT = "//input[@id='username']"
        PASSWORD_INPUT = "//input[@id='password']"
        LOGIN_BUTTON = _BUTTON("Login")

话虽这么说，我遇到了一个非常相似的问题。我通过为需要在定位器字符串中引用的变量创建一个单独的命名空间来解决这个问题。这允许我运行一些设置逻辑来确定在加载模块之前变量应该是什么。

# sys_params.py
class SystemParameters(object):
    homepage_arg1 = "default"

params = SystemParameters()
params.homepage_arg1 = "something"

# locators.py
from sys_params import params

class Locators:
    class HomePage:
        ACTION_BUTTON = f'//button[text()="{params.homepage_arg1}"]'

如果您需要模块更动态地加载（就像我所做的那样），那么您将面临必须在函数内部限定字符串范围的问题。

# locators.py
from sys_params import params

class Locators:
    class HomePage:
        @classmethod
        def ACTION_BUTTON(cls): return f'//button[text()="{params.homepage_arg1}"]'

## which gets used slightly differently
driver.click(Locators.HomePage.ACTION_BUTTON())