如何使用 Django AutoSlugField 输出作为 ImageField 文件名

Question

我正在使用 django-extensions.AutoSlugField 和 django.db.models.ImageField。

为 django.db.models.ImageField 自定义上传的图片名称，我做了什么：

from django.utils.text import slugify

# idea is to make image name the same as the automatically generated slug, however don't work
def update_image_name(instance, filename):
    # this debug instance, and instance.slug is empty string
    print(instance.__dict__)

    # I attempt to use slugify directly and see that it's not the same as the output generated by AutoSlugField
    # E.g. If I create display_name "shawn" 2nd time, AutoSlugField will return "shawn-2", but slugify(display_name) return "shawn"
    print(slugify(instance.display_name))

    return f"images/{instance.slug}.jpg"
 

class Object(models.Model):
    ...
    display_name = models.TextField()
    ...
    # to customize uploaded image name
    image = models.ImageField(blank=True, upload_to=update_image_name)
    ...
    # create slug automatically from display_name
    slug = AutoSlugField(blank=True, populate_from=["display_name"]

根据我的调试，当我在update_image_name中调用instance时，slug是空字符串。

如果我理解正确 slug 仅在事件保存时创建，因此当我调用 ImageField 实例时，尚未创建 slug，因此为空字符串。

我认为这可能与事件 post 保存有关。但是，我不确定这是否是真正的原因或如何做到这一点。

如何将自动生成的 slug 作为我的自定义图像名称？

Answer 1

这是一个棘手的问题，因为字段的保存顺序很重要。

我建议的暴力攻击是重写模型的 save 方法，并在确保 slug 已设置之前手动调用 create_slug 方法：

from django.utils.encoding import force_str
[...]

class Object(models.Model):
    [...]

    def save(self, *args, **kwargs):
        self.slug = force_str(self._meta.get_field('slug').create_slug(self, False))
        super(Object, self).save(*args, **kwargs)

就是AutoSlugField的作用，参考代码here。 self._meta.get_field('slug') 获取 slug 字段定义，然后我们只需调用 create_slug 方法。

在 Python 3.7.9 和 Django 3.1.5 下测试如下：

from django.core.files.uploadedfile import SimpleUploadedFile
o = Object()
o.display_name = "foo bar"
o.image = SimpleUploadedFile(name='test_image.png', content=open('/path/to/test/image.png', 'rb').read(), content_type='image/png')
o.save()

然后我看到update_image_namereturnimages/foo-bar.jpg.

如何使用 Django AutoSlugField 输出作为 ImageField 文件名

How to use Django AutoSlugField output as ImageField filename

django

django-models

django-views

django-forms

django-extensions