mirror of
https://github.com/openswarm-ai/openswarm.git
synced 2026-08-17 18:25:42 +02:00
201 lines
9.2 KiB
Python
201 lines
9.2 KiB
Python
import os
|
|
import json
|
|
import colorsys
|
|
from pathlib import Path
|
|
from debugger_backend.DebugFile import DebugFile
|
|
from debugger_backend.DEFAULTS import DEFAULT_COLOR, DEFAULT_TOGGLED, DEFAULT_SET_MANUALLY, DEFAULT_EMOJI
|
|
from debugger_backend.path_mngr import get_abspath, get_root_rel_path
|
|
|
|
class Directory:
|
|
def __init__(self, path, color=DEFAULT_COLOR, is_toggled=DEFAULT_TOGGLED,
|
|
set_manually=DEFAULT_SET_MANUALLY, emoji=DEFAULT_EMOJI):
|
|
self.path = path
|
|
# print(f"Directory init: {self.path}")
|
|
self.children = [] # Can contain DebugFile or other Directory objects
|
|
self.color = color
|
|
self.is_toggled = is_toggled
|
|
self.set_manually = set_manually
|
|
self.emoji = emoji
|
|
|
|
def __str__(self):
|
|
return f"Directory: {self.path}\nNum Children: {len(self.children)}\nColor: {self.color}\nToggled: {self.is_toggled}\nSet Manually: {self.set_manually}"
|
|
|
|
def get_abspath(self):
|
|
return get_abspath(self.path)
|
|
|
|
def add_child(self, child):
|
|
"""
|
|
Adds a child to the directory (either a DebugFile or another Directory).
|
|
"""
|
|
self.children.append(child)
|
|
|
|
def get_ordered_abspaths_and_instances(self):
|
|
# print("[get_ordered_abspaths]: START")
|
|
curr_file_path = os.path.abspath(__file__)
|
|
root_dir = os.path.dirname(os.path.dirname(os.path.dirname(curr_file_path)))
|
|
# print(f"[get_ordered_abspaths]: Curr path: {curr_file_path}")
|
|
# print(f"[get_ordered_abspaths]: Dir path: {root_dir}")
|
|
def construct_ordered_abspaths(dir: Directory, ordered_abspaths: list):
|
|
dir_path = dir.path
|
|
full_path = os.path.join(root_dir, dir_path)
|
|
ordered_abspaths.append({"abspath": full_path, "instance": dir})
|
|
# print(f"\t[construct_ordered_abspaths]: Full path: {full_path}")
|
|
for child in dir.children:
|
|
child_abspath = os.path.join(root_dir, child.path).lower()
|
|
if os.path.isdir(child_abspath):
|
|
construct_ordered_abspaths(child, ordered_abspaths)
|
|
elif os.path.isfile(child_abspath):
|
|
# print(f"\t[construct_ordered_abspaths]: Child is file: {child_abspath}")
|
|
ordered_abspaths.append({"abspath": child_abspath, "instance": child})
|
|
else:
|
|
print(f"\033[38;5;120mEntry is non existent: {child_abspath}\033[0m")
|
|
# print(f"\t[construct_ordered_abspaths]: Finished for dir: {full_path}")
|
|
# print(f"\t[construct_ordered_abspaths]: RETURNING FROM DIR: {full_path}")
|
|
return ordered_abspaths
|
|
ordered_abspaths_and_instances = construct_ordered_abspaths(self, [])
|
|
# print("[get_ordered_abspaths]: Finished getting ordered abspaths and instances")
|
|
# for abspath_and_instance in ordered_abspaths_and_instances:
|
|
# abspath = abspath_and_instance["abspath"]
|
|
# print(f"\t[get_ordered_abspaths]: Abspath: {abspath}")
|
|
ordered_abspaths = [abspath_and_instance["abspath"] for abspath_and_instance in ordered_abspaths_and_instances]
|
|
ordered_instances = [abspath_and_instance["instance"] for abspath_and_instance in ordered_abspaths_and_instances]
|
|
return ordered_abspaths, ordered_instances
|
|
|
|
|
|
def build_structure(self):
|
|
print("[build_structure]: START")
|
|
root_dir = self.get_abspath()
|
|
# print(f"[build_structure]: Root dir: {root_dir}")
|
|
excluded_dirs = [".venv", "debugger", "node_modules", ".git", "__pycache__"]
|
|
project_structure = []
|
|
|
|
def construct_project_structure(dir_path: str, parent_dir: Directory):
|
|
# print(f"[build_structure]: Scanning dir: {dir_path}")
|
|
with os.scandir(dir_path) as it:
|
|
for entry in it:
|
|
# print(f"[build_structure]: Entry: {entry.path}")
|
|
if any(excluded_dir in entry.path for excluded_dir in excluded_dirs):
|
|
# print(f"[build_structure]: Excluding {entry.path}")
|
|
continue
|
|
root_rel_path = get_root_rel_path(entry.path)
|
|
if entry.is_dir():
|
|
subdir = Directory(root_rel_path)
|
|
construct_project_structure(entry.path, subdir)
|
|
parent_dir.add_child(subdir)
|
|
elif entry.is_file():
|
|
debug_file = DebugFile(filename=entry.name, path=root_rel_path)
|
|
if debug_file.calls_debug_function():
|
|
parent_dir.add_child(debug_file)
|
|
else:
|
|
raise Exception(f"[build_structure]: Entry is not dir or file: {entry.path}")
|
|
|
|
construct_project_structure(root_dir, self)
|
|
# [print(f"[build_structure]: {file}") for file in project_structure]
|
|
# print(f"[build_structure]: END")
|
|
return
|
|
|
|
def to_dict(self):
|
|
"""
|
|
Converts the Directory object to a dictionary format, recursively.
|
|
"""
|
|
return {
|
|
"name": os.path.basename(self.path),
|
|
"color": self.color,
|
|
"is_toggled": self.is_toggled,
|
|
"set_manually": self.set_manually,
|
|
"emoji": self.emoji,
|
|
"children": [child.to_dict() if isinstance(child, DebugFile) else child.to_dict() for child in self.children]
|
|
}
|
|
|
|
def prune_empty(self):
|
|
# Recursively prune empty directories
|
|
# Base case) if the current directory has no children, return
|
|
# Recursive case) for each of the directories in the current directory, call prune_empty
|
|
# then remove the directory from the children of the current directory if it has no children
|
|
for child in self.children[:]:
|
|
if isinstance(child, Directory):
|
|
# Recursively prune empty subdirectories
|
|
child.prune_empty()
|
|
# If the subdirectory is empty after pruning, remove it
|
|
if len(child.children) == 0:
|
|
self.children.remove(child)
|
|
|
|
def propagate_toggled_state(self):
|
|
"""
|
|
Propagates the toggled state down the hierarchy.
|
|
"""
|
|
for child in self.children:
|
|
if isinstance(child, DebugFile) and not child.set_manually:
|
|
child.is_toggled = self.is_toggled
|
|
elif isinstance(child, Directory) and not child.set_manually:
|
|
child.is_toggled = self.is_toggled
|
|
child.propagate_toggled_state()
|
|
|
|
def propagate_color(self, parent_color=DEFAULT_COLOR):
|
|
"""
|
|
Propagates the color from parent to children.
|
|
"""
|
|
if self.color == DEFAULT_COLOR:
|
|
self.color = lighten_color(parent_color)
|
|
for child in self.children:
|
|
if isinstance(child, DebugFile) and child.color == DEFAULT_COLOR:
|
|
child.color = lighten_color(self.color)
|
|
elif isinstance(child, Directory):
|
|
child.propagate_color(self.color)
|
|
|
|
def load_from_json(self, json_data):
|
|
"""
|
|
Loads a directory structure from a JSON file into this Directory instance.
|
|
"""
|
|
for item in json_data:
|
|
if 'children' in item:
|
|
subdir = Directory(
|
|
path=os.path.join(self.path, item['name']),
|
|
color=item.get('color', DEFAULT_COLOR),
|
|
is_toggled=item.get('is_toggled', DEFAULT_TOGGLED),
|
|
set_manually=item.get('set_manually', DEFAULT_SET_MANUALLY),
|
|
emoji=item.get('emoji', DEFAULT_EMOJI)
|
|
)
|
|
|
|
subdir.load_from_json(item['children'])
|
|
self.add_child(subdir)
|
|
else:
|
|
# debug_file = DebugFile.from_dict(item, self)
|
|
debug_file = DebugFile(
|
|
filename=item['name'],
|
|
path=os.path.join(self.path, item['name']),
|
|
color=item.get('color', DEFAULT_COLOR),
|
|
is_toggled=item.get('is_toggled', DEFAULT_TOGGLED),
|
|
set_manually=item.get('set_manually', DEFAULT_SET_MANUALLY),
|
|
emoji=item.get('emoji', DEFAULT_EMOJI),
|
|
directory=self
|
|
)
|
|
self.add_child(debug_file)
|
|
|
|
def reset_colors(self):
|
|
"""
|
|
Resets the color of all DebugFile and Directory objects in this directory structure to the default color.
|
|
"""
|
|
self.color = DEFAULT_COLOR
|
|
for child in self.children:
|
|
if isinstance(child, DebugFile):
|
|
child.color = DEFAULT_COLOR
|
|
elif isinstance(child, Directory):
|
|
child.reset_colors()
|
|
|
|
|
|
def lighten_color(color, amount=0.1):
|
|
"""
|
|
Lightens the given color by the specified amount.
|
|
"""
|
|
try:
|
|
color = color.lstrip('#')
|
|
r, g, b = int(color[:2], 16), int(color[2:4], 16), int(color[4:6], 16)
|
|
h, l, s = colorsys.rgb_to_hls(r / 255.0, g / 255.0, b / 255.0)
|
|
l = min(1, l + amount)
|
|
r, g, b = colorsys.hls_to_rgb(h, l, s)
|
|
return '#{:02x}{:02x}{:02x}'.format(int(r * 255), int(g * 255), int(b * 255))
|
|
except Exception as e:
|
|
print(f"Error lightening color {color}: {e}")
|
|
return color
|