diff --git a/src/pyplaml/puml_parser.py b/src/pyplaml/puml_parser.py index 6a4ced13d41faed349ed305f470a17e5d6d2762f..2effa7f1bb8fa167bba573278841bf67e4a016a6 100644 --- a/src/pyplaml/puml_parser.py +++ b/src/pyplaml/puml_parser.py @@ -108,10 +108,15 @@ class PUMLParser(object): if text[0] == "<": edge.arrow_from_source = edge.get_dir() != Direction.LEFT text = text[1:] - + elif text[-1] == "<": + edge.arrow_from_source = edge.get_dir() != Direction.LEFT + text = text[:-1] elif text[0] == ">": edge.arrow_from_source = edge.get_dir() == Direction.RIGHT text = text[1:] + elif text[-1] == ">": + edge.arrow_from_source = edge.get_dir() == Direction.RIGHT + text = text[:-1] edge.edge_text = text