Django REST Framework + Django REST Swagger + ImageField

Django REST Framework + Django REST Swagger + ImageField

我创建了一个带有 ImageField 的简单模型,我想用 django-rest-framework + django-rest-swagger 创建一个 api 视图,该视图已记录并能够上传文件。

这是我得到的:

models.py

from django.utils import timezone
from django.db import models

class MyModel(models.Model):

    source = models.ImageField(upload_to=u'/photos')
    is_active = models.BooleanField(default=False)
    created_at = models.DateTimeField(default=timezone.now)

    def __unicode__(self):
        return u"photo {0}".format(self.source.url)

serializer.py

from .models import MyModel

class MyModelSerializer(serializers.ModelSerializer):

    class Meta:
        model = MyModel
        fields = [
            'id',
            'source',
            'created_at',
        ]

views.py

from rest_framework import generics
from .serializer import MyModelSerializer

class MyModelView(generics.CreateAPIView):
    serializer_class = MyModelSerializer
    parser_classes = (FileUploadParser, )

    def post(self, *args, **kwargs):
        """
            Create a MyModel
            ---
            parameters:
                - name: source
                  description: file
                  required: True
                  type: file
            responseMessages:
                - code: 201
                  message: Created
        """
        return super(MyModelView, self).post(self, *args, **kwargs)

urls.py

from weddings.api.views import MyModelView

urlpatterns = patterns(
    '',
    url(r'^/api/mymodel/$', MyModelView.as_view()),
)

对我来说这应该很简单。但是,我无法使上传工作。我总是收到此错误响应:

我已经从 django-rest-framework 阅读了文档的这一部分:

If the view used with FileUploadParser is called with a filename URL keyword argument, then that argument will be used as the filename. If it is called without a filename URL keyword argument, then the client must set the filename in the Content-Disposition HTTP header. For example Content-Disposition: attachment; filename=upload.jpg.

然而,Header 正在请求负载 属性 中由 django-rest-swagger 传递(来自 chrome 控制台)。

如果需要更多信息,请告诉我。

我正在使用 Django==1.8.8djangorestframework==3.3.2django-rest-swagger==0.3.4

我通过对您的代码进行一些更改来实现此功能。

首先,在models.py中,将ImageField名称更改为file,并使用相对路径上传文件夹。当您将文件作为二进制流上传时,它在文件键 (request.data.get('file')) 下的 request.data 字典中可用,因此最干净的选择是将它映射到具有相同名称的模型字段。

from django.utils import timezone
from django.db import models


class MyModel(models.Model):

    file = models.ImageField(upload_to=u'photos')
    is_active = models.BooleanField(default=False)
    created_at = models.DateTimeField(default=timezone.now)

    def __unicode__(self):
        return u"photo {0}".format(self.file.url)

serializer.py 中,将源字段重命名为文件:

class MyModelSerializer(serializers.ModelSerializer):

    class Meta:
        model = MyModel
        fields = ('id', 'file', 'created_at')

在views.py中,不调用super,而是调用create():

from rest_framework import generics
from rest_framework.parsers import FileUploadParser

from .serializer import MyModelSerializer


class MyModelView(generics.CreateAPIView):
    serializer_class = MyModelSerializer
    parser_classes = (FileUploadParser,)

    def post(self, request, *args, **kwargs):
        """
            Create a MyModel
            ---
            parameters:
                - name: file
                  description: file
                  required: True
                  type: file
            responseMessages:
                - code: 201
                  message: Created
        """
        return self.create(request, *args, **kwargs)

我已经使用 Postman Chrome 扩展程序对此进行了测试。我已将图像上传为二进制文件,并手动设置了两个 headers:

Content-Disposition: attachment; filename=upload.jpg
Content-Type: */*

根据我的经验,FileUploadParser 使用这种格式的请求:

    curl -X POST -H "Content-Type:multipart/form-data" \
                 -F "file=@{filename};type=image/jpg" \
                 https://endpoint.com/upload-uri/

您视图中的 request.data['file'] 将包含该文件。

也许如果你尝试 Content-Type:multipart/form-data header,你会很幸运。

这是我想出的最终解决方案:

from rest_framework import generics
from rest_framework.parsers import FormParser, MultiPartParser
from .serializer import MyModelSerializer

class MyModelView(generics.CreateAPIView):
    serializer_class = MyModelSerializer
    parser_classes = (FormParser, MultiPartParser)

    def post(self, *args, **kwargs):
        """
            Create a MyModel
            ---
            parameters:
                - name: source
                  description: file
                  required: True
                  type: file
            responseMessages:
                - code: 201
                  message: Created
        """
        return super(MyModelView, self).post(self, *args, **kwargs)

我所要做的就是将解析器从 FileUploadParser 更改为 (FormParser, MultiPartParser)