diff --git a/py/selenium/webdriver/remote/webdriver.py b/py/selenium/webdriver/remote/webdriver.py index 2dad5568df63c..d6bbe9c7e3e77 100644 --- a/py/selenium/webdriver/remote/webdriver.py +++ b/py/selenium/webdriver/remote/webdriver.py @@ -23,6 +23,7 @@ import types import typing import warnings +import zipfile from abc import ABCMeta from base64 import b64decode from base64 import urlsafe_b64encode @@ -1162,6 +1163,10 @@ def download_file(self, file_name: str, target_directory: str) -> None: with open(target_file, "wb") as file: file.write(base64.b64decode(contents)) + with zipfile.ZipFile(target_file, "r") as zip_ref: + zip_ref.extractall(target_directory) + + def delete_downloadable_files(self) -> None: """Deletes all downloadable files.""" if "se:downloadsEnabled" not in self.capabilities: