From 88295ce77d3501c6417ef8f6b469099fa3ce384d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?J=C3=B6rg=20Thalheim?= <joerg@thalheim.io>
Date: Fri, 21 Jul 2023 11:21:09 +0200
Subject: [PATCH] treefmt: add ruff + black

---
 flake-parts/formatting.nix | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/flake-parts/formatting.nix b/flake-parts/formatting.nix
index a1b2159..7d2e6b9 100644
--- a/flake-parts/formatting.nix
+++ b/flake-parts/formatting.nix
@@ -1,4 +1,4 @@
-{ self
+{ lib
 , inputs
 , ...
 }: {
@@ -11,5 +11,17 @@
     treefmt.flakeFormatter = true;
     treefmt.programs.nixpkgs-fmt.enable = true;
     treefmt.programs.shellcheck.enable = true;
+    treefmt.settings.formatter.python = {
+      command = "sh";
+      options = [
+        "-eucx"
+        ''
+          ${lib.getExe pkgs.ruff} --fix "$@"
+          ${lib.getExe pkgs.black} "$@"
+        ''
+        "--" # this argument is ignored by bash
+      ];
+      includes = [ "*.py" ];
+    };
   };
 }