Selenium如果參數爲true，請單擊按鈕

Question

我想創建一個帶參數的函數，如果參數爲True，則該按鈕將被單擊否則不會。我可以使用這個嗎？

def buttonClick(self, Button): 
    if Button == True: 
     self.driver.find_element_by_id('button').click 

Answer 1

兩個主要的東西從我的頭頂修復：

• 你可以避免== True部分
• 你是不是調用click方法 - 添加()

固定版本：

def buttonClick(self, should_click_button): 
    if should_click_button: 
     self.driver.find_element_by_id('button').click() 

使用範例：

instance = MyClass() 
instance.buttonClick(True) 
instance.buttonClick(False) 

您還可以設置的默認值參數：

def buttonClick(self, should_click_button=False): 
    if should_click_button: 
     self.driver.find_element_by_id('button').click() 

現在，如果你並不需要點擊按鈕，根本不通過參數：

instance = MyClass() 
instance.buttonClick(True) 
instance.buttonClick()