44 lines
1.7 KiB
Python
44 lines
1.7 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.response import result
|
|||
|
|
from dataset.serializers.file_serializers import FileSerializer
|
|||
|
|
|
|||
|
|
|
|||
|
|
class FileView(APIView):
|
|||
|
|
authentication_classes = [TokenAuth]
|
|||
|
|
parser_classes = [MultiPartParser]
|
|||
|
|
|
|||
|
|
@action(methods=['POST'], detail=False)
|
|||
|
|
@swagger_auto_schema(operation_summary="上传文件",
|
|||
|
|
operation_id="上传文件",
|
|||
|
|
manual_parameters=[openapi.Parameter(name='file',
|
|||
|
|
in_=openapi.IN_FORM,
|
|||
|
|
type=openapi.TYPE_FILE,
|
|||
|
|
required=True,
|
|||
|
|
description='上传文件')],
|
|||
|
|
tags=["文件"])
|
|||
|
|
def post(self, request: Request):
|
|||
|
|
return result.success(FileSerializer(data={'file': request.FILES.get('file')}).upload())
|
|||
|
|
|
|||
|
|
class Operate(APIView):
|
|||
|
|
@action(methods=['GET'], detail=False)
|
|||
|
|
@swagger_auto_schema(operation_summary="获取图片",
|
|||
|
|
operation_id="获取图片",
|
|||
|
|
tags=["文件"])
|
|||
|
|
def get(self, request: Request, file_id: str):
|
|||
|
|
return FileSerializer.Operate(data={'id': file_id}).get()
|