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

refactor: update get_ai_comments to return line nums, comments and bang status

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-12-01 07:54:11 -08:00
parent a64f14db51
commit 7ee69628a3
1 changed files with 6 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

14
aider/watch.py
View file

@ -173,11 +173,8 @@ class FileWatcher:
# Refresh all AI comments from tracked files # Refresh all AI comments from tracked files
ai_comments = {} ai_comments = {}
for fname in self.coder.abs_fnames: for fname in self.coder.abs_fnames:
line_nums, has_bang = self.get_ai_comments(fname) line_nums, comments, has_bang = self.get_ai_comments(fname)
if line_nums: 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 ai_comments[fname] = comments
has_bangs = has_bang has_bangs = has_bang
@ -196,10 +193,10 @@ class FileWatcher:
dump(res) dump(res)
return res return res
# ai actually, have this function return the line nums, the comments and has_bang!
def get_ai_comments(self, filepath): def get_ai_comments(self, filepath):
"""Extract AI comment line numbers and bang status from a file""" """Extract AI comment line numbers, comments and bang status from a file"""
line_nums = [] line_nums = []
comments = []
has_bang = False has_bang = False
content = self.io.read_text(filepath) content = self.io.read_text(filepath)
for i, line in enumerate(content.splitlines(), 1): for i, line in enumerate(content.splitlines(), 1):
@ -207,11 +204,12 @@ class FileWatcher:
comment = match.group(0).strip() comment = match.group(0).strip()
if comment: if comment:
line_nums.append(i) line_nums.append(i)
comments.append(comment)
if comment.strip().endswith("!"): if comment.strip().endswith("!"):
has_bang = True has_bang = True
if not line_nums: if not line_nums:
return None, False return None, None, False
return line_nums, has_bang return line_nums, comments, has_bang
def main(): def main():