forked from auxolotl/core
29 lines
1.2 KiB
Nix
29 lines
1.2 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;
|
||
|
})
|