Python selenium 将屏幕截图保存在新创建的文件夹中

Question

我想创建一个文件夹，用于在测试失败时保存屏幕截图，如果仅在测试失败时创建该文件夹，那就太好了。

这目前不起作用，它没有将屏幕截图保存在文件夹内:

try:
    os.makedirs('./screenshots')
except OSError:
    pass

def check_exists_by_xpath(xpath):
    try:
        driver.find_element_by_xpath(xpath)
    except NoSuchElementException:
        return False
        driver.save_screenshot('screenshots/screenie.png')
    return True

Answer 1

这是因为 save_screenshot() 调用无法访问，该函数在进行屏幕截图之前返回。修复版本:

def check_exists_by_xpath(xpath):
    try:
        driver.find_element_by_xpath(xpath)
    except NoSuchElementException:
        driver.save_screenshot('screenshots/screenie.png')
        return False
    return True