Kyle
f32203b630
- FastAPI + SQLite backend with routers for diagrams, devices, wires, bundles, connectors, export, and Octopart/Nexar search - Konva.js canvas: ortho/direct/curved routing, wire crossings, harness mode, snap-to-grid, multi-select, drag, resize, undo - Wire features: color/stripe, twisted pair helix, shielded glow, gauge size label, multi-core bundle grouping, length/unit tracking - Device library: connector, terminal block, cable, splice, label, group, relay, fuse, switch, component with custom connector support - Exports: BOM CSV, assembly TXT, JSON, formboard layout, PNG image - Auto-migration for schema changes via ALTER TABLE at startup Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
104 lines
4.8 KiB
Python
104 lines
4.8 KiB
Python
from sqlalchemy import Column, Integer, String, Float, JSON, ForeignKey, DateTime, Text, Boolean
|
|
from sqlalchemy.orm import relationship
|
|
from .database import Base
|
|
import datetime
|
|
|
|
|
|
class Diagram(Base):
|
|
__tablename__ = "diagrams"
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
name = Column(String, nullable=False, default="Untitled Diagram")
|
|
description = Column(Text, default="")
|
|
created_at = Column(DateTime, default=datetime.datetime.utcnow)
|
|
updated_at = Column(DateTime, default=datetime.datetime.utcnow)
|
|
|
|
sheets = relationship("Sheet", back_populates="diagram", cascade="all, delete-orphan", order_by="Sheet.position")
|
|
devices = relationship("Device", back_populates="diagram", cascade="all, delete-orphan")
|
|
wires = relationship("Wire", back_populates="diagram", cascade="all, delete-orphan")
|
|
wire_bundles = relationship("WireBundle", back_populates="diagram", cascade="all, delete-orphan")
|
|
|
|
|
|
class WireBundle(Base):
|
|
__tablename__ = "wire_bundles"
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
diagram_id = Column(Integer, ForeignKey("diagrams.id"), nullable=False)
|
|
label = Column(String, default="")
|
|
jacket_color = Column(String, default="#2a2a2a")
|
|
|
|
diagram = relationship("Diagram", back_populates="wire_bundles")
|
|
wires = relationship("Wire", back_populates="bundle")
|
|
|
|
|
|
class Sheet(Base):
|
|
__tablename__ = "sheets"
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
diagram_id = Column(Integer, ForeignKey("diagrams.id", ondelete="CASCADE"), nullable=False)
|
|
name = Column(String, nullable=False, default="Sheet 1")
|
|
position = Column(Integer, default=0)
|
|
|
|
diagram = relationship("Diagram", back_populates="sheets")
|
|
devices = relationship("Device", back_populates="sheet", cascade="all, delete-orphan")
|
|
wires = relationship("Wire", back_populates="sheet", cascade="all, delete-orphan")
|
|
|
|
|
|
class Device(Base):
|
|
__tablename__ = "devices"
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
diagram_id = Column(Integer, ForeignKey("diagrams.id"), nullable=False)
|
|
sheet_id = Column(Integer, ForeignKey("sheets.id", ondelete="CASCADE"), nullable=True)
|
|
device_type = Column(String, nullable=False)
|
|
label = Column(String, default="")
|
|
reference = Column(String, default="")
|
|
x = Column(Float, default=100)
|
|
y = Column(Float, default=100)
|
|
width = Column(Float, default=120)
|
|
height = Column(Float, default=60)
|
|
properties = Column(JSON, default={})
|
|
pins = Column(JSON, default=[])
|
|
|
|
diagram = relationship("Diagram", back_populates="devices")
|
|
sheet = relationship("Sheet", back_populates="devices")
|
|
|
|
|
|
class Wire(Base):
|
|
__tablename__ = "wires"
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
diagram_id = Column(Integer, ForeignKey("diagrams.id"), nullable=False)
|
|
sheet_id = Column(Integer, ForeignKey("sheets.id", ondelete="CASCADE"), nullable=True)
|
|
label = Column(String, default="")
|
|
from_device_id = Column(Integer, ForeignKey("devices.id"), nullable=True)
|
|
from_pin = Column(String, default="")
|
|
to_device_id = Column(Integer, ForeignKey("devices.id"), nullable=True)
|
|
to_pin = Column(String, default="")
|
|
color_primary = Column(String, default="#FF0000")
|
|
color_stripe = Column(String, nullable=True)
|
|
gauge = Column(String, default="18 AWG")
|
|
length = Column(Float, nullable=True)
|
|
length_unit = Column(String, default="in")
|
|
notes = Column(Text, default="")
|
|
waypoints = Column(JSON, default=[])
|
|
twisted_pair = Column(Boolean, default=False)
|
|
twist_pitch = Column(Float, default=16.0)
|
|
shielded = Column(Boolean, default=False)
|
|
|
|
bundle_id = Column(Integer, ForeignKey("wire_bundles.id"), nullable=True)
|
|
show_size_label = Column(Boolean, default=False)
|
|
|
|
diagram = relationship("Diagram", back_populates="wires")
|
|
sheet = relationship("Sheet", back_populates="wires")
|
|
bundle = relationship("WireBundle", back_populates="wires", foreign_keys=[bundle_id])
|
|
|
|
|
|
class CustomConnector(Base):
|
|
__tablename__ = "custom_connectors"
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
name = Column(String, nullable=False)
|
|
manufacturer = Column(String, default="")
|
|
part_number = Column(String, default="")
|
|
category = Column(String, default="Custom")
|
|
description = Column(String, default="")
|
|
pin_count = Column(Integer, nullable=False, default=2)
|
|
pin_labels = Column(JSON, default=[])
|
|
created_at = Column(DateTime, default=datetime.datetime.utcnow)
|
|
updated_at = Column(DateTime, default=datetime.datetime.utcnow)
|