* buildPythonPackage: added an argument `pythonPath' to specify Python

dependencies that are *not* propagated to the user environment
  (as opposed to `propagatedBuildInputs').  For instance, if you
  install `iotop', you typically don't want its Python dependencies
  polluting the user environment.
  
* buildPythonPackage: some cleanup (e.g. use function argument
  defaults instead of `if attrs ? foo then attrs.foo else []').

svn path=/nixpkgs/branches/modular-python/; revision=26571

pkgs/development/python-modules/generic/default.nix

@@ -5,46 +5,40 @@
 
 { python, setuptools, makeWrapper, lib }:
 
-{ name, namePrefix ? "python-", src, meta, patches ? []
+{ name, namePrefix ? "python-"
-, installCommand ? ""
+
+, buildInputs ? []
+
+, # List of packages that should be added to the PYTHONPATH
+  # environment variable in programs built by this function.  Packages
+  # in the standard `propagatedBuildInputs' variable are also added.
+  # The difference is that `pythonPath' is not propagated to the user
+  # environment.  This is preferrable for programs because it doesn't
+  # pollute the user environment.
+  pythonPath ? []
+
+, installCommand ?
+    ''
+      easy_install --prefix="$out" .
+    ''
+    
+, buildPhase ? "true"
+
 , doCheck ? true, checkPhase ? "python setup.py test"
+
 , postInstall ? ""
+
 , ... } @ attrs:
 
-let
+# Keep extra attributes from ATTR, e.g., `patchPhase', etc.
-    defaultInstallCommand = ''easy_install --prefix="$out" .'';
+python.stdenv.mkDerivation (attrs // {
+  inherit doCheck buildPhase checkPhase;
 
-    # Return the list of recursively propagated build inputs of PKG.
+  name = namePrefix + name;
-    recursiveBuildInputs =
-      pkg:
-        [ pkg ] ++
-        (if pkg ? propagatedBuildNativeInputs
-         then lib.concatLists (map recursiveBuildInputs
-                                   pkg.propagatedBuildNativeInputs)
-         else []);
 
-in
+  buildInputs = [ python makeWrapper setuptools ] ++ buildInputs ++ pythonPath;
 
-python.stdenv.mkDerivation (
+  pythonPath = [ setuptools] ++ pythonPath;
-  # Keep extra attributes from ATTR, e.g., `patchPhase', etc.
-  attrs
-
-  //
-
-  (rec {
-  inherit src meta patches doCheck checkPhase;
-
-  name = namePrefix + attrs.name;
-
-  buildInputs = [ python setuptools makeWrapper ] ++
-    (if attrs ? buildInputs then attrs.buildInputs else []);
-
-  propagatedBuildInputs = [ setuptools ] ++
-    (if attrs ? propagatedBuildInputs
-     then attrs.propagatedBuildInputs
-     else []);
-
-  buildPhase = "true";
 
   # XXX: Should we run `easy_install --always-unzip'?  It doesn't seem
   # to have a noticeable impact on small scripts.
@@ -53,42 +47,53 @@ python.stdenv.mkDerivation (
 
     echo "installing \`${name}' with \`easy_install'..."
     export PYTHONPATH="$out/lib/${python.libPrefix}/site-packages:$PYTHONPATH"
-    ${if installCommand == "" then defaultInstallCommand else installCommand}
+    ${installCommand}
 
     ${postInstall}
   '';
 
-  postFixup = ''
+  postFixup =
-    # Wrap scripts that are under `{s,}bin/' so that they get the right
+    ''
-    # $PYTHONPATH.
+      declare -A pythonPathsSeen
-    for i in "$out/bin/"* "$out/sbin/"*
-    do
-      if head -n1 "$i" | grep -q "${python}"
-      then
-          echo "wrapping \`$i'..."
     
-          # Compute a $PATH prefix for the program.
+      addToPythonPath() {
-          program_PATH=""
+          local dir="$1"
-          ${lib.concatStrings
+          if [ -n "''${pythonPathsSeen[$dir]}" ]; then return; fi
-            (map (path:
+          pythonPathsSeen[$dir]=1
-                  ''if [ -d "${path}/bin" ]
+          addToSearchPath program_PYTHONPATH $dir/lib/${python.libPrefix}/site-packages
-                    then
+          addToSearchPath program_PATH $dir/bin
-                        program_PATH="${path}/bin'' + "\$" + ''{program_PATH:+:}$program_PATH"
+          local prop="$dir/nix-support/propagated-build-native-inputs"
+          if [ -e $prop ]; then
+              local i
+              for i in $(cat $prop); do
+                  addToPythonPath $i
+              done
           fi
-                   '')
+      }
-                 (lib.concatMap recursiveBuildInputs propagatedBuildInputs))}
     
+      wrapPythonPrograms() {
+          local dir="$1"
+          local pythonPath="$2"
+          local i
+
+          pythonPathsSeen=()
+          program_PYTHONPATH=
+          program_PATH=
+          for i in $pythonPath; do
+              addToPythonPath $i
+          done
+
+          for i in $(find "$out" -type f -perm +0100); do
+              if head -n1 "$i" | grep -q "${python}"; then
+                  echo "wrapping \`$i'..."
                   wrapProgram "$i" \
-            --prefix PYTHONPATH ":"                 \
+                    --prefix PYTHONPATH ":" $program_PYTHONPATH \
-            ${lib.concatStringsSep ":"
+                    --prefix PATH ":" $program_PATH
-               ([ "$out/lib/${python.libPrefix}/site-packages" ] ++
-                (map (path: path + "/lib/${python.libPrefix}/site-packages")
-                     (lib.concatMap recursiveBuildInputs
-                                    propagatedBuildInputs)))} \
-            --prefix PATH ":" "$program_PATH"
-
               fi
           done
+      }
+
+      wrapPythonPrograms $out "$out $pythonPath"
 
       # If a user installs a Python package, she probably also wants its
       # dependencies in the user environment (since Python modules don't
@@ -98,4 +103,4 @@ python.stdenv.mkDerivation (
           ln -s $out/nix-support/propagated-build-inputs $out/nix-support/propagated-user-env-packages
       fi
     '';
-}))
+})

pkgs/os-specific/linux/iotop/default.nix

@@ -9,7 +9,7 @@ buildPythonPackage rec {
     sha256 = "1dfvw3khr2rvqllvs9wad9ca3ld4i7szqf0ibq87rn36ickrf3ll";
   };
 
-  propagatedBuildInputs = [ pythonPackages.curses ];
+  pythonPath = [ pythonPackages.curses ];
 
   doCheck = false;