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

noop

Browse source
This commit is contained in:
Paul Gauthier 2023-05-25 08:49:40 -07:00
parent 1aa9cde119
commit f70aea7cd1
1 changed files with 23 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

25
aider/repomap.py
View file

@ -6,12 +6,12 @@ import tiktoken
from collections import defaultdict from collections import defaultdict
from aider import prompts, utils from aider import prompts, utils
from aider.dump import dump
# Global cache for tags # Global cache for tags
TAGS_CACHE = {} TAGS_CACHE = {}
from aider.dump import dump
def to_tree(tags): def to_tree(tags):
tags = sorted(tags) tags = sorted(tags)
@ -170,12 +170,32 @@ class RepoMap:
return tags return tags
def find_py_files(directory):
if not os.path.isdir(directory):
return [directory]
py_files = []
for root, dirs, files in os.walk(directory):
for file in files:
if file.endswith(".py"):
py_files.append(os.path.join(root, file))
return py_files
if __name__ == "__main__": if __name__ == "__main__":
import random import random
import graphviz import graphviz
fnames = sys.argv[1:] fnames = sys.argv[1:]
"""
fnames = []
for dname in sys.argv[1:]:
fnames += find_py_files(dname)
fnames = sorted(fnames)
"""
rm = RepoMap() rm = RepoMap()
# res = rm.get_tags_map(fnames) # res = rm.get_tags_map(fnames)
# print(res) # print(res)
@ -187,6 +207,7 @@ if __name__ == "__main__":
show_fnames = set() show_fnames = set()
for fname in fnames: for fname in fnames:
dump(fname)
show_fname = os.path.relpath(fname, root) show_fname = os.path.relpath(fname, root)
show_fnames.add(show_fname) show_fnames.add(show_fname)