# `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; }
)