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