使用带有 selenium 的 pom 框架的页面类的 Pycharm Intellisense

Question

嘿，我正在尝试解决这个问题，但遇到了一些麻烦。基本上我正在尝试用我的 selenium 实现页面对象模型框架。我创建了一个页面类，我将我的驱动程序对象传递给下面的内容：

from selenium import webdriver

class LoginPage:

   def __init__(self, driver):

       self.driver = driver

       self.email_field_xpath = self.driver.find_element_by_xpath('/html/body/div[3]/div[4]/div/div[4]/div/form/div['
                                                                  '1]/input')

       self.password_field_xpath = self.driver.find_element_by_xpath('/html/body/div[3]/div[4]/div/div['
                                                                     '4]/div/form/div[''2]/input')

       self.submit_button_xpath = self.driver.find_element_by_xpath('/html/body/div[3]/div[4]/div/div[4]'
                                                                    '/div/form/button')

   def send_text_to_email_field(self):

       self.email_field_xpath.clear()

虽然代码在我运行主要单元测试时有效，但我似乎无法让 Intellisense 在此类中工作。我知道类不知道驱动程序对象是什么，所以它可能不知道建议什么。有没有办法让班级知道驱动程序的类型是让Intellisense开始工作？

例如输入“self.driver.find”会提示“find_element_by_id”等

谢谢

Answer 1

我也面临同样的问题。有人可以回答这个问题吗？