Django迁移和FileSystemStorage取决于设置



在我的Django应用程序中,我使用FileSystemStorage来生成文件。我像这样初始化它:

import os
from urlparse import urljoin
from django.conf import settings
from django.core.files.storage import FileSystemStorage
gen_files_storage = FileSystemStorage(location=os.path.join(settings.MEDIA_ROOT, 'generated/'), base_url=urljoin(settings.MEDIA_URL, 'generated/'))

当我想创建一个新文件时,我使用:

from django.core.files.base import ContentFile
from django.db import models
def next_number():
    # num = ...
    return num
gen_file = models.FileField(storage=gen_files_storage)
gen_file.save('file%s.txt' % next_number(), ContentFile(''))

很好。唯一的问题是FileSystemStorage的路径在Django迁移中是"硬编码"的。因为我使用不同的开发(更改)和生产设置,通常manage.py makemigrations命令生成迁移只是因为路径更改,尽管数据库中的一切保持不变。

我知道有一个解决方案使用FileSystemStorage的子类(见下面我的回答),但有一个更好的解决方案吗?

有一个解决方案涉及FileSystemStorage的自定义@deconstructible子类:

import os
from urlparse import urljoin
from django.conf import settings
from django.core.files.storage import FileSystemStorage
from django.utils.deconstruct import deconstructible
@deconstructible
class MyFileSystemStorage(FileSystemStorage):
    def __init__(self, subdir):
        self.subdir = subdir
        super(MyFileSystemStorage, self).__init__(location=os.path.join(settings.MEDIA_ROOT, self.subdir), base_url=urljoin(settings.MEDIA_URL, self.subdir))
    def __eq__(self, other):
        return self.subdir == other.subdir

然后我可以像这样初始化存储:

import os
from urlparse import urljoin
from django.conf import settings
from django.core.files.storage import FileSystemStorage
gen_files_storage = MyFileSystemStorage('generated/')

这样Django迁移就不会注意到我设置中的变化。有更好的方法吗?

升级到Django 3.1+修复此问题:https://docs.djangoproject.com/en/3.2/releases/3.1/#file-storage

只需将一个可调用对象传递给storage参数。

from django.db import models
from django.conf import settings
from django.core.files.storage import get_storage_class

def _get_storage():
    storage_class = get_storage_class(settings.MY_STORAGE_CLASS)  # ie. 'django.core.files.storage.FileSystemStorage'
    return storage_class()
class MyModel(models.Model):
    myfile = models.FileField(max_length=255, blank=True, storage=_get_storage)

解决方案是永远不要在生产环境中运行makemigrations。在生产服务器上运行migrate,但是忽略关于运行makemigrations的警告,如果它们与此问题有关。

想想看:makemigrations生成Python代码,所以在生产环境中运行它和在服务器上开发是一样的。根据您的服务器设置,您的生产站点可能会正确地提供这些文件,而不考虑makemigrations警告。

我的问题是相关的,但略有不同。该字段使用的存储类可以根据以下设置进行更改:生产中的默认本地远程存储。我实现了FileField的一个子类,它在解构用于迁移生成的字段时忽略了存储 kwarg。

from django.db.models import FileField
class VariableStorageFileField(FileField):
    """
    Disregard the storage kwarg when creating migrations for this field
    """
    def deconstruct(self):
        name, path, args, kwargs = super(VariableStorageFileField, self).deconstruct()
        kwargs.pop('storage', None)
        return name, path, args, kwargs

可以这样使用:

class MyModel(models.Model):
    storage = get_storage_class(getattr(settings, 'LARGE_FILE_STORAGE', None))()
    file = VariableStorageFileField(blank=True, null=True, storage=storage)

最新更新