Skip to content

Commit

Permalink
PERF: faster pip install, when extracting a wheel, do not recheck/rec…
Browse files Browse the repository at this point in the history
…reate the parent directory before extracting each file
  • Loading branch information
rmmancom committed Jun 21, 2024
1 parent 66f4a5d commit ead576d
Showing 1 changed file with 8 additions and 6 deletions.
14 changes: 8 additions & 6 deletions src/pip/_internal/operations/install/wheel.py
Original file line number Diff line number Diff line change
Expand Up @@ -358,12 +358,6 @@ def _getinfo(self) -> ZipInfo:
return self._zip_file.getinfo(self.src_record_path)

def save(self) -> None:
# directory creation is lazy and after file filtering
# to ensure we don't install empty dirs; empty dirs can't be
# uninstalled.
parent_dir = os.path.dirname(self.dest_path)
ensure_dir(parent_dir)

# When we open the output file below, any existing file is truncated
# before we start writing the new contents. This is fine in most
# cases, but can cause a segfault if pip has loaded a shared
Expand Down Expand Up @@ -580,7 +574,15 @@ def is_entrypoint_wrapper(file: "File") -> bool:
script_scheme_files = map(ScriptFile, script_scheme_files)
files = chain(files, script_scheme_files)

existing_parents = set()
for file in files:
# directory creation is lazy and after file filtering
# to ensure we don't install empty dirs; empty dirs can't be
# uninstalled.
parent_dir = os.path.dirname(file.dest_path)
if parent_dir not in existing_parents:
ensure_dir(parent_dir)
existing_parents.add(parent_dir)
file.save()
record_installed(file.src_record_path, file.dest_path, file.changed)

Expand Down

0 comments on commit ead576d

Please sign in to comment.