Add a python module

2 files changed, 26 insertions, 0 deletions

diff --git a/modules/base.nix b/modules/base.nix
index 1ebc61a..0f1218e 100644
--- a/modules/base.nix
+++ b/modules/base.nix
@@ -11,6 +11,7 @@
   imports = [
     ./host-common.nix
     ./dev-tools.nix
+    ./python.nix
     ./shell.nix
     ./term.nix
   ];
diff --git a/modules/python.nix b/modules/python.nix
new file mode 100644
index 0000000..56eadb3
--- /dev/null
+++ b/modules/python.nix
@@ -0,0 +1,25 @@
+{
+  config,
+  pkgs,
+  lib,
+  userOptions,
+  ...
+}:
+let
+  pythonVersion = lib.replaceStrings [ "." ] [ "" ] userOptions.pythonVersion;
+  python = pkgs."python${pythonVersion}";
+  pythonPackages = pkgs."python${pythonVersion}Packages";
+in
+{
+  home.packages = with pkgs; [
+    python
+    pipenv
+    pythonPackages.virtualenv
+    pythonPackages.uv
+  ];
+
+  programs.pyenv = {
+    enable = true;
+    enableZshIntegration = true;
+  };
+}