调用函数:select_by_value 硒 Python

Calling function: select_by_value Selenium Python

我还是初学者...所以我在不同的文件中有以下 类 和功能:

def select_element_by_value(self, value, *selector):
    select = Select(self.find_element(*selector))
    select.select_by_value(value)

然后我在另一个文件中有这个:

class SiteList(HomePage):

SITE_LIST_SELECTOR = (By.ID, "site_label")

    def select_site(self):
        self.select_element_by_value("10", *self.SITE_LIST_SELECTOR)

最后我的脚本文件包含:

site_table = SiteList(driver=self.driver)
site_table.select_site()

它可以工作,但我希望能够 select 直接从脚本文件本身获取值(“10”),而不是像现在在其他文件的函数中那样。我该怎么做?

像这样?

class SiteList(HomePage):

SITE_LIST_SELECTOR = (By.ID, "site_label")

    def select_site(self, value):
        self.select_element_by_value(value, *self.SITE_LIST_SELECTOR)

site_table = SiteList(driver=self.driver)
site_table.select_site("10")