From 15449ae24bd7bbc22b550e8c5d4339664b7f7ac5 Mon Sep 17 00:00:00 2001 From: Pavel 'LEdoian' Turinsky Date: Thu, 11 Jun 2020 22:51:55 +0000 Subject: [PATCH] TreeLib: Fix type check --- seminar/treelib.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/seminar/treelib.py b/seminar/treelib.py index 2fb53c8e..a17d9aee 100644 --- a/seminar/treelib.py +++ b/seminar/treelib.py @@ -229,7 +229,7 @@ def create_node_after(predecessor, type, **kwargs): from seminar.models import TreeNode if predecessor is None: raise TreeLibError("Nelze vyrábět sirotky! (predecessor=None)") - if not isinstance(type, TreeNode): + if not issubclass(type, TreeNode): raise TreeLibError("Nový node není node!") new_node = type.objects.create(**kwargs) new_node.root = predecessor.root @@ -246,7 +246,7 @@ def create_child(parent, type, **kwargs): from seminar.models import TreeNode if parent is None: raise TreeLibError("Nelze vyrábět sirotky! (parent=None)") - if not isinstance(type, TreeNode): + if not issubclass(type, TreeNode): raise TreeLibError("Nový node není node!") new_node = type.objects.create(**kwargs) new_node.root = parent.root @@ -284,7 +284,7 @@ def create_node_before(successor, type, **kwargs): from seminar.models import TreeNode if successor is None: raise TreeLibError("Nelze vyrábět sirotky! (successor=None)") - if not isinstance(type, TreeNode): + if not issubclass(type, TreeNode): raise TreeLibError("Nový node není node!") if safe_pred(successor) is not None: # Easy: přidáme za předchůdce