Add a python module

1 files changed, 25 insertions, 0 deletions

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;
+  };
+}