Python get_related_fields Examples

Example #1

File: importer.py Project: sdelements/django-multi-import

    def get_queryset(self):
        queryset = self.model.objects
        serializer = self.empty_serializer

        for field in serializers.get_related_fields(serializer):
            queryset = queryset.select_related(field.source)

        for field in serializers.get_many_related_fields(serializer):
            queryset = queryset.prefetch_related(field.source)

        return queryset

Example #2

File: test_helpers_serializers.py Project: sdelements/django-multi-import

    def test_returns_correct_related_fields(self):
        serializer = BookSerializer()
        fields = serializers.get_related_fields(serializer)

        self.assertEqual(fields[0].field_name, 'author')