时间戳

首页 > 解决方案 > 在 Django Rest 框架中获取图像字段绝对路径 - 非请求流

python - 在 Django Rest 框架中获取图像字段绝对路径 - 非请求流

问题描述

我有一个周期性的 celery 任务，它需要将对象的表示存储在特定的 json 字段中。

这是简化的模型结构。父 <-- ChildWrapper <-- 子图像

所以基本上我有一个'ChildImage'模型指的是'ChildWrapper'，它又指的是'Parent'。

class Parent(TimeStampedModel):
    label = models.CharField(max_length=30, unique=True)
    live_content = JSONField(blank=True, null=True)
    is_template = models.BooleanField(default=False)
    reference_image = models.ImageField(upload_to=get_web_row_reference_image_path, blank=True, null=True)
    # Around 8 Other Fields

    def __str__(self):
        return '%s' % self.label


class ChildWrapper(TimeStampedModel):
    name = models.CharField(max_length=25, blank=True, null=True)
    row = models.ForeignKey(Parent, on_delete=models.CASCADE, related_name='web_column')
    order = models.PositiveIntegerField(default=0)
    # Around 20 Other Fields

    def __str__(self):
        return '%s' % self.name


class ChildImage(TimeStampedModel):
    image = models.ImageField(upload_to=get_web_image_path)
    column = models.ForeignKey(ChildWrapper, on_delete=models.CASCADE, related_name='web_image')
    # Around 10 Other Fields
 
    def __str__(self):
        return '%s' % self.column

这是为模型定义的序列化程序。

class ChildImageSerializer(serializers.ModelSerializer):
    class Meta:
        model = ChildImage
        fields = '__all__'

class ChildWrapperSerializer(serializers.ModelSerializer):
    web_image = ChildImageSerializer(read_only=True, many=True)
    class Meta:
        model = ChildWrapper
        fields = '__all__'

class ParentSerializer(serializers.ModelSerializer):
    web_column = ChildWrapperSerializer(many=True, read_only=True)
    class Meta:
        model = Parent
        fields = '__all__'

这是执行所需的定期芹菜任务

@app.task(bind=True)
def update_data(self):
    # Get Parent By a condition.
    parent = Parent.objects.filter(to_update=True).first()

    parent.live_content = None
    parent.live_content = ParentSerializer(parent).data
    print(parent.live_content)
    parent.save()

上面的任务得到子图像的输出，像这样，图像字段是相对路径而不是绝对路径。

{
    "id": 1
    "image": '/api/col/info.jpg'
}

有什么方法可以获取图像字段的绝对路径？

{
    "id": 1
    "image": "http://localhost:8000/admin/media/api/col/info.jpg"
}

PS： 我不能将请求上下文作为ParentSerializer(parent, context={'request': request})传递给序列化程序，因为这里不涉及请求对象。

标签： pythondjangodjango-rest-frameworkcelery

解决方案

我认为您有两种方法可以解决此问题。

第一个，是通过请求。您可以采用这种方法：

class ChildImageSerializer(serializers.ModelSerializer):
    img_url = serializers.SerializerMethodField()
    class Meta:
        model = ChildImage
        fields = '__all__'

    def get_img_url(self, obj):
        return self.context['request'].build_absolute_uri(obj.image.url)

class ChildWrapperSerializer(serializers.ModelSerializer):
    web_image = serializers.SerializerMethodField()
    class Meta:
        model = ChildWrapper
        fields = '__all__'

    def get_web_image(self, obj):
        serializer_context = {'request': self.context.get('request') }
        children = ChildImage.objects.filter(row=obj)
        serializer = ChildImageSerializer(children, many=True, context=serializer_context)
        return serializer.data

class ParentSerializer(serializers.ModelSerializer):
    web_column = serializers.SerializerMethodField()
    class Meta:
        model = Parent
        fields = '__all__'

    def get_web_column(self, obj):
        serializer_context = {'request': self.context.get('request') }
        children = ChildWrapper.objects.filter(row=obj)
        serializer = ChildWrapperSerializer(children, many=True, context=serializer_context)
        return serializer.data

在这里，我SerializerMethodField用来将请求传递给下一个序列化程序。

第二种方法是使用Django Sites Framework（@dirkgroten 提到）。您可以执行以下操作：

class ChildImageSerializer(serializers.ModelSerializer):
    img_url = serializers.SerializerMethodField()
    class Meta:
        model = ChildImage
        fields = '__all__'

    def get_img_url(self, obj):
        return 'http://%s%s%s' % (Site.objects.get_current().domain, settings.MEDIA_URL, obj.img.url)

更新：我完全错过了芹菜部分。对于生产，我认为您不必担心它们在 S3 中，绝对路径应该来自obj.image.url. 在 dev 和 stage 中，您可以使用给定的示例获取绝对路径。所以，试试这样：

class ChildImageSerializer(serializers.ModelSerializer):
    img_url = serializers.SerializerMethodField()
    class Meta:
        model = ChildImage
        fields = '__all__'

    def get_img_url(self, obj):
        if settings.DEBUG:  # debug enabled for dev and stage
            return 'http://%s%s%s' % (Site.objects.get_current().domain, settings.MEDIA_URL, obj.img.url)
        return obj.img.url

或者，有一种方法可以django-crequest在 celery 中使用请求，但我不确定它是否方便您。

推荐阅读