4

我有一个用于处理上传文件的抽象模型类 UploadItem。我希望每个子类都能够定义 upload_to 路径。为此，我将回调传递给 FileField 的构造函数。

这是一个例子：

class UploadItem(models.Model):
    file = models.FileField(upload_to=UploadItem.get_directory) 


    class Meta:
        abstract = True
# I want videos to be storred in 'videos/' directory
class Video(UploadItem):
    def get_directory(self, instance, filename):
        return 'videos/'

但这不起作用，我收到此错误：

file = models.FileField(upload_to=UploadItem.get_directory) 
NameError: name 'UploadItem' is not defined
4

2 回答 2

6

鉴于在评估时，该错误是自然的

file = models.FileField(upload_to=UploadItem.get_directory)

该类UploadItem尚未定义。您可以执行以下操作以使其工作：

def get_directory():
    pass

class UploadItem(models.Model):
    file = models.FileField(upload_to=get_directory)

    class Meta:
        abstract = True

但这并不能解决您的所有问题。get_directory在类中添加（或覆盖）方法Video不会改变模型属性upload_to的file属性。

更新

文档说upload_to可以是可调用的。

这也可能是一个可调用的，例如一个函数，将被调用以获取上传路径，包括文件名。此可调用对象必须能够接受两个参数，并返回要传递给存储系统的 Unix 样式路径（带有正斜杠）。

鉴于此，我们可以编写一个自定义回调函数，如下所示：

categories_and_paths = { 'video': 'videos/', 'photo': 'photos/' } # etc.
def get_directory(instance, filename):
    category = instance.category
    return categories_and_paths.get(category, '')

Instance这里将是相应模型的实例。为此，每个模型实例都应该有一个类别字段。我们可以在模型主体中添加一个。

class Video(UploadItem):
    category = 'video'
于 2010-09-08T16:45:58.700 回答
5

如果您需要使用基类属性作为子类的一部分，这可以通过一些调整以类似的方式完成：

import os

# ...

def get_directory(instance, filename):
    return instance.get_file_directory(filename)


class UploadItem(models.Model):
    FILE_DIRECTORY = 'files/'

    file = models.FileField(upload_to=get_directory) 

    class Meta:
        abstract = True

    @staticmethod
    def get_file_directory(filename):
        return os.path.join(UploadItem.FILE_DIRECTORY, filename)

然后在子类中：

class Video(UploadItem):
    FILE_DIRECTORY = 'videos/'

    def get_file_directory(self, filename):
        filename = os.path.join(self.FILE_DIRECTORY, filename)
        return super().get_document_path(filename)

这样，您将拥有upload_to等于：

  • files为了UploadItem
  • files/videos为了Video

对于需要共享一些公共基础属性的更复杂的对象可能很有用。

于 2016-12-26T14:24:58.400 回答