20 lines
636 B
Python
20 lines
636 B
Python
"""GVS 自定义 JSON 编码器与渲染器——处理 UUIDObj 等非标准类型的序列化"""
|
|
|
|
from rest_framework.renderers import JSONRenderer as _DRFJSONRenderer
|
|
from rest_framework.utils.encoders import JSONEncoder as _DRFJSONEncoder
|
|
from gvsdsdk.model_utils import UUIDObj
|
|
|
|
|
|
class JSONEncoder(_DRFJSONEncoder):
|
|
"""扩展 DRF 的 JSONEncoder，支持 UUIDObj 序列化"""
|
|
|
|
def default(self, obj):
|
|
if isinstance(obj, UUIDObj):
|
|
return obj.str
|
|
return super().default(obj)
|
|
|
|
|
|
class JSONRenderer(_DRFJSONRenderer):
|
|
"""使用自定义 JSONEncoder 的渲染器"""
|
|
encoder_class = JSONEncoder
|