Add support for wire length units

kvid · Oct 5, 2020 · 1a973df · 1a973df
1 parent 55d8dba
commit 1a973df
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/wireviz/DataClasses.py b/src/wireviz/DataClasses.py
@@ -88,6 +88,7 @@ class Cable:
     gauge_unit: Optional[str] = None
     show_equiv: bool = False
     length: float = 0
+    lengthunit: Optional[str] = None      
     color: Optional[str] = None
     wirecount: Optional[int] = None
     shield: bool = False
@@ -119,6 +120,9 @@ def __post_init__(self):
 
         self.connections = []
 
+        if self.lengthunit is None: #Default wire length units to meters if left undeclared
+            self.lengthunit = 'm'
+
         if self.wirecount:  # number of wires explicitly defined
             if self.colors:  # use custom color palette (partly or looped if needed)
                 pass