ArmorandClaude
772f0f78f6
- Add diagram_layouts and diagram_nodes tables - Seed layout coordinates extracted from SVG - Extend snapshot API to return layout data - Rewrite demo page to fetch from API dynamically - Support ?mission= and ?at= URL parameters - Modal details populated from database component data - Show/hide nodes based on component existence at snapshot time Co-Authored-By: Claude <noreply@anthropic.com>
103 lines
4.4 KiB
Python
103 lines
4.4 KiB
Python
# topology_api/seed/seed_layout.py
|
|
"""Seed diagram layout and node positions for the engineering overview SVG."""
|
|
import json
|
|
from topology_api.db import write_cursor
|
|
|
|
|
|
def seed():
|
|
with write_cursor() as cur:
|
|
# Get base
|
|
cur.execute("SELECT id FROM bases WHERE code = 'guanghan'")
|
|
base_id = cur.fetchone()["id"]
|
|
|
|
# Upsert layout
|
|
cur.execute("""
|
|
INSERT INTO diagram_layouts (id, base_id, layout_key, name,
|
|
canvas_width, canvas_height, is_default)
|
|
VALUES (gen_random_uuid(), %(base)s, 'engineering_overview_v1',
|
|
'工程总图 v1', 1500, 1900, true)
|
|
ON CONFLICT (base_id, layout_key) DO UPDATE
|
|
SET name = EXCLUDED.name, is_default = true
|
|
RETURNING id
|
|
""", {"base": base_id})
|
|
layout_id = cur.fetchone()["id"]
|
|
print(f"Layout: {layout_id}")
|
|
|
|
# Component key → component_id mapping
|
|
cur.execute("SELECT component_key, id FROM components WHERE base_id = %(base)s",
|
|
{"base": base_id})
|
|
comp_map = {r["component_key"]: r["id"] for r in cur.fetchall()}
|
|
|
|
# SVG node positions extracted from guanghan_topology_demo.html
|
|
# node_key → (x, y, width, height, style_key, label_override, z_index)
|
|
nodes = [
|
|
# Core cross
|
|
("core_cabin", comp_map.get("core_cabin"), 630, 470, 240, 200,
|
|
"core-card", None, 10),
|
|
("habitat_2", comp_map.get("habitat_2"), 630, 220, 240, 140,
|
|
"node-card", None, 5),
|
|
("habitat_1", comp_map.get("habitat_1"), 630, 760, 240, 120,
|
|
"node-card", None, 5),
|
|
("greenhouse", comp_map.get("greenhouse"), 330, 490, 220, 160,
|
|
"node-card", None, 5),
|
|
("lab_module", comp_map.get("lab_module"), 950, 490, 220, 160,
|
|
"node-card", None, 5),
|
|
("wugang_2", comp_map.get("wugang_2"), 40, 495, 210, 150,
|
|
"vehicle-card", None, 3),
|
|
|
|
# Power cross
|
|
("power_module_hub", comp_map.get("power_module_hub"), 650, 1260, 200, 180,
|
|
"power-node-card", None, 8),
|
|
("power_storage_rack", comp_map.get("power_storage_rack"), 600, 990, 300, 160,
|
|
"power-card", "北部模块", 5),
|
|
("nuclear_reactor", comp_map.get("nuclear_reactor"), 620, 1540, 260, 150,
|
|
"power-card", "南部模块", 5),
|
|
|
|
# Power west/east groups (multiple DB components per SVG node)
|
|
# power-west groups: isru_module, mining_module, ore_fuel_storage
|
|
("power_west_group", None, 280, 1250, 260, 200,
|
|
"power-card", "西部模块", 5),
|
|
# power-east groups: fuel_storage_mod, centrifuge_mod
|
|
("power_east_group", None, 960, 1250, 260, 200,
|
|
"power-card", "东部模块", 5),
|
|
|
|
# Bingluns
|
|
("binglun_1", comp_map.get("binglun_1"), 30, 1280, 210, 140,
|
|
"vehicle-card", None, 2),
|
|
("binglun_2", comp_map.get("binglun_2"), 645, 1740, 210, 120,
|
|
"vehicle-card", None, 2),
|
|
("binglun_3", comp_map.get("binglun_3"), 1260, 1280, 210, 140,
|
|
"vehicle-card", None, 2),
|
|
]
|
|
|
|
# Delete existing nodes and re-insert
|
|
cur.execute("DELETE FROM diagram_nodes WHERE layout_id = %(lid)s",
|
|
{"lid": layout_id})
|
|
|
|
for node_key, comp_id, x, y, w, h, style, label, z in nodes:
|
|
cur.execute("""
|
|
INSERT INTO diagram_nodes
|
|
(id, layout_id, component_id, node_key, x, y,
|
|
width, height, style_key, label_override, z_index)
|
|
VALUES (gen_random_uuid(), %(lid)s, %(cid)s, %(nkey)s,
|
|
%(x)s, %(y)s, %(w)s, %(h)s,
|
|
%(style)s, %(label)s, %(z)s)
|
|
""", {
|
|
"lid": layout_id, "cid": comp_id, "nkey": node_key,
|
|
"x": x, "y": y, "w": w, "h": h,
|
|
"style": style, "label": label, "z": z,
|
|
})
|
|
|
|
print(f"Nodes inserted: {len(nodes)}")
|
|
|
|
# Node group mappings (for modal details aggregation)
|
|
groups = {
|
|
"power_west_group": ["isru_module", "mining_module", "ore_fuel_storage"],
|
|
"power_east_group": ["fuel_storage_mod", "centrifuge_mod"],
|
|
}
|
|
print(f"Node groups: {json.dumps(groups, ensure_ascii=False)}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
seed()
|