python: remove LD_LIBRARY_PATH hack from running python environment

Fixes cachix#1111
vlaci · Oct 30, 2024 · 58ee89f · 58ee89f
1 parent 5c046ee
commit 58ee89f
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/src/modules/languages/python.nix b/src/modules/languages/python.nix
@@ -14,6 +14,9 @@ let
     python = cfg.package;
     requiredPythonModules = cfg.package.pkgs.requiredPythonModules;
     makeWrapperArgs = [
+      "--set"
+      "DEVENV_LD_LIBRARY_PATH_PREFIX"
+      libraries
       "--prefix"
       "LD_LIBRARY_PATH"
       ":"
@@ -39,6 +42,21 @@ let
     follows = [ "nixpkgs" ];
   };
 
+  pth_file =
+    let
+      exec_content = ''
+        ld_library_path_var = "DYLD_LIBRARY_PATH" if sys.platform == "darwin" else "LD_LIBRARY_PATH"
+        ld_library_path = os.environ.get(ld_library_path_var)
+        ld_library_path_prefix = os.environ.get("DEVENV_LD_LIBRARY_PATH_PREFIX")
+        if ld_library_path and ld_library_path_prefix:
+            if ld_library_path == ld_library_path_prefix:
+                del os.environ[ld_library_path_var]
+            else:
+                os.environ[ld_library_path_var] = ld_library_path.removeprefix(ld_library_path_prefix + ":")
+      '';
+    in
+    pkgs.writeText "devenv.pth" ''import os; exec("""${builtins.replaceStrings [ "\n" ] [ "\\n" ] exec_content}""")'';
+
   initVenvScript =
     let
       USE_UV_SYNC = cfg.uv.sync.enable && builtins.compareVersions cfg.uv.package.version "0.4.4" >= 0;
@@ -80,6 +98,7 @@ let
           ''
         }
         echo "${package.interpreter}" > "$VENV_PATH/.devenv_interpreter"
+        cp ${pth_file} "$VENV_PATH/${package.sitePackages}/devenv.pth"
       fi
 
       source "$VENV_PATH"/bin/activate
@@ -157,6 +176,7 @@ let
         if "''${UV_SYNC_COMMAND[@]}"
         then
           echo "$ACTUAL_UV_CHECKSUM" > "$UV_CHECKSUM_FILE"
+          cp ${pth_file} "$VENV_PATH/${package.sitePackages}/devenv.pth"
         else
           echo "uv sync failed. Run 'uv sync' manually." >&2
         fi
@@ -206,6 +226,7 @@ let
         if ''${POETRY_INSTALL_COMMAND[@]}
         then
           echo "$ACTUAL_POETRY_CHECKSUM" > "$POETRY_CHECKSUM_FILE"
+          cp ${pth_file} ".venv/${package.sitePackages}/devenv.pth"
         else
           echo "Poetry install failed. Run 'poetry install' manually."
         fi