Skip to content

Instantly share code, notes, and snippets.

View slykar's full-sized avatar

Sylwester Kardziejonek slykar

View GitHub Profile
@slykar
slykar / polymorphic_tree.py
Last active December 21, 2023 01:26
Combining django-treebeard Materialized Path tree with django-model-utils InheritanceManager #django
from model_utils.managers import InheritanceManagerMixin, InheritanceQuerySetMixin
from treebeard.mp_tree import MP_NodeManager, MP_NodeQuerySet, MP_Node
class PolymorphicTreeQuerySet(InheritanceQuerySetMixin, MP_NodeQuerySet):
"""QuerySet combining functionality of polymorphic up-casting with Materialized Path tree structure"""
class PolymorphicTreeManager(InheritanceManagerMixin, MP_NodeManager):
_queryset_class = PolymorphicTreeQuerySet