python-flake/flake.nix at main · justinvson-pd/python-flake · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
{
  inputs = {
    nixpkgs.url = "github:NixOS/nixpkgs/nixos-unstable";
    flake-utils.url = "github:numtide/flake-utils";
  };
  outputs = { self, nixpkgs, flake-utils }:
    flake-utils.lib.eachDefaultSystem
      (system:
        let
          pkgs = import nixpkgs {
            inherit system;
          };

          pythonPackages = pkgs.python311Packages;

          venvDir = "./env";

          runPackages = with pkgs; [
            pythonPackages.python
            pythonPackages.venvShellHook
          ];

          devPackages = with pkgs; runPackages ++ [
            pythonPackages.pylint
            pythonPackages.flake8
            pythonPackages.black
          ];

          postHook = ''
            PYTHONPATH=\$PWD/\${venvDir}/\${pythonPackages.python.sitePackages}/:\$PYTHONPATH
          '';
        in
        with pkgs;
        {
          devShells.default = mkShell {
            inherit venvDir;
            buildInputs = [ devPackages ];
            postShellHook = postHook;
          };
        }
      );
}