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

Python selenium save screenshot in newly created folder

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

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

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

这是因为save_screenshot()调用不可达,截图前的函数returns。固定版本:

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