From aaf51d698bde8f9b51ef52b7853142a68c5b966a 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:21:23 +0100
Subject: [PATCH] Edge dir method both directions
---
src/pyplaml/diagram_edge.py | 9 +++++++--
1 file changed, 7 insertions(+), 2 deletions(-)
diff --git a/src/pyplaml/diagram_edge.py b/src/pyplaml/diagram_edge.py
index 966e68a..afec5c7 100644
--- a/src/pyplaml/diagram_edge.py
+++ b/src/pyplaml/diagram_edge.py
@@ -45,9 +45,14 @@ class DiagramEdge(DiagramObject):
self.redraw()
def get_dir(self) -> Direction | None:
- if self.source_rel != Relation.NONE:
+ is_left = self.source_rel != Relation.NONE
+ is_right = self.target_rel != Relation.NONE
+
+ if is_left and is_right:
+ return None
+ if is_left:
return Direction.LEFT
- elif self.target_rel != Relation.NONE:
+ elif is_right:
return Direction.RIGHT
return None
--
GitLab