Skip to content

Commit 8376c0a

Browse files
committed
add an integration test with 500pkgs
1 parent 79b2218 commit 8376c0a

File tree

5 files changed

+621
-93
lines changed

5 files changed

+621
-93
lines changed

python/private/pypi/whl_installer/wheel_deps.py

Lines changed: 16 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -236,20 +236,19 @@ def build(self) -> FrozenDeps:
236236
current_version = (
237237
f"{sys.version_info[0]}.{sys.version_info[1]}.{sys.version_info[2]}"
238238
)
239-
fname = Path(sys.argv[1])
240-
241-
metadata_contents = fname.read_text()
242-
metadata = Metadata.from_email(metadata_contents, validate=False)
243-
deps = Deps(
244-
name=metadata.name,
245-
requires_dist=[str(r) for r in metadata.requires_dist or []],
246-
).build()
247-
248-
print(
249-
json.dumps(
250-
{
251-
"deps": deps.deps,
252-
"version": current_version,
253-
}
254-
)
255-
)
239+
packages = Path(sys.argv[1]).read_text().split("\n")
240+
output = {
241+
"version": current_version,
242+
"deps": {},
243+
}
244+
for pkg in packages:
245+
metadata_contents = Path(f"{pkg}.METADATA").read_text()
246+
metadata = Metadata.from_email(metadata_contents, validate=False)
247+
deps = Deps(
248+
name=metadata.name,
249+
requires_dist=[str(r) for r in metadata.requires_dist or []],
250+
).build()
251+
output["deps"][metadata.name] = deps.deps
252+
253+
with Path(sys.argv[1] + ".out").open(mode="w") as f:
254+
json.dump(output, f, indent=" ", sort_keys=True)

0 commit comments

Comments
 (0)