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

wip

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

19
aider/repomap.py
View file

@ -1,5 +1,7 @@
import colorsys
import json import json
import os import os
import random
import shelve import shelve
import subprocess import subprocess
import sys import sys
@ -264,9 +266,13 @@ def find_py_files(directory):
return py_files return py_files
def call_map(): def get_random_color():
import random hue = random.randint(0, 255)
r, g, b = [int(x * 255) for x in colorsys.hsv_to_rgb(hue, 255, 64)]
return f"#{r:02x}{g:02x}{b:02x}80"
def call_map():
import graphviz import graphviz
fnames = sys.argv[1:] fnames = sys.argv[1:]
@ -372,12 +378,9 @@ def call_map():
weight = data["weight"] weight = data["weight"]
label = data["label"] label = data["label"]
r = random.randint(0, 128) color = get_random_color()
g = random.randint(0, 128) weight = weight * 2
b = random.randint(0, 128) dot.edge(refs, defs, penwidth=str(weight), color=color, fontcolor=color, label=label)
color = f"#{r:02x}{g:02x}{b:02x}80"
weight = weight * 2
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