1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42
|
from django.utils.module_loading import import_string
def lazy_serializer(path: str):
""" simulate initiated object but actually load class and init on first usage """
class LazySerializer:
def __init__(self, *args, **kwargs):
self.lazy_args, self.lazy_kwargs, self.lazy_obj = args, kwargs, None
def __getattr__(self, item):
if not self.lazy_obj:
self.lazy_obj = import_string(path)(*self.lazy_args, **self.lazy_kwargs)
return getattr(self.lazy_obj, item)
@property # type: ignore
def __class__(self):
return self.__getattr__('__class__')
@property
def __dict__(self):
return self.__getattr__('__dict__')
def __str__(self):
return self.__getattr__('__str__')()
def __repr__(self):
return self.__getattr__('__repr__')()
return LazySerializer
def forced_singular_serializer(serializer_class):
from drf_spectacular.drainage import set_override
from drf_spectacular.utils import extend_schema_serializer
patched_serializer_class = type(serializer_class.__name__, (serializer_class,), {})
extend_schema_serializer(many=False)(patched_serializer_class)
set_override(patched_serializer_class, 'suppress_collision_warning', True)
return patched_serializer_class
|