dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-23 22:04:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

refactor orig/upd into utils

Browse source
This commit is contained in:
Paul Gauthier 2023-05-11 20:57:27 -07:00
parent 5e593eb631
commit 797372c69e
2 changed files with 26 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

24
aider/coder.py
View file

@ -1,9 +1,7 @@
#!/usr/bin/env python
import os
import sys
import re
import traceback
import time
from openai.error import RateLimitError
@ -374,29 +372,9 @@ class Coder:
if live:
live.stop()
pattern = re.compile(
# Optional: Matches the start of a code block (e.g., ```python) and any following whitespace
r"(^```\S*\s*)?"
# Matches the file path
r"^(\S*)\s*"
# Optional: Matches the end of a code block (e.g., ```) and any following whitespace
r"(^```\S*\s*)?"
# Matches the start of the ORIGINAL section and captures its content
r"^<<<<<<< ORIGINAL\n(.*?\n?)"
# Matches sep between ORIGINAL and UPDATED sections, captures UPDATED content
r"^=======\n(.*?)"
# Matches the end of the UPDATED section
r"^>>>>>>> UPDATED",
re.MULTILINE | re.DOTALL,
)
def update_files(self, content, inp):
edited = set()
for match in self.pattern.finditer(content):
_, path, _, original, updated = match.groups()
path = path.strip()
for path, original, updated in utils.find_original_update_blocks(content):
full_path = os.path.abspath(os.path.join(self.root, path))
if full_path not in self.abs_fnames: