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

works

Browse source
This commit is contained in:
Paul Gauthier 2023-06-01 07:16:47 -07:00
parent 800ad12ae4
commit 6567577855
1 changed files with 17 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

28
aider/repomap.py
View file

@ -2,7 +2,8 @@ import colorsys
import json import json
import os import os
import random import random
import shelve
# import shelve
import subprocess import subprocess
import sys import sys
import tempfile import tempfile
@ -211,16 +212,18 @@ class RepoMap:
return True return True
def load_tags_cache(self): def load_tags_cache(self):
self.TAGS_CACHE = shelve.open(self.TAGS_CACHE_FILE) self.TAGS_CACHE = dict() # shelve.open(self.TAGS_CACHE_FILE)
def save_tags_cache(self): def save_tags_cache(self):
self.TAGS_CACHE.sync() # self.TAGS_CACHE.sync()
pass
def load_ident_cache(self): def load_ident_cache(self):
self.IDENT_CACHE = shelve.open(self.IDENT_CACHE_FILE) self.IDENT_CACHE = dict() # shelve.open(self.IDENT_CACHE_FILE)
def save_ident_cache(self): def save_ident_cache(self):
self.IDENT_CACHE.sync() # self.IDENT_CACHE.sync()
pass
def get_name_identifiers(self, fname, uniq=True): def get_name_identifiers(self, fname, uniq=True):
file_mtime = os.path.getmtime(fname) file_mtime = os.path.getmtime(fname)
@ -267,9 +270,11 @@ def find_py_files(directory):
def get_random_color(): def get_random_color():
hue = random.randint(0, 255) hue = random.random()
r, g, b = [int(x * 255) for x in colorsys.hsv_to_rgb(hue, 255, 64)] r, g, b = [int(x * 255) for x in colorsys.hsv_to_rgb(hue, 1, 0.75)]
return f"#{r:02x}{g:02x}{b:02x}80" res = f"#{r:02x}{g:02x}{b:02x}"
dump(hue, res)
return res
def call_map(): def call_map():
@ -377,10 +382,11 @@ def call_map():
for refs, defs, data in G.edges(data=True): for refs, defs, data in G.edges(data=True):
weight = data["weight"] weight = data["weight"]
label = data["label"] label = data["label"]
dump(refs, defs, data)
color = get_random_color() color = get_random_color()
weight = weight * 2 weight = weight * 2
dot.edge(refs, defs, penwidth=str(weight), color=color, fontcolor=color, label=label) dot.edge(refs, defs, penwidth=str(weight), color=color, fontcolor=color, label=label)
top_rank = sorted([(rank, node) for (node, rank) in ranked.items()], reverse=True) top_rank = sorted([(rank, node) for (node, rank) in ranked.items()], reverse=True)
# Print the PageRank of each node # Print the PageRank of each node