diff --git a/pytm/pytm.py b/pytm/pytm.py index 169b3c5..9a5fb7c 100644 --- a/pytm/pytm.py +++ b/pytm/pytm.py @@ -128,7 +128,7 @@ def _print_components(classes): print("Elements:") _print_components(list(_iter_subclasses(Element))) - print("\nAtributes:") + print("\nAttributes:") enumerated = set(_iter_subclasses(OrderedEnum)) enumerated.update({Data, Action, Lifetime}) _print_components(list(enumerated)) @@ -139,6 +139,7 @@ def _print_components(classes): "Agent": Agent, "Actor": Actor, "Asset": Asset, + "Element": Element, "Boundary": Boundary, "Classification": Classification, "Data": Data, @@ -411,26 +412,6 @@ def _apply_response_defaults(self, flow: Dataflow) -> None: "isEncrypted", getattr(flow.source.controls, "isEncrypted", False) ) - def _color(self): - if self.inScope is True: - return "black" - else: - return "grey69" - - def display_name(self): - return self.name - - def _label(self): - return "\\n".join(wrap(html.escape(self.display_name()), 18)) - - def _shape(self): - return "square" - - def _safeset(self, attr, value): - try: - setattr(self, attr, value) - except ValueError: - pass def _apply_forward_defaults(self, flow: Dataflow) -> None: flow._safeset( "protocol", getattr(flow.sink, "protocol", getattr(flow, "protocol", ""))