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

Refactor Coder class to use get_rel_fname method instead of os.path.relpath.

Browse source
This commit is contained in:
Paul Gauthier 2023-05-12 15:30:43 -07:00
parent 888ab713f9
commit a1cb2e0cd2
1 changed files with 7 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

11
aider/coder.py
View file

@ -105,7 +105,7 @@ class Coder:
new_files = [] new_files = []
for fname in self.abs_fnames: for fname in self.abs_fnames:
relative_fname = os.path.relpath(fname, repo.working_tree_dir) relative_fname = self.get_rel_fname(fname)
tracked_files = set(repo.git.ls_files().splitlines()) tracked_files = set(repo.git.ls_files().splitlines())
if relative_fname not in tracked_files: if relative_fname not in tracked_files:
new_files.append(relative_fname) new_files.append(relative_fname)
@ -134,7 +134,7 @@ class Coder:
prompt = "" prompt = ""
for fname in fnames: for fname in fnames:
relative_fname = os.path.relpath(fname, self.root) relative_fname = self.get_rel_fname(fname)
prompt += utils.quoted_file(fname, relative_fname) prompt += utils.quoted_file(fname, relative_fname)
return prompt return prompt
@ -461,7 +461,7 @@ class Coder:
diffs = "" diffs = ""
relative_dirty_files = [] relative_dirty_files = []
for fname in file_list: for fname in file_list:
relative_fname = os.path.relpath(fname, repo.working_tree_dir) relative_fname = self.get_rel_fname(fname)
relative_dirty_files.append(relative_fname) relative_dirty_files.append(relative_fname)
try: try:
@ -533,8 +533,11 @@ class Coder:
return commit_hash, commit_message return commit_hash, commit_message
def get_rel_fname(self, abs_fname):
return self.get_rel_fname(fname)
def get_inchat_relative_files(self): def get_inchat_relative_files(self):
files = [os.path.relpath(fname, self.root) for fname in self.abs_fnames] files = [self.get_rel_fname(fname) for fname in self.abs_fnames]
return sorted(set(files)) return sorted(set(files))
def get_all_relative_files(self): def get_all_relative_files(self):