0

我想在 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

所需的软件包是boto3django-storages以防万一任何想帮助回答的人都需要知道。

4

1 回答 1

0

我想出了一种使用会话密钥将图像/文件上传到 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, null=False, 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 的模板保持不变。

上传成功消息,其中包含上传文件的完整 url 路径

于 2021-01-26T12:31:24.203 回答