Use a special technique to remove git repos on Windows

Author: mdboom

bench_runner/git.py

@@ -14,6 +14,7 @@
 
 
 from .util import PathLike
+from . import util
 
 
 def get_log(
@@ -147,7 +148,7 @@ def clone(
         if is_hash and (dirname / ".git").is_dir() and get_git_hash(dirname) == branch:
             # This is a git repo, and the hash matches
             return
-        shutil.rmtree(dirname)
+        util.smart_rmtree(dirname)
 
     # Fetching a hash and fetching a branch require different approaches
 

bench_runner/util.py

@@ -5,6 +5,7 @@
 from pathlib import Path
 import shutil
 import subprocess
+import stat
 import sys
 from typing import Iterable, Iterator, Literal, TypeAlias, Union
 
@@ -105,6 +106,37 @@ def valid_version(version: str) -> bool:
         return False
 
 
+if sys.platform.startswith("win"):
+    if sys.version_info >= (3, 12):
+
+        def smart_rmtree(path: PathLike) -> None:
+            def onexc(func, path, exc):
+                # Is the error an access error?
+                if not os.access(path, os.W_OK):
+                    os.chmod(path, stat.S_IWUSR)
+                    func(path)
+                else:
+                    raise exc
+
+            shutil.rmtree(path, onexc=onexc)
+
+    else:
+
+        def smart_rmtree(path: PathLike) -> None:
+            def onerror(func, path, error):
+                # Is the error an access error?
+                if not os.access(path, os.W_OK):
+                    os.chmod(path, stat.S_IWUSR)
+                    func(path)
+                else:
+                    raise exc[1]
+
+            shutil.rmtree(path, onerror=onerror)
+
+else:
+    smart_rmtree = shutil.rmtree
+
+
 if os.getenv("GITHUB_ACTIONS") == "true":
 
     @contextlib.contextmanager