From 67760737cc81c57b35fb0755fad98821297356b3 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tom=C3=A1=C5=A1=20Orvisk=C3=BD?= <orvisky.tomas@centrum.cz>
Date: Sun, 28 Jan 2024 13:26:34 +0100
Subject: [PATCH] Label arrows also on end
---
src/pyplaml/puml_parser.py | 7 ++++++-
1 file changed, 6 insertions(+), 1 deletion(-)
diff --git a/src/pyplaml/puml_parser.py b/src/pyplaml/puml_parser.py
index 6a4ced1..2effa7f 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
--
GitLab