From aee430ee47e1e80e4d555722e3d6ba1bf7899c00 Mon Sep 17 00:00:00 2001
From: Benjamin Sparks <b.sparks@alugha.com>
Date: Wed, 21 May 2025 16:02:39 +0200
Subject: [PATCH] python3Packages.nltk.dataDir: init

Add helper function to install NLTK data packages like `wordnet` and `punkt-tab`.
Should be used in conjunction with `makeWrapperArgs`,
e.g. `makeWrapperArgs = [ "--set" "NLTK_DATA" "${nltk.dataDir(d: [ d.wordnet d.punkt-tab ])}" ];`
---
 pkgs/development/python-modules/nltk/data-dir.nix | 15 +++++++++++++++
 pkgs/development/python-modules/nltk/default.nix  |  1 +
 2 files changed, 16 insertions(+)
 create mode 100644 pkgs/development/python-modules/nltk/data-dir.nix

diff --git a/pkgs/development/python-modules/nltk/data-dir.nix b/pkgs/development/python-modules/nltk/data-dir.nix
new file mode 100644
index 000000000000..d18ab730597b
--- /dev/null
+++ b/pkgs/development/python-modules/nltk/data-dir.nix
@@ -0,0 +1,15 @@
+{
+  lib,
+  pkgs,
+  python3Packages,
+}:
+lib.makeOverridable (
+  { ... }@nltkDataPkgs:
+  f:
+  pkgs.symlinkJoin {
+    inherit (python3Packages.nltk) meta;
+    name = "nltk-data-dir";
+
+    paths = f nltkDataPkgs;
+  }
+) python3Packages.nltk.data
diff --git a/pkgs/development/python-modules/nltk/default.nix b/pkgs/development/python-modules/nltk/default.nix
index 55831f7708ae..bdb79081ec81 100644
--- a/pkgs/development/python-modules/nltk/default.nix
+++ b/pkgs/development/python-modules/nltk/default.nix
@@ -41,6 +41,7 @@ buildPythonPackage rec {
 
   passthru = {
     data = pkgs.nltk-data;
+    dataDir = pkgs.callPackage ./data-dir.nix { };
   };
 
   meta = with lib; {