feat: add Tree relationships

nick-solly · Jan 3, 2024 · 29eb67d · 29eb67d
1 parent 12027c3
commit 29eb67d
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/django_diagram/app.py b/django_diagram/app.py
@@ -30,6 +30,18 @@ class App:
             FirstToSecondCardinality.EXACTLY_ONE,
             SecondToFirstCardinality.EXACTLY_ONE,
         ),
+        "TreeForeignKey": (
+            FirstToSecondCardinality.ONE_OR_MORE,
+            SecondToFirstCardinality.EXACTLY_ONE,
+        ),
+        "TreeManyToManyField": (
+            FirstToSecondCardinality.ONE_OR_MORE,
+            SecondToFirstCardinality.ONE_OR_MORE,
+        ),
+        "TreeOneToOneField": (
+            FirstToSecondCardinality.EXACTLY_ONE,
+            SecondToFirstCardinality.EXACTLY_ONE,
+        ),
     }
 
     DEFAULT_TITLE = "Django ER Diagram"