From dc16380cfa76f507a062f40b014a62ea3fdc9135 Mon Sep 17 00:00:00 2001 From: Nick Crews Date: Fri, 7 Oct 2022 17:38:09 -0800 Subject: [PATCH] Make `get_screenshot_as_file()` work with pathlib.Paths It currently expects a string (or something with .lower()), but accepting Paths seems quite reasonable. --- py/selenium/webdriver/remote/webdriver.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/py/selenium/webdriver/remote/webdriver.py b/py/selenium/webdriver/remote/webdriver.py index 6e0113cd9e4a2..6684278a0d84a 100644 --- a/py/selenium/webdriver/remote/webdriver.py +++ b/py/selenium/webdriver/remote/webdriver.py @@ -920,7 +920,7 @@ def get_screenshot_as_file(self, filename) -> bool: driver.get_screenshot_as_file('/Screenshots/foo.png') """ - if not filename.lower().endswith(".png"): + if not str(filename).lower().endswith(".png"): warnings.warn( "name used for saved screenshot does not match file " "type. It should end with a `.png` extension", UserWarning,