10 examples of 'django queryset first' in Python

15
def get_first(self, *args, **kwargs):
16
    objs = list(self.filter(*args, **kwargs)[:1])
17
    if not objs:
18
        return None
19
    return objs[0]

97
def get_queryset(self):
98
    return super(SuccessManager, self).get_queryset().filter(
99
        success=True)

42
def get_queryset(self):
43
    location = self.request.query_params.get('location')
44
    qs = self.queryset
45
    if location:
46
        qs = qs.filter(location=location)
47
    return qs

33
def test_queryset(self):
34
    res = self.client.get('/list/authors/')
35
    self.assertEqual(res.status_code, 200)
36
    self.assertTemplateUsed(res, 'generic_views/author_list.html')
37
    self.assertEqual(list(res.context['object_list']), list(Author.objects.all()))
38
    self.assertIsInstance(res.context['view'], View)
39
    self.assertIs(res.context['author_list'], res.context['object_list'])
40
    self.assertIsNone(res.context['paginator'])
41
    self.assertIsNone(res.context['page_obj'])
42
    self.assertFalse(res.context['is_paginated'])

338
def test_manager_last(self):
339
    self.assertIsNone(SomeModel.objects.last())
340
341
    SomeModel(p='foo', q=1)
342
    SomeModel(p='bar', q=2)
343
344
    obj = SomeModel.objects.last()
345
346
    self.assertEqual(obj.q, 2)

39
def get_object_or_none(klass, *args, **kwargs):
40
    """
41
        Uses get() to return an object or None if the object does not exist.
42
43
        klass may be a Model, Manager, or QuerySet object. All other passed
44
        arguments and keyword arguments are used in the get() query.
45
    """
46
47
    from django.shortcuts import _get_queryset
48
    queryset = _get_queryset(klass)
49
    try:
50
        return queryset.get(*args, **kwargs)
51
    except queryset.model.DoesNotExist:
52
        return None

29
def get_object_or_empty(klass, *args, **kwargs):
30
    """Raises ApiEmpty if the object cannot be found"""
31
    queryset = _get_queryset(klass)
32
    try:
33
        return queryset.get(*args, **kwargs)
34
    except queryset.model.DoesNotExist:
35
        raise ApiEmpty

69
def get_list_queryset(self):
70
    obj = self.get_detail_object()
71
    return getattr(obj, self.related_name).all()

145
def test_queryset_of_manager(self):
146
    qs = PublicDefault.generic.all()
147
    self.assertEqual(4, len(qs))
148
    qs = qs.public()
149
    self.assertEqual(1, len(qs))
150
151
    # previous bug with queryset clones
152
    qs = PublicDefault.generic.filter(pk__gt=0).public()
153
    self.assertEqual(1, len(qs))

181
def get_queryset(self, request):
182
    query_set = super().get_queryset(request)
183
    if request.user.is_superuser:
184
        return query_set
185
    return query_set.filter(pk=request.user.pk)