Coverage for utilities/mptt.py: 100%

1from django.db.models import Manager 

2from mptt.managers import TreeManager as TreeManager_ 

3from mptt.querysets import TreeQuerySet as TreeQuerySet_ 

4 

5from .querysets import RestrictedQuerySet 

6 

7__all__ = ( 

8 'TreeManager', 

9 'TreeQuerySet', 

10) 

11 

12 

13class TreeQuerySet(TreeQuerySet_, RestrictedQuerySet): 

14 """ 

15 Mate django-mptt's TreeQuerySet with our RestrictedQuerySet for permissions enforcement. 

16 """ 

17 pass 

18 

19 

20class TreeManager(Manager.from_queryset(TreeQuerySet), TreeManager_): 

21 """ 

22 Extend django-mptt's TreeManager to incorporate RestrictedQuerySet(). 

23 """ 

24 pass