Removed useless base classes

Based on discussion here: neo4j-contrib#716 (comment)
73VW · Aug 10, 2023 · 74bf4bd · 74bf4bd
1 parent d6ada57
commit 74bf4bd
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 15 deletions.
diff --git a/neomodel/base.py b/neomodel/base.py
diff --git a/neomodel/core.py b/neomodel/core.py
@@ -5,7 +5,6 @@
 from neo4j.exceptions import ClientError
 
 from neomodel import config
-from neomodel.base import BaseNode
 from neomodel.exceptions import DoesNotExist, NodeClassAlreadyDefined
 from neomodel.hooks import hooks
 from neomodel.properties import Property, PropertyManager
@@ -315,7 +314,7 @@ def build_class_registry(cls):
 NodeBase = NodeMeta("NodeBase", (PropertyManager,), {"__abstract_node__": True})
 
 
-class StructuredNode(NodeBase, BaseNode):
+class StructuredNode(NodeBase):
     """
     Base class for all node definitions to inherit from.
 

diff --git a/neomodel/relationship.py b/neomodel/relationship.py
@@ -1,5 +1,3 @@
-from neomodel.base import BaseRel
-
 from .core import db
 from .hooks import hooks
 from .properties import Property, PropertyManager
@@ -22,7 +20,7 @@ def __new__(mcs, name, bases, dct):
 StructuredRelBase = RelationshipMeta("RelationshipBase", (PropertyManager,), {})
 
 
-class StructuredRel(StructuredRelBase, BaseRel):
+class StructuredRel(StructuredRelBase):
     """
     Base class for relationship objects
     """

diff --git a/neomodel/util.py b/neomodel/util.py
@@ -16,7 +16,6 @@
 from neo4j.graph import Relationship
 
 from neomodel import config, core
-from neomodel.base import BaseNode, BaseRel
 from neomodel.exceptions import (
     ConstraintValidationFailed,
     FeatureNotSupported,
@@ -66,9 +65,7 @@ def clear_neo4j_database(db, clear_constraints=False, clear_indexes=False):
 
 class Path(NeoPath):
     def __init__(self, nodes, *relationships):
-        assert all([isinstance(node, BaseNode) for node in nodes])
         for i, relationship in enumerate(relationships, start=1):
-            assert isinstance(relationship, BaseRel)
             start = relationship.start_node()
             end = relationship.end_node()
             if start not in nodes and end not in nodes: