Bug Fix Issue #4603 Access problem

mlflow/projects/docker.py

@@ -1,5 +1,6 @@
 import logging
 import os
+import stat
 import posixpath
 import shutil
 import tempfile
@@ -99,6 +100,30 @@ def _get_docker_image_uri(repository_uri, work_dir):
     return repository_uri + version_string
 
 
+def handle_readonly(func, path, exc_info):
+    """
+    Clear the readonly bit and reattempt the removal.
+
+    References:
+    - https://bugs.python.org/issue19643
+    - https://bugs.python.org/issue43657
+    """
+    exc_class, exc_instance = exc_info[:2]
+    should_reattempt = (
+        func in (os.unlink, os.rmdir)
+        and issubclass(exc_class, PermissionError)
+        and (
+            # Check whether the error code indicates "Permission Denied"
+            (os.name != "nt" and exc_instance.errno == 13)
+            or (os.name == "nt" and exc_instance.winerror == 5)
+        )
+    )
+    if not should_reattempt:
+        raise exc_instance
+    os.chmod(path, stat.S_IWRITE)
+    func(path)
+
+
 def _create_docker_build_ctx(work_dir, dockerfile_contents):
     """
     Creates build context tarfile containing Dockerfile and project code, returning path to tarfile
@@ -114,7 +139,7 @@ def _create_docker_build_ctx(work_dir, dockerfile_contents):
             output_filename=result_path, source_dir=dst_path, archive_name=_PROJECT_TAR_ARCHIVE_NAME
         )
     finally:
-        shutil.rmtree(directory)
+        shutil.rmtree(directory, onerror=handle_readonly)
     return result_path