Exemplo n.º 1
0
    def test_manager_from_queryset_succeed(self):
        class TestQuerySet(QuerySet):
            def shared_method(self):
                return 'queryset'

        class TestManager(models.Manager):
            def manager_only_method(self):
                return 'manager'

        CustomManager = compat.manager_from_queryset(TestManager, TestQuerySet)
        self.assertEqual(CustomManager.__name__, 'TestManagerFromTestQuerySet')

        manager = CustomManager()
        manager.model = Process

        self.assertEqual(manager.shared_method(), 'queryset')
        self.assertEqual(manager.manager_only_method(), 'manager')
        self.assertEqual(manager.get_queryset().shared_method(), 'queryset')
        self.assertFalse(hasattr(manager.get_queryset(), 'manager_only_method'))
Exemplo n.º 2
0
    def test_manager_from_queryset_succeed(self):
        class TestQuerySet(QuerySet):
            def shared_method(self):
                return 'queryset'

        class TestManager(models.Manager):
            def manager_only_method(self):
                return 'manager'

        CustomManager = compat.manager_from_queryset(TestManager, TestQuerySet)
        self.assertEqual(CustomManager.__name__, 'TestManagerFromTestQuerySet')

        manager = CustomManager()
        manager.model = Process

        self.assertEqual(manager.shared_method(), 'queryset')
        self.assertEqual(manager.manager_only_method(), 'manager')
        self.assertEqual(manager.get_queryset().shared_method(), 'queryset')
        self.assertFalse(hasattr(manager.get_queryset(),
                                 'manager_only_method'))
Exemplo n.º 3
0
from django.db.models import Manager, Q
from guardian.shortcuts import get_objects_for_user
from viewflow.compat import manager_from_queryset
from viewflow.managers import TaskQuerySet as BaseTaskQuerySet


class TaskQuerySet(BaseTaskQuerySet):
    def user_queue(self, user, flow_cls=None):
        """
        List of tasks permitted for user
        """
        queryset = self.filter(flow_task_type='HUMAN')

        if flow_cls is not None:
            queryset = queryset.filter(process__flow_cls=flow_cls)

        user_planets = get_objects_for_user(user, 'parcel.land_on_planet')
        has_permission = Q(owner__isnull=True, process__parcel__planet__in=user_planets) \
            | Q(owner=user)

        queryset = queryset.filter(has_permission)

        return queryset

TaskManager = manager_from_queryset(Manager, TaskQuerySet)