From 1754e574fded326d9f1458ff89e7545da0fc4dde Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Sat, 24 Jun 2023 14:53:19 -0700 Subject: [PATCH] cleanup --- aider/coders/wholefile_coder.py | 13 ++----------- 1 file changed, 2 insertions(+), 11 deletions(-) diff --git a/aider/coders/wholefile_coder.py b/aider/coders/wholefile_coder.py index 4f4b4dcaf..caf591307 100644 --- a/aider/coders/wholefile_coder.py +++ b/aider/coders/wholefile_coder.py @@ -32,8 +32,6 @@ class WholeFileCoder(Coder): output = [] lines = content.splitlines(keepends=True) - allowed_to_edit = False - fname = None new_lines = [] for i, line in enumerate(lines): @@ -53,7 +51,7 @@ class WholeFileCoder(Coder): ).splitlines() output += show_diff else: - if allowed_to_edit: + if self.allowed_to_edit(fname): edited.add(fname) if not self.dry_run: new_lines = "".join(new_lines) @@ -72,13 +70,6 @@ class WholeFileCoder(Coder): raise ValueError("No filename provided before ``` block") else: fname = lines[i - 1].strip() - - if mode == "update": - if self.allowed_to_edit(fname): - allowed_to_edit = True - else: - allowed_to_edit = False - elif fname: new_lines.append(line) else: @@ -101,7 +92,7 @@ class WholeFileCoder(Coder): return "\n".join(output) - if fname and allowed_to_edit: + if fname and self.allowed_to_edit(fname): edited.add(fname) if not self.dry_run: new_lines = "".join(new_lines)