如何在python列表上强制特定类型

Question

假设我有一个python类

class cls:
      x=5

我想创建一个列表，它只能包含cls类型的对象，比如

a = [obj1,obj2,obj3....]

其中所有对象必须为cls类型

我怎样才能强制执行呢

此外，我想谈谈提出这个问题的原因

在selenium中，我们有一个名为WebElement的类，现在一个方法findElements_by_xpath（xpath）返回WebElemnet的列表。所以如果我像下面这样做

elementlist = driver.findElements_by_xpath(xpath)

现在我将这个元素列表传递给function（元素列表）并做类似的事情

def fun(elementlist):
    for element in elementlist
       element.click()

虽然这可以很好地工作，但由于fun中的elementlist变量的类型为any，pycharm intellisence不起作用，并且没有给出与元素上可操作的方法相关的usggestion——当然，如果类型为webElement，则会给出usggestion

Answer 1

def fun(elementlist:List[WebElement]):
    for element in elementlist:
        element..click()

尝试定义类型，您需要的导入是：

from selenium.webdriver.remote.webelement import WebElement