Как я могу добавить save_screenshot к оператору If?

Question

Я хочу сделать скриншот, если кнопка входа отключена или нет.Где я могу добавить save_screenshot, чтобы сделать это?

def is_login_button_disabled(self):
        self.disabled_login_button = self.browser.find_element(*self.login_button_disabled)
        if self.disabled_login_button.is_displayed():
            print("Login Button Disabled")
        else:
            print("Login Button Enabled")

Спасибо.

Answer 1

Просто добавьте одну строку, например:

def is_login_button_disabled(self):
    self.disabled_login_button = self.browser.find_element(*self.login_button_disabled)
    if self.disabled_login_button.is_displayed():
        # take screen shot
        self.browser.save_screenshot("screenshot.png")
        print("Login Button Disabled")
    else:
        print("Login Button Enabled")

Редактировать:

Вы должны взять дополнительную переменную, такую ​​как image = "image" и count=0, которая увеличивается каждый раз после того, как снимок экранавзял

image_screenshot = image + str(count) + ".png"
self.browser.save_screenshot(image_screenshot)
count+=1

Примечание:

Определить изображение и считать в качестве глобальной переменной или передать в качестве параметра при вызове функции и определить в основной функции, например:

def main():
   image = "image"
   count = 0
   self.is_login_button_disabled(image,count)