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

refactor: simplify get_ai_comments to return only line numbers and bang status

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-12-01 07:49:14 -08:00
parent cd922e919e
commit 96c6d408fb
1 changed files with 12 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

23
aider/watch.py
View file

@ -173,9 +173,13 @@ class FileWatcher:
# Refresh all AI comments from tracked files
ai_comments = {}
for fname in self.coder.abs_fnames:
comments, line_nums, has_bang = self.get_ai_comments(fname)
ai_comments[fname] = comments
has_bangs = has_bang
line_nums, has_bang = self.get_ai_comments(fname)
if line_nums:
content = self.io.read_text(fname)
lines = content.splitlines()
comments = [lines[i-1].strip() for i in line_nums]
ai_comments[fname] = comments
has_bangs = has_bang
if not has_bangs:
return ""
@ -192,10 +196,8 @@ class FileWatcher:
dump(res)
return res
#ai don't return the comments, just line nums and has bang!
def get_ai_comments(self, filepath):
"""Extract all AI comments from a file, returning comments, line numbers and bang status"""
comments = []
"""Extract AI comment line numbers and bang status from a file"""
line_nums = []
has_bang = False
try:
@ -204,15 +206,14 @@ class FileWatcher:
if match := self.ai_comment_pattern.search(line):
comment = match.group(0).strip()
if comment:
comments.append(comment)
line_nums.append(i)
if comment.strip().endswith("!"):
has_bang = True
except Exception:
return None, None, False
if not comments:
return None, None, False
return comments, line_nums, has_bang
return None, False
if not line_nums:
return None, False
return line_nums, has_bang
def main():