如何为models.FileField动态选择存储选项？

Question

根据文件扩展名，我希望将文件存储在特定的 AWS 存储桶中。我尝试将函数传递给 storage 选项，类似于动态定义 upload_to 的方式。

但是，这并没有给出预期的结果。在我的模板中，当我尝试使用 href 到 document.docfile.url 时，链接不起作用。

检查外壳，发生这种情况

Document.objects.all()[0].docfile.storage.bucket
 <Bucket: <function aws_bucket at 0x110672050>>

Document.objects.all()[0].docfile.storage.bucket_name
 <function myproject.myapp.models.aws_bucket>

期望的行为是

Document.objects.all()[0].docfile.storage.bucket_name
 'asynch-uploader-txt'
Document.objects.all()[0].docfile.storage.bucket     
 <Bucket: asynch-uploader-txt>

这是我的models.py 文件：

# -*- coding: utf-8 -*-
from django.db import models
from storages.backends.s3boto import S3BotoStorage

def upload_file_to(instance, filename):
    import os
    from django.utils.timezone import now
    filename_base, filename_ext = os.path.splitext(filename)
    return 'files/%s_%s%s' % (
        filename_base,
        now().strftime("%Y%m%d%H%M%S"),
        filename_ext.lower(),
    )

def aws_bucket(instance, filename):
    import os
    filename_base, filename_ext = os.path.splitext(filename)
    return 'asynch-uploader-%s' %(filename_ext[1:])

class Document(models.Model):
    docfile = models.FileField(upload_to=upload_file_to,storage=S3BotoStorage(bucket=aws_bucket))

为什么aws_bucket 是作为函数而不是字符串传递的，就像upload_file_to 那样？我该如何纠正？

Answer 1

对于您想要做的事情，您最好制作一个自定义存储后端并覆盖 S3BotoStorage 的各个部分。 特别是如果您将bucket_name 设为属性，您应该能够获得您想要的行为。

编辑：

为了进一步扩展，S3BotoStorage.__init__ 的 source 将 bucket 作为可选参数。此外，bucket 在类中使用时为@param，使其易于覆盖。以下代码未经测试，但应该足以为您提供一个起点

class MyS3BotoStorage(S3BotoStorage):
    @property
    def bucket(self):
        if self._filename.endswith('.jpg'):
            return self._get_or_create_bucket('SomeBucketName')
        else:
            return self._get_or_create_bucket('SomeSaneDefaultBucket')

    def _save(self, name, content):
        # This part might need some work to normalize the name and all...
        self._filename = name
        return super(MyS3BotoStorage, self)._save(name, content)