dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-31 01:35:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

handle diff output for new file

Browse source
This commit is contained in:
Paul Gauthier 2023-07-07 11:58:27 -07:00
parent 93d4fe5f14
commit 552bdc4f71
1 changed files with 14 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

23
aider/coders/wholefile_coder.py
View file

@ -55,15 +55,18 @@ class WholeFileCoder(Coder):
full_path = (Path(self.root) / fname).absolute()
if mode == "diff" and full_path.exists():
orig_lines = self.io.read_text(full_path).splitlines(keepends=True)
if mode == "diff":
if full_path.exists():
orig_lines = self.io.read_text(full_path).splitlines(keepends=True)
show_diff = diffs.diff_partial_update(
orig_lines,
new_lines,
final=True,
).splitlines()
output += show_diff
show_diff = diffs.diff_partial_update(
orig_lines,
new_lines,
final=True,
).splitlines()
output += show_diff
else:
output += ["```"] + new_lines + ["```"]
else:
if self.allowed_to_edit(fname):
edited.add(fname)
@ -110,7 +113,7 @@ class WholeFileCoder(Coder):
# ending an existing block
full_path = (Path(self.root) / fname).absolute()
if mode == "diff" and full_path.exists():
if full_path.exists():
orig_lines = self.io.read_text(full_path).splitlines(keepends=True)
show_diff = diffs.diff_partial_update(
@ -118,6 +121,8 @@ class WholeFileCoder(Coder):
new_lines,
).splitlines()
output += show_diff
else:
output += ["```"] + new_lines + ["```"]
return "\n".join(output)