diff --git a/src/wireviz/DataClasses.py b/src/wireviz/DataClasses.py index 51a4a99..b024e18 100644 --- a/src/wireviz/DataClasses.py +++ b/src/wireviz/DataClasses.py @@ -116,6 +116,10 @@ class Connector: if len(loop) != 2: raise Exception('Loops must be between exactly two pins!') + for additional_component in self.additional_components: + if 'type' not in additional_component: + raise Exception('Additional components must have a type specified') + def activate_pin(self, pin): self.visible_pins[pin] = True @@ -200,6 +204,10 @@ class Cable: else: raise Exception('lists of part data are only supported for bundles') + for additional_component in self.additional_components: + if 'type' not in additional_component: + raise Exception('Additional components must have a type specified') + def connect(self, from_name, from_pin, via_pin, to_name, to_pin): from_pin = int2tuple(from_pin)