Kyle/Wiring-Designer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Scope git commit to current diagram + fix complete JSON export/import

Browse Source 
- Git commit: "Current diagram only" checkbox stages only that diagram's
  JSON file instead of all diagrams
- JSON export now includes: waypoints, twisted_pair, twist_pitch, shielded,
  show_size_label, bundle_id, bundle_label, sheet_id, sheets, wire_bundles
- JSON import reconstructs sheets and bundles with remapped IDs so restored
  diagrams are exact copies

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
Kyle
2026-04-24 20:02:19 -04:00
parent 7ae79bc9aa
commit 072e620d56
6 changed files with 130 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/app/routers/diagrams.py
+34 -4
View File
@@ -119,10 +119,38 @@ async def import_diagram(request: Request, db: Session = Depends(get_db)):
db.add(new_diag)
db.flush()
id_map = {}
# Sheets: old id → new Sheet object
sheet_map = {}
for s in data.get("sheets", []):
ns = models.Sheet(
diagram_id=new_diag.id,
name=s.get("name", "Sheet 1"),
position=s.get("position", 0),
)
db.add(ns)
db.flush()
if "id" in s:
sheet_map[s["id"]] = ns.id
# Wire bundles: old id → new bundle id
bundle_map = {}
for b in data.get("wire_bundles", []):
nb = models.WireBundle(
diagram_id=new_diag.id,
label=b.get("label", ""),
jacket_color=b.get("jacket_color", "#2a2a2a"),
)
db.add(nb)
db.flush()
if "id" in b:
bundle_map[b["id"]] = nb.id
# Devices: old id → new device id
dev_map = {}
for dev_data in data.get("devices", []):
new_dev = models.Device(
diagram_id=new_diag.id,
sheet_id=sheet_map.get(dev_data.get("sheet_id")),
device_type=dev_data.get("device_type", "generic"),
label=dev_data.get("label", ""),
reference=dev_data.get("reference", ""),
@@ -136,15 +164,16 @@ async def import_diagram(request: Request, db: Session = Depends(get_db)):
db.add(new_dev)
db.flush()
if "id" in dev_data:
id_map[dev_data["id"]] = new_dev.id
dev_map[dev_data["id"]] = new_dev.id
for wire_data in data.get("wires", []):
new_wire = models.Wire(
diagram_id=new_diag.id,
sheet_id=sheet_map.get(wire_data.get("sheet_id")),
label=wire_data.get("label", ""),
from_device_id=id_map.get(wire_data.get("from_device_id")),
from_device_id=dev_map.get(wire_data.get("from_device_id")),
from_pin=wire_data.get("from_pin", ""),
to_device_id=id_map.get(wire_data.get("to_device_id")),
to_device_id=dev_map.get(wire_data.get("to_device_id")),
to_pin=wire_data.get("to_pin", ""),
color_primary=wire_data.get("color_primary", "#FF0000"),
color_stripe=wire_data.get("color_stripe"),
@@ -157,6 +186,7 @@ async def import_diagram(request: Request, db: Session = Depends(get_db)):
twist_pitch=wire_data.get("twist_pitch", 16.0),
shielded=wire_data.get("shielded", False),
show_size_label=wire_data.get("show_size_label", False),
bundle_id=bundle_map.get(wire_data.get("bundle_id")),
)
db.add(new_wire)
backend/app/routers/export.py
+21 -2
View File
@@ -283,13 +283,25 @@ def export_json(diagram_id: int, db: Session = Depends(get_db)):
if not diagram:
raise HTTPException(status_code=404, detail="Diagram not found")
# Build bundle id→label map for wire export
bundle_label = {b.id: b.label for b in diagram.wire_bundles}
data = {
"id": diagram.id,
"name": diagram.name,
"description": diagram.description,
"sheets": [
{"id": s.id, "name": s.name, "position": s.position}
for s in diagram.sheets
],
"wire_bundles": [
{"id": b.id, "label": b.label, "jacket_color": b.jacket_color}
for b in diagram.wire_bundles
],
"devices": [
{
"id": d.id, "device_type": d.device_type, "label": d.label,
"id": d.id, "sheet_id": d.sheet_id,
"device_type": d.device_type, "label": d.label,
"reference": d.reference, "x": d.x, "y": d.y,
"width": d.width, "height": d.height,
"properties": d.properties, "pins": d.pins,
@@ -298,12 +310,19 @@ def export_json(diagram_id: int, db: Session = Depends(get_db)):
],
"wires": [
{
"id": w.id, "label": w.label,
"id": w.id, "sheet_id": w.sheet_id, "label": w.label,
"from_device_id": w.from_device_id, "from_pin": w.from_pin,
"to_device_id": w.to_device_id, "to_pin": w.to_pin,
"color_primary": w.color_primary, "color_stripe": w.color_stripe,
"gauge": w.gauge, "length": w.length, "length_unit": w.length_unit,
"notes": w.notes,
"waypoints": w.waypoints or [],
"twisted_pair": bool(w.twisted_pair),
"twist_pitch": w.twist_pitch,
"shielded": bool(w.shielded),
"show_size_label": bool(w.show_size_label),
"bundle_id": w.bundle_id,
"bundle_label": bundle_label.get(w.bundle_id),
}
for w in diagram.wires
],
backend/app/routers/git_mgr.py
+63 -11
View File
@@ -42,13 +42,23 @@ def _safe_name(diagram_id: int, name: str) -> str:
def _diagram_to_dict(diagram, db: Session) -> dict:
bundle_label = {b.id: b.label for b in diagram.wire_bundles}
return {
"id": diagram.id,
"name": diagram.name,
"description": diagram.description or "",
"sheets": [
{"id": s.id, "name": s.name, "position": s.position}
for s in diagram.sheets
],
"wire_bundles": [
{"id": b.id, "label": b.label, "jacket_color": b.jacket_color}
for b in diagram.wire_bundles
],
"devices": [
{
"id": d.id, "device_type": d.device_type, "label": d.label,
"id": d.id, "sheet_id": d.sheet_id,
"device_type": d.device_type, "label": d.label,
"reference": d.reference, "x": d.x, "y": d.y,
"width": d.width, "height": d.height,
"properties": d.properties, "pins": d.pins,
@@ -57,7 +67,7 @@ def _diagram_to_dict(diagram, db: Session) -> dict:
],
"wires": [
{
"id": w.id, "label": w.label,
"id": w.id, "sheet_id": w.sheet_id, "label": w.label,
"from_device_id": w.from_device_id, "from_pin": w.from_pin,
"to_device_id": w.to_device_id, "to_pin": w.to_pin,
"color_primary": w.color_primary, "color_stripe": w.color_stripe,
@@ -65,6 +75,7 @@ def _diagram_to_dict(diagram, db: Session) -> dict:
"notes": w.notes, "waypoints": w.waypoints or [],
"twisted_pair": bool(w.twisted_pair), "twist_pitch": w.twist_pitch,
"shielded": bool(w.shielded), "show_size_label": bool(w.show_size_label),
"bundle_id": w.bundle_id, "bundle_label": bundle_label.get(w.bundle_id),
}
for w in diagram.wires
],
@@ -78,10 +89,36 @@ def _import_data(data: dict, db: Session, name_override: Optional[str] = None) -
)
db.add(new_diag)
db.flush()
id_map = {}
sheet_map = {}
for s in data.get("sheets", []):
ns = models.Sheet(
diagram_id=new_diag.id,
name=s.get("name", "Sheet 1"),
position=s.get("position", 0),
)
db.add(ns)
db.flush()
if "id" in s:
sheet_map[s["id"]] = ns.id
bundle_map = {}
for b in data.get("wire_bundles", []):
nb = models.WireBundle(
diagram_id=new_diag.id,
label=b.get("label", ""),
jacket_color=b.get("jacket_color", "#2a2a2a"),
)
db.add(nb)
db.flush()
if "id" in b:
bundle_map[b["id"]] = nb.id
dev_map = {}
for dev in data.get("devices", []):
nd = models.Device(
diagram_id=new_diag.id,
sheet_id=sheet_map.get(dev.get("sheet_id")),
device_type=dev.get("device_type", "generic"),
label=dev.get("label", ""),
reference=dev.get("reference", ""),
@@ -93,14 +130,16 @@ def _import_data(data: dict, db: Session, name_override: Optional[str] = None) -
db.add(nd)
db.flush()
if "id" in dev:
id_map[dev["id"]] = nd.id
dev_map[dev["id"]] = nd.id
for w in data.get("wires", []):
nw = models.Wire(
diagram_id=new_diag.id,
sheet_id=sheet_map.get(w.get("sheet_id")),
label=w.get("label", ""),
from_device_id=id_map.get(w.get("from_device_id")),
from_device_id=dev_map.get(w.get("from_device_id")),
from_pin=w.get("from_pin", ""),
to_device_id=id_map.get(w.get("to_device_id")),
to_device_id=dev_map.get(w.get("to_device_id")),
to_pin=w.get("to_pin", ""),
color_primary=w.get("color_primary", "#FF0000"),
color_stripe=w.get("color_stripe"),
@@ -113,8 +152,10 @@ def _import_data(data: dict, db: Session, name_override: Optional[str] = None) -
twist_pitch=w.get("twist_pitch", 16.0),
shielded=w.get("shielded", False),
show_size_label=w.get("show_size_label", False),
bundle_id=bundle_map.get(w.get("bundle_id")),
)
db.add(nw)
db.commit()
db.refresh(new_diag)
return new_diag
@@ -162,6 +203,7 @@ def git_log():
class CommitBody(BaseModel):
message: str
diagram_id: Optional[int] = None # None = all diagrams
@router.post("/commit")
@@ -171,14 +213,24 @@ def git_commit(body: CommitBody, db: Session = Depends(get_db)):
repo_root = _repo_root()
diagrams_dir = _diagrams_dir(repo_root)
# Export all diagrams to diagrams/ folder
diagrams = db.query(models.Diagram).all()
for diagram in diagrams:
if body.diagram_id is not None:
# Export only the specified diagram
diagram = db.query(models.Diagram).filter(models.Diagram.id == body.diagram_id).first()
if not diagram:
raise HTTPException(404, "Diagram not found")
to_export = [diagram]
else:
to_export = db.query(models.Diagram).all()
exported_files = []
for diagram in to_export:
data = _diagram_to_dict(diagram, db)
fname = _safe_name(diagram.id, diagram.name)
(diagrams_dir / fname).write_text(json.dumps(data, indent=2), encoding="utf-8")
exported_files.append(f"diagrams/{fname}")
add_r = _run("add", "diagrams/", cwd=repo_root)
# Stage only the exported files (not the whole diagrams/ dir)
add_r = _run("add", *exported_files, cwd=repo_root)
if add_r.returncode != 0:
raise HTTPException(500, f"git add failed: {add_r.stderr}")
@@ -186,7 +238,7 @@ def git_commit(body: CommitBody, db: Session = Depends(get_db)):
if commit_r.returncode != 0:
out = commit_r.stdout + commit_r.stderr
if "nothing to commit" in out:
return {"status": "nothing_to_commit", "output": "Nothing to commit — diagrams unchanged since last commit"}
return {"status": "nothing_to_commit", "output": "Nothing to commit — diagram unchanged since last commit"}
raise HTTPException(500, f"git commit failed: {commit_r.stderr}")
return {"status": "committed", "output": commit_r.stdout.strip()}