Coverage for /var/srv/projects/api.amasfac.comuna18.com/tmp/venv/lib/python3.9/site-packages/mptt/querysets.py: 65%

1from django.db import models 

2 

3from mptt import utils 

4 

5 

6class TreeQuerySet(models.query.QuerySet): 

7 def as_manager(cls): 

8 # Address the circular dependency between `Queryset` and `Manager`. 

9 from mptt.managers import TreeManager 

10 

11 manager = TreeManager.from_queryset(cls)() 

12 manager._built_with_as_manager = True 

13 return manager 

14 

15 as_manager.queryset_only = True 

16 as_manager = classmethod(as_manager) 

17 

18 def get_descendants(self, *args, **kwargs): 

19 """ 

20 Alias to `mptt.managers.TreeManager.get_queryset_descendants`. 

21 """ 

22 return self.model._tree_manager.get_queryset_descendants(self, *args, **kwargs) 

23 

24 get_descendants.queryset_only = True 

25 

26 def get_ancestors(self, *args, **kwargs): 

27 """ 

28 Alias to `mptt.managers.TreeManager.get_queryset_ancestors`. 

29 """ 

30 return self.model._tree_manager.get_queryset_ancestors(self, *args, **kwargs) 

31 

32 get_ancestors.queryset_only = True 

33 

34 def get_cached_trees(self): 

35 """ 

36 Alias to `mptt.utils.get_cached_trees`. 

37 """ 

38 return utils.get_cached_trees(self)