1 files changed, 17 insertions, 13 deletions

diff --git a/tools/sops-inventory/default.nix b/tools/sops-inventory/default.nix
index d6a3d0c4..0f868629 100644
--- a/tools/sops-inventory/default.nix
+++ b/tools/sops-inventory/default.nix
@@ -1,20 +1,24 @@
-{ system, self, ... }:
+{ system, self, uv2nix, gitignore, ... }:
 
 let
   pkgs = self.legacyPackages.${system};
-in
-with pkgs.poetry2nix;
-mkPoetryApplication {
-  projectDir = cleanPythonSources { src = ./.; };
-
-  nativeBuildInputs = with pkgs; [ makeWrapper ];
-
-  overrides = overrides.withDefaults (self: super: {
-    pyyaml = super.pyyaml.override { preferWheel = true; };
-  });
-
+  workspace = uv2nix.lib.workspace.loadWorkspace {
+    workspaceRoot = gitignore.lib.gitignoreSource ./.;
+  };
+  pythonSet = self.lib.pythonSet {
+    inherit pkgs;
+    python = pkgs.python3;
+    overlay = workspace.mkPyprojectOverlay {
+      sourcePreference = "wheel";
+    };
+  };
+  virtualEnv = pythonSet.mkVirtualEnv "sops-inventory" workspace.deps.default;
+in virtualEnv.overrideAttrs (oldAttrs: {
+  buildInputs = (oldAttrs.buildInputs or []) ++ [ pkgs.makeWrapper ];
   postInstall = ''
+    ${oldAttrs.postInstall or ""}
+
     wrapProgram $out/bin/sops-inventory \
       --prefix PATH : ${pkgs.lib.makeBinPath (with pkgs; [git])}
   '';
-}
+})