django serialization hack

serialization.py

from django.db import models
from django.db.models.query import ModelIterable


class BaseSerializer:
    SELECT_RELATED = []
    PREFETCH_RELATED = []

    def prepare_queryset(self, qs):
        if self.SELECT_RELATED:
            qs = qs.select_related(*self.SELECT_RELATED)

        if self.PREFETCH_RELATED:
            qs = qs.prefetch_related(*self.PREFETCH_RELATED)
        return qs

    def prepare_object(self, obj):
        obj.serialize = lambda: self.serialize(obj)

    def serialize(self, obj):
        return obj.to_dict_json()


class _SerializableModelIterable(ModelIterable):
    @classmethod
    def factory(cls, serializer):
        def new(*args, **kwargs):
            self = cls(*args, **kwargs)
            self._serializer = serializer
            return self
        return new

    def __iter__(self):
        for obj in super().__iter__():
            self._serializer.prepare_object(obj)
            yield obj


class SerializableQuerySet(models.QuerySet):
    def to_serialize(self, serializer_cls):
        serializer = serializer_cls()
        self._iterable_class = _SerializableModelIterable.factory(serializer)
        return serializer.prepare_queryset(self)