49 lines
1.8 KiB
Python
49 lines
1.8 KiB
Python
# coding=utf-8
|
||
"""
|
||
@project: maxkb
|
||
@Author：虎
|
||
@file： image_serializers.py
|
||
@date：2024/4/22 16:36
|
||
@desc:
|
||
"""
|
||
import uuid
|
||
|
||
from django.db.models import QuerySet
|
||
from django.http import HttpResponse
|
||
from rest_framework import serializers
|
||
|
||
from common.exception.app_exception import NotFound404
|
||
from common.field.common import UploadedImageField
|
||
from common.util.field_message import ErrMessage
|
||
from dataset.models import Image
|
||
from django.utils.translation import gettext_lazy as _
|
||
|
||
|
||
class ImageSerializer(serializers.Serializer):
|
||
image = UploadedImageField(required=True, error_messages=ErrMessage.image(_('image')))
|
||
|
||
def upload(self, with_valid=True):
|
||
if with_valid:
|
||
self.is_valid(raise_exception=True)
|
||
image_id = uuid.uuid1()
|
||
image = Image(id=image_id, image=self.data.get('image').read(), image_name=self.data.get('image').name)
|
||
image.save()
|
||
return f'/api/image/{image_id}'
|
||
|
||
class Operate(serializers.Serializer):
|
||
id = serializers.UUIDField(required=True)
|
||
|
||
def get(self, with_valid=True):
|
||
if with_valid:
|
||
self.is_valid(raise_exception=True)
|
||
image_id = self.data.get('id')
|
||
image = QuerySet(Image).filter(id=image_id).first()
|
||
if image is None:
|
||
raise NotFound404(404, _('Image not found'))
|
||
if image.image_name.endswith('.svg'):
|
||
return HttpResponse(image.image, status=200, headers={'Content-Type': 'image/svg+xml'})
|
||
# gif
|
||
elif image.image_name.endswith('.gif'):
|
||
return HttpResponse(image.image, status=200, headers={'Content-Type': 'image/gif'})
|
||
return HttpResponse(image.image, status=200, headers={'Content-Type': 'image/png'})
|