From cf57f0e0cd193a94e747c8c044de26f0c9826419 Mon Sep 17 00:00:00 2001
From: KV <kvid@users.noreply.github.com>
Date: Thu, 30 Jul 2020 01:21:58 +0200
Subject: [PATCH] Add an image attribute to both Connectors and Cables

---
 src/wireviz/DataClasses.py | 2 ++
 src/wireviz/Harness.py     | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/src/wireviz/DataClasses.py b/src/wireviz/DataClasses.py
index 6b3b1e3..2420683 100644
--- a/src/wireviz/DataClasses.py
+++ b/src/wireviz/DataClasses.py
@@ -18,6 +18,7 @@ class Connector:
     type: Optional[str] = None
     subtype: Optional[str] = None
     pincount: Optional[int] = None
+    image: Optional[str] = None
     notes: Optional[str] = None
     pinlabels: List[Any] = field(default_factory=list)
     pins: List[Any] = field(default_factory=list)
@@ -91,6 +92,7 @@ class Cable:
     color: Optional[str] = None
     wirecount: Optional[int] = None
     shield: bool = False
+    image: Optional[str] = None
     notes: Optional[str] = None
     colors: List[Any] = field(default_factory=list)
     color_code: Optional[str] = None
diff --git a/src/wireviz/Harness.py b/src/wireviz/Harness.py
index 635fae9..1413e42 100644
--- a/src/wireviz/Harness.py
+++ b/src/wireviz/Harness.py
@@ -98,6 +98,7 @@ class Harness:
                      f'{connector.pincount}-pin' if connector.show_pincount else None,
                      connector.color, '<!-- colorbar -->' if connector.color else None],
                     '<!-- connector table -->' if connector.style != 'simple' else None,
+                    [f'<img src="{connector.image}"/>' if connector.image else None],
                     [html_line_breaks(connector.notes)]]
             html.extend(nested_html_table(rows))
 
@@ -173,6 +174,7 @@ class Harness:
                      f'{cable.length} m' if cable.length > 0 else None,
                      cable.color, '<!-- colorbar -->' if cable.color else None],
                     '<!-- wire table -->',
+                    [f'<img src="{cable.image}"/>' if cable.image else None],
                     [html_line_breaks(cable.notes)]]
             html.extend(nested_html_table(rows))