34 lines
1.3 KiB
Nix
34 lines
1.3 KiB
Nix
# `fetchPypi` function for fetching artifacts from PyPI.
|
|
{ fetchurl, makeOverridable }:
|
|
|
|
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 (builtins.removeAttrs attrs [ "format" ]);
|
|
|
|
in makeOverridable
|
|
({ format ? "setuptools", sha256 ? "", hash ? "", ... }@attrs:
|
|
let url = computeUrl (builtins.removeAttrs attrs [ "sha256" "hash" ]);
|
|
in fetchurl { inherit url sha256 hash; })
|