Update the types of dataclass attributes according to usage

Fixes wireviz#156
kvid · Sep 1, 2020 · 0d2ad00 · 0d2ad00
1 parent df90d83
commit 0d2ad00
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/wireviz/DataClasses.py b/src/wireviz/DataClasses.py
@@ -22,8 +22,8 @@ class Connector:
  pinlabels: List[Any] = field(default_factory=list)
  pins: List[Any] = field(default_factory=list)
  color: Optional[str] = None
- show_name: bool = None
- show_pincount: bool = None
+ show_name: Optional[bool] = None
+ show_pincount: Optional[bool] = None
  hide_disconnected_pins: bool = False
  autogenerate: bool = False
  loops: List[Any] = field(default_factory=list)
@@ -90,7 +90,7 @@ class Cable:
  length: float = 0
  color: Optional[str] = None
  wirecount: Optional[int] = None
- shield: bool = False
+ shield: Union[bool, str] = False # False | True | color
  notes: Optional[str] = None
  colors: List[Any] = field(default_factory=list)
  color_code: Optional[str] = None