Django-如何创建文件并将其保存到模型的FileField中?


110

这是我的模特。我想要做的是生成一个新文件,并在保存模型实例时覆盖现有文件:

class Kitten(models.Model):
    claw_size = ...
    license_file = models.FileField(blank=True, upload_to='license')

    def save(self, *args, **kwargs):
        #Generate a new license file overwriting any previous version
        #and update file path
        self.license_file = ???
        super(Request,self).save(*args, **kwargs)

我看到很多有关如何上传文件的文档。但是,如何生成文件,将其分配给模型字段并将Django存储在正确的位置呢?

Answers:


152

您想看看Django文档中的FileField和FieldFile,尤其是FieldFile.save()

基本上,FileField在访问时声明为的字段为您提供class的实例FieldFile,该实例为您提供了几种与基础文件进行交互的方法。因此,您需要做的是:

self.license_file.save(new_name, new_contents)

new_name您要分配的文件名在哪里,并且new_contents是文件的内容。请注意,new_contents该实例必须是django.core.files.File或的一个实例django.core.files.base.ContentFile(有关详细信息,请参见给定的手册链接)。这两个选择可以归结为:

# Using File
f = open('/path/to/file')
self.license_file.save(new_name, File(f))
# Using ContentFile
self.license_file.save(new_name, ContentFile('A string with the file content'))

1
好的,我认为这可以工作,但是我正在进入某种递归循环,在save方法中调用它。它只是永远创建文件。
格雷格

11
对于递归问题,我必须使用arg save = False调用self.license_file.save。
格雷格

1
该(ContentFile)与django-wkhtmltopdf convert_to_pdf命令返回的文件字符串完美配合。谢谢!!
Nostalg.io

除此之外,如果在打开文件时未指定文件模式,则会出现错误。因此,f = open('/path/to/file', 'r')对于ZIP类型的文件,f = open('/path/to/file.zip', 'rb')
rajagopalx

1
就我而言,以上内容并未将文件保存到文件夹中。原来的问题是我正在使用docker-compose和celery worker一起运行django应用。django应用程序卷MEDIA_ROOT未与celery worker中的相同卷共享。共享命名的卷将其固定(参考)。
shadi

28

接受的答案当然是一个很好的解决方案,但是这就是我生成CSV并从视图中提供它的方法。

我认为这是值得的,因为我花了点时间来摆弄所有想要的行为(覆盖现有文件,存储到正确的位置,不创建重复的文件等)。

的Django 1.4.1

Python 2.7.3

#Model
class MonthEnd(models.Model):
    report = models.FileField(db_index=True, upload_to='not_used')

import csv
from os.path import join

#build and store the file
def write_csv():
    path = join(settings.MEDIA_ROOT, 'files', 'month_end', 'report.csv')
    f = open(path, "w+b")

    #wipe the existing content
    f.truncate()

    csv_writer = csv.writer(f)
    csv_writer.writerow(('col1'))

    for num in range(3):
        csv_writer.writerow((num, ))

    month_end_file = MonthEnd()
    month_end_file.report.name = path
    month_end_file.save()

from my_app.models import MonthEnd

#serve it up as a download
def get_report(request):
    month_end = MonthEnd.objects.get(file_criteria=criteria)

    response = HttpResponse(month_end.report, content_type='text/plain')
    response['Content-Disposition'] = 'attachment; filename=report.csv'

    return response

1

close()在文件保存过程中,如果有异常情况,最好使用上下文管理器或进行调用。如果您的存储后端关闭等情况,可能会发生。

任何覆盖行为都应在存储后端中配置。例如S3Boto3Storage有一个设置AWS_S3_FILE_OVERWRITE。如果您正在使用FileSystemStorage,则可以编写自定义mixin

如果您希望发生任何自定义的副作用(例如最近更新的时间戳记),则可能还需要调用模型的save方法而不是FileField的save方法。如果是这种情况,您还可以将文件的name属性设置为文件的名称-相对于MEDIA_ROOT。它默认为文件的完整路径,如果不设置它,可能会引起问题-请参见File .__ init __()File.name

这是一个示例实例,其中self是模型实例,其中my_file是FileField / ImageFile,它调用save()整个模型实例,而不仅仅是FileField:

import os
from django.core.files import File

with open(filepath, 'rb') as fi:
    self.my_file = File(fi, name=os.path.basename(fi.name))
    self.save()
By using our site, you acknowledge that you have read and understand our Cookie Policy and Privacy Policy.
Licensed under cc by-sa 3.0 with attribution required.