You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
19 lines
531 B
Python
19 lines
531 B
Python
4 years ago
|
from rest_framework.utils.encoders import JSONEncoder as DRFJSONEncoder
|
||
|
|
from rest_framework.renderers import JSONRenderer as DRFJSONRenderer
|
||
|
|
|
||
|
|
from .serializers import b64encode
|
||
|
|
|
||
|
|
|
||
|
|
class JSONEncoder(DRFJSONEncoder):
|
||
|
|
def default(self, obj):
|
||
|
|
if isinstance(obj, bytes) or isinstance(obj, memoryview):
|
||
|
|
return b64encode(obj)
|
||
|
|
return super().default(obj)
|
||
|
|
|
||
|
|
|
||
|
|
class JSONRenderer(DRFJSONRenderer):
|
||
|
|
"""
|
||
|
|
Renderer which serializes to JSON with support for our base64
|
||
|
|
"""
|
||
|
|
encoder_class = JSONEncoder
|