mirror of
https://github.com/openswarm-ai/openswarm.git
synced 2026-09-11 20:27:44 +02:00
111 lines
5.0 KiB
Python
111 lines
5.0 KiB
Python
import os
|
|
import json
|
|
import colorsys
|
|
from typing import Union
|
|
from debugger_backend.Directory import Directory
|
|
from debugger_backend.DEFAULTS import DEFAULT_COLOR, DEFAULT_TOGGLED, DEFAULT_SET_MANUALLY, TOGGLE_FILE, DEFAULT_EMOJI, ROOT_DIR
|
|
from debugger_backend.DebugFile import DebugFile
|
|
from collections import OrderedDict
|
|
|
|
def merge_directories(json_dir: Directory, scanned_dir: Directory):
|
|
"""Merge json_dir into scanned_dir; json values win on overlap, matched by full path."""
|
|
json_abspaths, json_instances = json_dir.get_ordered_abspaths_and_instances()
|
|
scanned_abspaths, scanned_instances = scanned_dir.get_ordered_abspaths_and_instances()
|
|
|
|
def find_matching_in_structure(scanned_child: Union[DebugFile, Directory], json_dir: Directory):
|
|
assert json_dir in json_instances, f"JSON_DIR: {json_dir.path} not in json_instances"
|
|
assert scanned_child in scanned_instances, f"SCANNED_CHILD: {scanned_child.path} not in scanned_instances"
|
|
scanned_id = scanned_instances.index(scanned_child)
|
|
scanned_abspath = scanned_abspaths[scanned_id]
|
|
json_instance = None
|
|
try:
|
|
json_id = json_abspaths.index(scanned_abspath)
|
|
json_instance = json_instances[json_id]
|
|
except ValueError:
|
|
pass
|
|
return json_instance
|
|
|
|
def construct_merged_dir(json_dir: Directory, scanned_dir: Directory):
|
|
for scanned_child in scanned_dir.children:
|
|
matching_json_child = find_matching_in_structure(scanned_child, json_dir)
|
|
|
|
if isinstance(scanned_child, DebugFile) and matching_json_child:
|
|
scanned_child.color = matching_json_child.color
|
|
scanned_child.is_toggled = matching_json_child.is_toggled
|
|
scanned_child.set_manually = matching_json_child.set_manually
|
|
scanned_child.emoji = matching_json_child.emoji
|
|
|
|
elif isinstance(scanned_child, Directory) and matching_json_child:
|
|
scanned_child.color = matching_json_child.color
|
|
scanned_child.is_toggled = matching_json_child.is_toggled
|
|
scanned_child.set_manually = matching_json_child.set_manually
|
|
scanned_child.emoji = matching_json_child.emoji
|
|
|
|
construct_merged_dir(matching_json_child, scanned_child)
|
|
else:
|
|
scanned_child.color = DEFAULT_COLOR
|
|
scanned_child.is_toggled = DEFAULT_TOGGLED
|
|
scanned_child.set_manually = DEFAULT_SET_MANUALLY
|
|
scanned_child.emoji = DEFAULT_EMOJI
|
|
|
|
construct_merged_dir(json_dir, scanned_dir)
|
|
|
|
|
|
def update_debug_toggles(save_to_file=True) -> Directory:
|
|
json_loaded_dir = None
|
|
if os.path.exists(TOGGLE_FILE):
|
|
with open(TOGGLE_FILE, 'r', encoding='utf-8') as file:
|
|
try:
|
|
json_data = json.load(file)
|
|
json_loaded_dir = Directory("")
|
|
json_loaded_dir = Directory(path="",
|
|
color=json_data[0].get('color', DEFAULT_COLOR),
|
|
is_toggled=json_data[0].get('is_toggled', DEFAULT_TOGGLED),
|
|
set_manually=json_data[0].get('set_manually', DEFAULT_SET_MANUALLY),
|
|
emoji=json_data[0].get('emoji', DEFAULT_EMOJI)
|
|
)
|
|
json_loaded_dir.load_from_json(json_data[0]['children'])
|
|
|
|
except json.JSONDecodeError:
|
|
ValueError("Error: JSON file could not be decoded.")
|
|
else:
|
|
print("No JSON file found")
|
|
scanned_dir = Directory(path="",
|
|
color=json_loaded_dir.color if json_loaded_dir else DEFAULT_COLOR,
|
|
is_toggled=json_loaded_dir.is_toggled if json_loaded_dir else DEFAULT_TOGGLED,
|
|
set_manually=json_loaded_dir.set_manually if json_loaded_dir else DEFAULT_SET_MANUALLY,
|
|
emoji=json_loaded_dir.emoji if json_loaded_dir else DEFAULT_EMOJI
|
|
)
|
|
scanned_dir.build_structure()
|
|
scanned_dir.prune_empty()
|
|
scanned_dir.propagate_toggled_state()
|
|
|
|
if json_loaded_dir:
|
|
merge_directories(json_loaded_dir, scanned_dir)
|
|
|
|
scanned_dir.propagate_color()
|
|
output = dir_to_output_format(scanned_dir)
|
|
|
|
if save_to_file:
|
|
with open(TOGGLE_FILE, 'w', encoding='utf-8') as file:
|
|
json.dump(output, file, ensure_ascii=False, indent=4)
|
|
return scanned_dir
|
|
|
|
def dir_to_output_format(input_dir):
|
|
root_node = {
|
|
"name": "root",
|
|
"color": input_dir.color,
|
|
"is_toggled": input_dir.is_toggled,
|
|
"set_manually": input_dir.set_manually,
|
|
"emoji": input_dir.emoji,
|
|
"children": input_dir.to_dict()["children"]
|
|
}
|
|
return [ordered(root_node)]
|
|
|
|
def ordered(obj):
|
|
if isinstance(obj, dict):
|
|
return OrderedDict((k, ordered(v)) for k, v in obj.items())
|
|
if isinstance(obj, list):
|
|
return [ordered(x) for x in obj]
|
|
return obj
|