From d8a5108f9e093efc24af3a0d952727eaa04d8271 Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Sat, 18 May 2024 19:40:02 -0700 Subject: [PATCH] Added functionality to include tree context in linter results. --- aider/linter.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/aider/linter.py b/aider/linter.py index 0f4018524..b1aa3336d 100644 --- a/aider/linter.py +++ b/aider/linter.py @@ -36,7 +36,7 @@ class Linter: else: return fname - def run_cmd(self, cmd, rel_fname): + def run_cmd(self, cmd, rel_fname, code): cmd += " " + rel_fname cmd = cmd.split() @@ -52,6 +52,11 @@ class Linter: res = f"# Running: {cmd}\n\n" res += errors + filenames_linenums = find_filenames_and_linenums(errors, [rel_fname]) + if filenames_linenums: + filename, linenums = filenames_linenums.items().next() + res += tree_context(rel_fname, code, linenums) + return res def lint(self, fname): @@ -68,7 +73,7 @@ class Linter: return cmd(fname, rel_fname, code) if cmd: - return self.run_cmd(cmd, rel_fname) + return self.run_cmd(cmd, rel_fname, code) return basic_lint(rel_fname, code)