[py] fix bug where downloaded file was not getting unzipped

Author: titusfortner

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: