48 lines
1.9 KiB
Python
48 lines
1.9 KiB
Python
# coding=utf-8
|
||
"""
|
||
@project: maxkb
|
||
@Author：虎
|
||
@file： image.py
|
||
@date：2024/4/22 16:23
|
||
@desc:
|
||
"""
|
||
from drf_yasg import openapi
|
||
from drf_yasg.utils import swagger_auto_schema
|
||
from rest_framework.decorators import action
|
||
from rest_framework.parsers import MultiPartParser
|
||
from rest_framework.views import APIView
|
||
from rest_framework.views import Request
|
||
|
||
from common.auth import TokenAuth
|
||
from common.log.log import log
|
||
from common.response import result
|
||
from dataset.serializers.image_serializers import ImageSerializer
|
||
from django.utils.translation import gettext_lazy as _
|
||
|
||
|
||
class Image(APIView):
|
||
authentication_classes = [TokenAuth]
|
||
parser_classes = [MultiPartParser]
|
||
|
||
@action(methods=['POST'], detail=False)
|
||
@swagger_auto_schema(operation_summary=_('Upload image'),
|
||
operation_id=_('Upload image'),
|
||
manual_parameters=[openapi.Parameter(name='file',
|
||
in_=openapi.IN_FORM,
|
||
type=openapi.TYPE_FILE,
|
||
required=True,
|
||
description=_('Upload image'))],
|
||
tags=[_('Image')])
|
||
@log(menu=_('Image'), operate=_('Upload image'))
|
||
def post(self, request: Request):
|
||
return result.success(ImageSerializer(data={'image': request.FILES.get('file')}).upload())
|
||
|
||
class Operate(APIView):
|
||
@action(methods=['GET'], detail=False)
|
||
@swagger_auto_schema(operation_summary=_('Get Image'),
|
||
operation_id=_('Get Image'),
|
||
tags=[_('Image')])
|
||
@log(menu=_('Image'), operate=_('Get Image'))
|
||
def get(self, request: Request, image_id: str):
|
||
return ImageSerializer.Operate(data={'id': image_id}).get()
|