dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-30 09:14:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

refactor: add dry run mode to apply_edits method

Browse source
This commit is contained in:
Paul Gauthier 2024-10-29 14:28:38 -07:00 • committed by Paul Gauthier (aider)
parent 3ccae4eff7
commit 28d9f6f8da
1 changed files with 14 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

15
aider/coders/editblock_coder.py
View file

@ -35,9 +35,13 @@ class EditBlockCoder(Coder):
return edits return edits
def apply_edits(self, edits): def apply_edits_dry_run(self, edits):
return self.apply_edits(edits, dry_run=True)
def apply_edits(self, edits, dry_run=False):
failed = [] failed = []
passed = [] passed = []
updated = []
for edit in edits: for edit in edits:
path, original, updated = edit path, original, updated = edit
@ -52,14 +56,21 @@ class EditBlockCoder(Coder):
if new_content: if new_content:
break break
# ai: update full_path->path!
updated.append((path, original, updated))
if new_content: if new_content:
if not dry_run:
self.io.write_text(full_path, new_content) self.io.write_text(full_path, new_content)
passed.append(edit) passed.append(edit)
else: else:
failed.append(edit) failed.append(edit)
if dry_run:
return updated
if not failed: if not failed:
return passed return
blocks = "block" if len(failed) == 1 else "blocks" blocks = "block" if len(failed) == 1 else "blocks"