core/pkgs/by-name/py/python/wrapper.nix

70 lines
2.1 KiB
Nix
Raw Normal View History

2024-05-02 00:46:19 +00:00
{ lib, stdenv, buildEnv, makeBinaryWrapper
# manually pased
2024-05-13 21:24:10 +00:00
, python, requiredPythonModules
2024-05-02 00:46:19 +00:00
# extra opts
2024-05-13 21:24:10 +00:00
, extraLibs ? [ ], extraOutputsToInstall ? [ ], postBuild ? ""
, ignoreCollisions ? false, permitUserSite ? false
# Wrap executables with the given argument.
, makeWrapperArgs ? [ ], }:
2024-05-02 00:46:19 +00:00
# Create a python executable that knows about additional packages.
let
env = let
2024-05-13 21:24:10 +00:00
paths = requiredPythonModules (extraLibs ++ [ python ]);
2024-05-02 00:46:19 +00:00
pythonPath = "${placeholder "out"}/${python.sitePackages}";
pythonExecutable = "${placeholder "out"}/bin/${python.executable}";
in buildEnv {
name = "${python.name}-env";
inherit paths;
inherit ignoreCollisions;
extraOutputsToInstall = [ "out" ] ++ extraOutputsToInstall;
nativeBuildInputs = [ makeBinaryWrapper ];
postBuild = ''
if [ -L "$out/bin" ]; then
unlink "$out/bin"
fi
mkdir -p "$out/bin"
for path in ${lib.concatStringsSep " " paths}; do
if [ -d "$path/bin" ]; then
cd "$path/bin"
for prg in *; do
if [ -f "$prg" ]; then
rm -f "$out/bin/$prg"
if [ -x "$prg" ]; then
2024-05-13 21:24:10 +00:00
makeWrapper "$path/bin/$prg" "$out/bin/$prg" --set NIX_PYTHONPREFIX "$out" --set NIX_PYTHONEXECUTABLE ${pythonExecutable} --set NIX_PYTHONPATH ${pythonPath} ${
lib.optionalString (!permitUserSite)
''--set PYTHONNOUSERSITE "true"''
} ${lib.concatStringsSep " " makeWrapperArgs}
2024-05-02 00:46:19 +00:00
fi
fi
done
fi
done
'' + postBuild;
inherit (python) meta;
passthru = python.passthru // {
interpreter = "${env}/bin/${python.executable}";
inherit python;
env = stdenv.mkDerivation {
name = "interactive-${python.name}-environment";
nativeBuildInputs = [ env ];
buildCommand = ''
echo >&2 ""
echo >&2 "*** Python 'env' attributes are intended for interactive nix-shell sessions, not for building! ***"
echo >&2 ""
exit 1
'';
2024-05-13 21:24:10 +00:00
};
2024-05-02 00:46:19 +00:00
};
};
in env