mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
26 lines
633 B
Python
26 lines
633 B
Python
|
from tree_queries.query import TreeManager as TreeManager_
|
||
|
|
from tree_queries.query import TreeQuerySet as TreeQuerySet_
|
||
|
|
|
||
|
|
from django.db.models import Manager
|
||
|
|
from .querysets import RestrictedQuerySet
|
||
|
|
|
||
|
|
__all__ = (
|
||
|
|
'TreeManager',
|
||
|
|
'TreeQuerySet',
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
class TreeQuerySet(TreeQuerySet_, RestrictedQuerySet):
|
||
|
|
"""
|
||
|
|
Mate django-tree-queries TreeQuerySet with our RestrictedQuerySet for permissions enforcement.
|
||
|
|
"""
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
class TreeManager(Manager.from_queryset(TreeQuerySet), TreeManager_):
|
||
|
|
"""
|
||
|
|
Extend django-tree-queries TreeManager to incorporate RestrictedQuerySet().
|
||
|
|
"""
|
||
|
|
|
||
|
|
_with_tree_fields = True
|