mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
24 lines
726 B
Python
24 lines
726 B
Python
from mptt.managers import TreeManager as TreeManager_
|
|
from mptt.querysets import TreeQuerySet as TreeQuerySet_
|
|
|
|
from django.db.models import Manager
|
|
from .querysets import RestrictedQuerySet
|
|
|
|
|
|
class TreeQuerySet(TreeQuerySet_, RestrictedQuerySet):
|
|
"""
|
|
Mate django-mptt's TreeQuerySet with our RestrictedQuerySet for permissions enforcement.
|
|
"""
|
|
pass
|
|
|
|
|
|
class TreeManager(Manager.from_queryset(TreeQuerySet), TreeManager_):
|
|
"""
|
|
Extend django-mptt's TreeManager to incorporate RestrictedQuerySet().
|
|
"""
|
|
def db_manager(self, using=None, hints=None):
|
|
manager = super().db_manager(using, hints)
|
|
|
|
# Return an unrestricted QuerySet for use by MPTT
|
|
return manager.unrestricted()
|