43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
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
|