15

我需要从 excel 文件和带有图像的文件夹中导入一些数据，excel 中的每一行都描述了每个条目，并在文件夹中有一个文件名列表（与条目相关的照片）。

我已经完成了一个脚本，它在数据库中创建每个条目并通过 django shell 保存它，但我不知道如何实例化 InMemoryUploadedFile 以将其与模型一起保存。

在 django 1.0 中，我有一个小班，让我可以做我需要的事情，但是随着 django 1.1 的变化，它不再起作用了。

class ImportFile(file):
    def __init__(self, *args, **kwargs):
        super(ImportFile, self).__init__(*args, **kwargs)
        self._file = self
        self.size = os.path.getsize(self.name)

    def __len__(self):
        return self.size

    def chunks(self, chunk_size=None):
        self._file.seek(0)
        yield self.read()

我正在使用这个类和这段代码来加载图像并将它们与模型实例一起保存。

for photo in photos:
    f = ImportFile(os.path.join(IMPORT_DIR, 'fotos', photo), 'r')
    p = Photo(name=f.name, image=f, parent=supply.supply_ptr)
    name = str(uuid1()) + os.path.splitext(f.name)[1]
    p.image.save(name, f)
    p.save()

问题是，如何从 python 中的文件创建 InMemoryUploadedFile 或 TemporaryUploadedFile？，或任何其他可以在这种情况下工作的东西。

4

2 回答 2

23

最后我找到了答案。

from django.core.files import File

f = File(open(os.path.join(IMPORT_DIR, 'fotos', photo), 'r'))
p = Photo(name=f.name, image=f, parent=supply.supply_ptr)
name = str(uuid1()) + os.path.splitext(f.name)[1]
p.image.save(name, f)
p.save()
于 2009-08-05T11:07:11.410 回答
7

如果使用Python 3需要进行一项调整 - 将 'r' 更改为 'rb'：

f = File(open(os.path.join(IMPORT_DIR, 'fotos', photo), 'rb'))
于 2014-07-11T13:45:43.367 回答