core/pkgs/by-name/py/python/python-packages-base.nix

{ pkgs, stdenv, lib, python }:

self:

let
  inherit (self) callPackage;

  namePrefix = python.libPrefix + "-";

  # Derivations built with `buildPythonPackage` can already be overridden with `override`, `overrideAttrs`, and `overrideDerivation`.
  # This function introduces `overridePythonAttrs` and it overrides the call to `buildPythonPackage`.
  makeOverridablePythonPackage = f:
    lib.mirrorFunctionArgs f (origArgs:
      let
        args = lib.fix (lib.extends (_: previousAttrs: {
          passthru = (previousAttrs.passthru or { }) // {
            overridePythonAttrs = newArgs:
              makeOverridablePythonPackage f (overrideWith newArgs);
          };
        }) (_: origArgs));
        result = f args;
        overrideWith = newArgs:
          args
          // (if pkgs.lib.isFunction newArgs then newArgs args else newArgs);
      in if builtins.isAttrs result then
        result
      else if builtins.isFunction result then {
        overridePythonAttrs = newArgs:
          makeOverridablePythonPackage f (overrideWith newArgs);
        __functor = self: result;
      } else
        result);

  mkPythonDerivation = if python.isPy3k then
    ./mk-python-derivation.nix
  else
    ./python2/mk-python-derivation.nix;

  buildPythonPackage = makeOverridablePythonPackage (lib.makeOverridable
    (callPackage mkPythonDerivation {
      inherit
        namePrefix; # We want Python libraries to be named like e.g. "python3.6-${name}"
      inherit toPythonModule; # Libraries provide modules
    }));

  buildPythonApplication = makeOverridablePythonPackage (lib.makeOverridable
    (callPackage mkPythonDerivation {
      namePrefix = ""; # Python applications should not have any prefix
      toPythonModule = x: x; # Application does not provide modules.
    }));

  # Check whether a derivation provides a Python module.
  hasPythonModule = drv: drv ? pythonModule && drv.pythonModule == python;

  # Get list of required Python modules given a list of derivations.
  requiredPythonModules = drvs:
    let modules = lib.filter hasPythonModule drvs;
    in lib.unique ([ python ] ++ modules
      ++ lib.concatLists (lib.catAttrs "requiredPythonModules" modules));

  # Create a PYTHONPATH from a list of derivations. This function recurses into the items to find derivations
  # providing Python modules.
  makePythonPath = drvs:
    lib.makeSearchPath python.sitePackages (requiredPythonModules drvs);

  removePythonPrefix = lib.removePrefix namePrefix;

  # Convert derivation to a Python module.
  toPythonModule = drv:
    drv.overrideAttrs (oldAttrs: {
      # Use passthru in order to prevent rebuilds when possible.
      passthru = (oldAttrs.passthru or { }) // {
        pythonModule = python;
        pythonPath = [ ]; # Deprecated, for compatibility.
        requiredPythonModules = requiredPythonModules drv.propagatedBuildInputs;
      };
    });

  # Convert a Python library to an application.
  toPythonApplication = drv:
    drv.overrideAttrs (oldAttrs: {
      passthru = (oldAttrs.passthru or { }) // {
        # Remove Python prefix from name so we have a "normal" name.
        # While the prefix shows up in the store path, it won't be
        # used by `nix-env`.
        name = removePythonPrefix oldAttrs.name;
        pythonModule = false;
      };
    });

  disabled = drv:
    throw "${
      removePythonPrefix (drv.pname or drv.name)
    } not supported for interpreter ${python.executable}";

  disabledIf = x: drv: if x then disabled drv else drv;

in {
  inherit lib pkgs stdenv;
  inherit (python.passthru)
    isPy27 isPy37 isPy38 isPy39 isPy310 isPy311 isPy3k isPyPy pythonAtLeast
    pythonOlder;
  inherit buildPythonPackage buildPythonApplication;
  inherit hasPythonModule requiredPythonModules makePythonPath disabled
    disabledIf;
  inherit toPythonModule toPythonApplication;

  python = toPythonModule python;
  # Dont take pythonPackages from "global" pkgs scope to avoid mixing python versions
  pythonPackages = self;
}