111 lines
3.6 KiB
Python
111 lines
3.6 KiB
Python
import os
|
|
import yaml
|
|
from collections import OrderedDict
|
|
from flask import Flask, jsonify, send_from_directory, request, redirect, url_for, session, render_template
|
|
from werkzeug.security import check_password_hash
|
|
import database # Imports the database.py you created earlier
|
|
|
|
# --- Path Configurations ---
|
|
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
FRONTEND_DIR = os.path.join(BASE_DIR, '..', 'frontend')
|
|
YML_PATH = os.path.join(BASE_DIR, '..\\mxpic\\PDKs\\Silterra\\directories.yaml')
|
|
COMPS_ROOT = os.path.join(BASE_DIR, '..\\mxpic\\PDKs\\Silterra')
|
|
|
|
# --- YAML & PDK Parsing Helper Functions (Unchanged) ---
|
|
def countSpaces(line):
|
|
"""Count leading spaces (tab=4)."""
|
|
expanded = line.expandtabs(4)
|
|
return len(expanded) - len(expanded.lstrip(' '))
|
|
|
|
def buildTree(filepath):
|
|
"""Build nested tree from indented yaml."""
|
|
if not os.path.exists(filepath):
|
|
return OrderedDict()
|
|
|
|
with open(filepath, 'r', encoding='utf-8') as f:
|
|
lines = f.readlines()
|
|
|
|
rootIdx = None
|
|
for i, line in enumerate(lines):
|
|
if line.strip().startswith('root') and ':' in line.strip():
|
|
rootIdx = i
|
|
break
|
|
if rootIdx is None:
|
|
return OrderedDict()
|
|
|
|
entries = []
|
|
for line in lines[rootIdx + 1:]:
|
|
stripped = line.strip()
|
|
if not stripped or stripped.startswith('#'):
|
|
continue
|
|
if stripped.startswith('- '):
|
|
spaceNum = countSpaces(line)
|
|
name = stripped[2:].strip()
|
|
if name.strip():
|
|
entries.append((spaceNum, name))
|
|
|
|
if not entries:
|
|
return OrderedDict()
|
|
|
|
minIndent = min(indent for indent, _ in entries)
|
|
nest = OrderedDict()
|
|
levelStack = [(minIndent - 1, nest)]
|
|
|
|
for spaceNum, name in entries:
|
|
while levelStack and levelStack[-1][0] >= spaceNum:
|
|
levelStack.pop()
|
|
parent = levelStack[-1][1]
|
|
child = OrderedDict()
|
|
parent[name] = child
|
|
levelStack.append((spaceNum, child))
|
|
|
|
return nest
|
|
|
|
def findComps(baseDir):
|
|
"""Scan component folders, return map of paths -> component info."""
|
|
compMap = {}
|
|
refDir = os.path.dirname(baseDir)
|
|
for root, dirs, files in os.walk(baseDir):
|
|
ymlFiles = [f for f in files if f.endswith('.yml')]
|
|
if ymlFiles:
|
|
parentDir = os.path.dirname(root)
|
|
relPath = os.path.relpath(parentDir, refDir)
|
|
parts = () if relPath == '.' else tuple(relPath.split(os.sep))
|
|
compName = os.path.basename(root)
|
|
compMap[parts] = {
|
|
'folder': compName,
|
|
'yml': ymlFiles[0]
|
|
}
|
|
dirs.clear()
|
|
return compMap
|
|
|
|
def addCompsToTree(tree, compMap):
|
|
"""Insert component nodes into the tree."""
|
|
for pathSeg, compItem in compMap.items():
|
|
compName = compItem['folder']
|
|
curNode = tree
|
|
try:
|
|
for seg in pathSeg:
|
|
curNode = curNode[seg]
|
|
except KeyError:
|
|
continue
|
|
curNode[compName] = OrderedDict({
|
|
"__type__": "component",
|
|
"__name__": compName,
|
|
"__yml__": compItem['yml']
|
|
})
|
|
return tree
|
|
|
|
def readCompYaml(compName):
|
|
"""Load YAML from component folder."""
|
|
for root, dirs, files in os.walk(COMPS_ROOT):
|
|
if os.path.basename(root) == compName:
|
|
dirs.clear()
|
|
ymlFiles = [f for f in files if f.endswith('.yml')]
|
|
if ymlFiles:
|
|
ymlPath = os.path.join(root, ymlFiles[0])
|
|
with open(ymlPath, 'r', encoding='utf-8') as f:
|
|
return yaml.safe_load(f)
|
|
return None
|
|
|