summaryrefslogtreecommitdiff
path: root/pkgs/development/interpreters/python/wrapper.nix
blob: 914e4b354c49a956beb669d8d491342869da395f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
{
  lib,
  stdenv,
  buildEnv,
  runCommand,
  makeBinaryWrapper,

  # manually pased
  python,
  requiredPythonModules,

  # extra opts
  extraLibs ? [ ],
  extraOutputsToInstall ? [ ],
  postBuild ? "",
  ignoreCollisions ? false,
  permitUserSite ? false,
  # Wrap executables with the given argument.
  makeWrapperArgs ? [ ],
}:

# Create a python executable that knows about additional packages.
let
  env =
    let
      paths = requiredPythonModules (extraLibs ++ [ python ]) ++ [
        (runCommand "bin" { } ''
          mkdir -p $out/bin
        '')
      ];
      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 = ''
        for path in ${lib.concatStringsSep " " paths}; do
          if [ -d "$path/bin" ]; then
            cd "$path/bin"
            for prg in *; do
              if [ -f "$prg" ] && [ -x "$prg" ]; then
                rm -f "$out/bin/$prg"
                if [ "$prg" = "${python.executable}" ]; then
                  makeWrapper "${python.interpreter}" "$out/bin/$prg" \
                    --inherit-argv0 \
                    --resolve-argv0 \
                    ${lib.optionalString (!permitUserSite) ''--set PYTHONNOUSERSITE "true"''} \
                    ${lib.concatStringsSep " " makeWrapperArgs}
                elif [ "$(readlink "$prg")" = "${python.executable}" ]; then
                  ln -s "${python.executable}" "$out/bin/$prg"
                else
                  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}
                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
          '';
        };
      };
    };
in
env