Django FileField,在运行时确定upload_to

时间:2009-07-27 21:22:39

标签: python django django-models

我正在尝试设置我的上传,以便如果用户joe上传文件,则转到MEDIA_ROOT / joe,而不是让每个人的文件都转到MEDIA_ROOT。问题是我不知道如何在模型中定义它。以下是目前的情况:

class Content(models.Model):
    name = models.CharField(max_length=200)
    user = models.ForeignKey(User)
    file = models.FileField(upload_to='.')

所以我想要的不是'。'作为upload_to,让它成为用户的名字。

据我所知,从Django 1.0开始,您可以定义自己的函数来处理upload_to,但该函数不知道用户是谁,所以我有点迷失。

感谢您的帮助!

5 个答案:

答案 0 :(得分:248)

你可能已经阅读了the documentation,所以这里有一个简单的例子让它变得有意义:

def content_file_name(instance, filename):
    return '/'.join(['content', instance.user.username, filename])

class Content(models.Model):
    name = models.CharField(max_length=200)
    user = models.ForeignKey(User)
    file = models.FileField(upload_to=content_file_name)

正如您所看到的,您甚至不需要使用给定的文件名 - 如果您愿意,也可以在upload_to中调用该文件名。

答案 1 :(得分:12)

这确实有帮助。为了更简洁起见,我决定在我的情况下使用lambda:

file = models.FileField(
    upload_to=lambda instance, filename: '/'.join(['mymodel', str(instance.pk), filename]),
)

答案 2 :(得分:3)

关于使用'instance'对象的pk值的注释。根据文件:

  

在大多数情况下,此对象尚未保存到数据库中,因此如果它使用默认的AutoField,则可能还没有其主键字段的值。

因此,使用pk的有效性取决于您的特定模型的定义方式。

答案 3 :(得分:0)

如果您在迁移时遇到问题,则可能应该使用@deconstructible装饰器。

import datetime
import os
import unicodedata

from django.core.files.storage import default_storage
from django.utils.deconstruct import deconstructible
from django.utils.encoding import force_text, force_str


@deconstructible
class UploadToPath(object):
    def __init__(self, upload_to):
        self.upload_to = upload_to

    def __call__(self, instance, filename):
        return self.generate_filename(filename)

    def get_directory_name(self):
        return os.path.normpath(force_text(datetime.datetime.now().strftime(force_str(self.upload_to))))

    def get_filename(self, filename):
        filename = default_storage.get_valid_name(os.path.basename(filename))
        filename = force_text(filename)
        filename = unicodedata.normalize('NFKD', filename).encode('ascii', 'ignore').decode('ascii')
        return os.path.normpath(filename)

    def generate_filename(self, filename):
        return os.path.join(self.get_directory_name(), self.get_filename(filename))

用法:

class MyModel(models.Model):
    file = models.FileField(upload_to=UploadToPath('files/%Y/%m/%d'), max_length=255)

答案 4 :(得分:0)

如果你有一个用户实例,让有一个快速设置来生成

<model-slug>/<username>-<first_name>-<last_name>/filename-random.png

例如: /medias/content/ft0004-john-doe/filename-lkl9237.png


def upload_directory_name(instance, filename):

    user = getattr(instance, 'user', None)
    if user:
        name = f"{user.username}-{user.get_full_name().replace(' ', '-')}"
    else:
        name=str(instance)
    model_name = instance._meta.verbose_name.replace(' ', '-')
    return str(os.path.pathsep).join([model_name, name, filename])


class Content(models.Model):
    name = models.CharField(max_length=200)
    user = models.ForeignKey(User)
    file = models.FileField(upload_to=upload_directory_name)


[@SmileyChris 的修改版]