Python SelectRelated Examples

Example #1

File: test_qs.py Project: maxipavlovic/django-rql

def test_npr_apply_multi():
    qs = NPR(Prefetch('publisher'), 'publisher').rebuild(SelectRelated('author')).apply(default_qs)

    assert len(qs._prefetch_related_lookups) == 2
    assert qs._prefetch_related_lookups[1] == 'author__publisher'

    pr_obj = qs._prefetch_related_lookups[0]
    assert isinstance(pr_obj, Prefetch)
    assert pr_obj.prefetch_through == pr_obj.prefetch_to == 'author__publisher'

Example #2

File: filters.py Project: sdurgi17/django_rql_select_example

class ProductFilters(RQLFilterClass):
    MODEL = Product
    SELECT = True
    FILTERS = (
        {
            'filter': 'id',
            'ordering': True,
        },
        {
            'filter': 'name',
            'search': True,
        },
        {
            'namespace': 'category',
            'filters': ('id', 'name'),
            'qs': SelectRelated('category'),
        },
        {
            'namespace': 'company',
            'filters': ('id', 'name'),
            'hidden': True,
            'qs': SelectRelated('company'),
        },
    )

Example #3

File: test_qs.py Project: maxipavlovic/django-rql

def test_sr_apply():
    qs = SelectRelated('author', 'author__publisher').apply(default_qs)
    assert qs.query.select_related == {'author': {'publisher': {}}}

Example #4

File: test_qs.py Project: maxipavlovic/django-rql

def test_chain_parent_sr():
    qs = Chain(NSR('publisher')).rebuild(Chain(SelectRelated('author'))).apply(default_qs)
    assert qs.query.select_related == {'author': {'publisher': {}}}

Example #5

File: test_qs.py Project: maxipavlovic/django-rql

def test_chain_without_parent():
    qs = Chain(SelectRelated('author__publisher'), SelectRelated('author')).apply(default_qs)
    assert qs.query.select_related == {'author': {'publisher': {}}}

Example #6

File: test_qs.py Project: maxipavlovic/django-rql

def test_nsr_apply_sr_parent():
    qs = NestedSelectRelated('publisher').rebuild(SelectRelated('author')).apply(default_qs)
    assert qs.query.select_related == {'author': {'publisher': {}}}

Example #7

File: test_qs.py Project: maxipavlovic/django-rql

    assert qs._prefetch_related_lookups == ('pages',)


def test_nested_prefetch_object_in_parent():
    with pytest.raises(AssertionError) as e:
        NestedPrefetchRelated('pages').rebuild(PrefetchRelated(Prefetch('pages')))

    assert str(e.value) == 'Only simple parent relations are supported.'


def test_npr_apply_no_parent():
    qs = NestedPrefetchRelated('pages').apply(default_qs)
    assert qs._prefetch_related_lookups == ('pages',)


@pytest.mark.parametrize('parent', (PrefetchRelated('author'), SelectRelated('author')))
def test_npr_apply_parent(parent):
    qs = NestedPrefetchRelated('publisher').rebuild(parent).apply(default_qs)
    assert qs._prefetch_related_lookups == ('author__publisher',)


def test_npr_apply_multi():
    qs = NPR(Prefetch('publisher'), 'publisher').rebuild(SelectRelated('author')).apply(default_qs)

    assert len(qs._prefetch_related_lookups) == 2
    assert qs._prefetch_related_lookups[1] == 'author__publisher'

    pr_obj = qs._prefetch_related_lookups[0]
    assert isinstance(pr_obj, Prefetch)
    assert pr_obj.prefetch_through == pr_obj.prefetch_to == 'author__publisher'

Example #8

def test_nested_prefetch_object_in_parent():
    with pytest.raises(AssertionError) as e:
        NestedPrefetchRelated('pages').rebuild(
            PrefetchRelated(Prefetch('pages')))

    assert str(e.value) == 'Only simple parent relations are supported.'


def test_npr_apply_no_parent():
    qs = NestedPrefetchRelated('pages').apply(default_qs)
    assert qs._prefetch_related_lookups == ('pages', )


@pytest.mark.parametrize('parent',
                         (PrefetchRelated('author'), SelectRelated('author')))
def test_npr_apply_parent(parent):
    qs = NestedPrefetchRelated('publisher').rebuild(parent).apply(default_qs)
    assert qs._prefetch_related_lookups == ('author__publisher', )


def test_npr_apply_multi():
    qs = NPR(Prefetch('publisher'),
             'publisher').rebuild(SelectRelated('author')).apply(default_qs)

    assert len(qs._prefetch_related_lookups) == 2
    assert qs._prefetch_related_lookups[1] == 'author__publisher'

    pr_obj = qs._prefetch_related_lookups[0]
    assert isinstance(pr_obj, Prefetch)
    assert pr_obj.prefetch_through == pr_obj.prefetch_to == 'author__publisher'