在我的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)