Files
nixpkgs/pkgs/build-support/fetchpypi/default.nix
2025-10-30 22:38:49 +02:00

68 lines
1.4 KiB
Nix

# `fetchPypi` function for fetching artifacts from PyPI.
{
lib,
fetchurl,
}:
let
computeUrl =
{
format ? "setuptools",
...
}@attrs:
let
computeWheelUrl =
{
pname,
version,
dist ? "py2.py3",
python ? "py2.py3",
abi ? "none",
platform ? "any",
}:
# Fetch a wheel. By default we fetch an universal wheel.
# See https://www.python.org/dev/peps/pep-0427/#file-name-convention for details regarding the optional arguments.
"https://files.pythonhosted.org/packages/${dist}/${builtins.substring 0 1 pname}/${pname}/${pname}-${version}-${python}-${abi}-${platform}.whl";
computeSourceUrl =
{
pname,
version,
extension ? "tar.gz",
}:
# Fetch a source tarball.
"mirror://pypi/${builtins.substring 0 1 pname}/${pname}/${pname}-${version}.${extension}";
compute = (
if format == "wheel" then
computeWheelUrl
else if format == "setuptools" then
computeSourceUrl
else
throw "Unsupported format ${format}"
);
in
compute (removeAttrs attrs [ "format" ]);
in
lib.makeOverridable (
{
format ? "setuptools",
sha256 ? "",
hash ? "",
...
}@attrs:
let
url = computeUrl (
removeAttrs attrs [
"sha256"
"hash"
]
);
in
fetchurl {
inherit url sha256 hash;
}
)