如何使用 S3Boto3Storage 在 django 中上传图像，其中路径是从 django 中的视图动态设置的

问题描述

我想在 django 中使用 session_key 作为存储桶内的目录将图像保存到 S3 存储桶。

我创建了一个测试页面，用于将图像上传到存储桶中的设置位置，但不知道如何使用 session_key 动态设置上传位置。

我查看了 django-storages 的文档，如果不是因为我使用的是 ModelForm，我可以找到一种方法。

这是我的代码（我省略了带有存储桶名称和凭据的 settings.py）：

storage_backends.py

from storages.backends.s3boto3 import S3Boto3Storage

class TestS3MediaStorage(S3Boto3Storage):
    location = 'dev/'
    default_acl = 'public-read'
    file_overwrite = False

models.py

from .storage_backends import TestS3MediaStorage

class TestS3Upload(models.Model):
    uploaded_at = models.DateTimeField(auto_Now_add=True)
    file = models.FileField(storage=TestS3MediaStorage())

forms.py

from .models import TestS3Upload

class TestS3UploadForm(forms.ModelForm):

    class Meta:
        model = TestS3Upload
        fields = ['file']

views.py

from django.shortcuts import render
from django.http import HttpResponse

from .forms import TestS3UploadForm

def test_s3_upload(request):

    # create session if it doesn't already exist
    if not request.session.session_key:
        request.session.create()

    # not quite sure how to use this to set upload destination
    session_key = request.session.session_key

    if request.method == 'POST':

        form = TestS3UploadForm(request.POST,request.FILES)

        if form.is_valid():
            form.save()
            return HttpResponse("upload successful!")
    else:
        form = TestS3UploadForm()

    return render(
        request,'uploader/test_s3_upload.html',{
            'form': form
        }
    )

test_s3_upload.html

<h1>Test S3 file upload</h1>
<form method="POST" enctype="multipart/form-data">
    {% csrf_token %}
    {{ form.as_p }}
    <button type="submit">Upload photo to S3 Bucket</button>
</form>

当我运行我的代码并上传图像时，例如 car.jpg，它上传成功但 S3 存储桶内的路径是

<bucket-name>/dev/car.jpg

我想要

<bucket-name>/dev/<session-key>/car.jpg

需要的包是 boto3 和 django-storages，以防任何想要帮助回答的人需要知道。

解决方法

我想出了一种使用会话密钥将图像/文件上传到 s3 到目录的方法。它也适用于一般上传文件，而不仅仅是 s3。

首先，我向模型添加了一个属性来存储会话密钥。

class TestS3Upload(models.Model):
    session_key = models.CharField(max_length=50,null=False,blank=False)
    ...

然后我在模型表单上包含了一个隐藏字段，我在视图中预先填充了 session_key 值。

forms.py

class TestS3UploadForm(forms.ModelForm):

    class Meta:
        model = TestS3Upload
        fields = ['file','session_key']
        widgets = {'session_key': forms.HiddenInput()}

views.py

def test_s3_upload(request):
    # create session if it doesn't already exist
    if not request.session.session_key:
        request.session.create()

    session_key = request.session.session_key
    ...
    form = TestS3UploadForm(initial={'session_key': session_key})

然后我在我的 models.py 中创建了一个函数，它使用模型中的 session_key 返回一个路径，并将模型的文件字段 upload_to 属性设置为这个函数

...
import os

def upload_to_session_key_dir(instance,filename):
    return os.path.join(instance.session_key,filename)

class TestS3Upload(models.Model):
    session_key = models.CharField(max_length=50,blank=False)
    uploaded_at = models.DateTimeField(auto_now_add=True)

    file = models.FileField(upload_to=upload_to_session_key_dir)

现在保存表单时，它会将文件上传到包含 session_key 的目录中。

最后的views.py

from django.shortcuts import render
from django.http import HttpResponse
from .forms import TestS3UploadForm
from .models import TestS3Upload

def test_s3_upload(request):

    # create session if it doesn't already exist
    if not request.session.session_key:
        request.session.create()

    session_key = request.session.session_key

    if request.method == 'POST':

        form = TestS3UploadForm(request.POST,request.FILES)

        if form.is_valid():
            form.save()

            filename = "{}/{}".format(session_key,form.cleaned_data['file'].name)
            s3_upload_path = TestS3Upload.objects.get(file=filename).file.url

            return HttpResponse("Image successfully uploaded to bucket at location: {}".format(s3_upload_path))
    else:
        form = TestS3UploadForm(initial={'session_key': session_key})

    return render(
        request,'upload/test_s3_upload.html',{
            'form': form
        }
    )

视图 test_s3_upload.html 的模板保持不变。