Question

我正在使用python的xlrd和csv来解析xls到csv。当用户上传xls或xlsx文件时，我能够解析并生成csv。虽然它生成实际文件，但我想更新或创建新的InMemoryUploadFile对象，我可以用它来创建FileField对象的记录。有什么办法吗？我试图查找这些示例，但大多数示例都是针对InMemoryUploadFile的图像文件。

以下是我尝试的示例代码 -

    import csv
    import xlrd
    import StringIO
    from django.core.files.uploadedfile import InMemoryUploadedFile        
    .......
    .......

    #xls_file is excel file
    file_name = '%s.csv'%(xls_file.name)
    workbook = xlrd.open_workbook(file_contents=xls_file.read())
    all_worksheets = workbook.sheet_names()

    # for now starting with reading the first sheet
    worksheet_name = all_worksheets[0]
    worksheet = workbook.sheet_by_name(worksheet_name)

    csv_output = StringIO.StringIO()
    csv_data = []
    for rownum in xrange(worksheet.nrows):
        csv_data.append(
            [unicode(
                entry).encode("utf-8") for entry in worksheet.row_values(
                rownum)])
    writer = csv.writer(csv_output)
    for row in csv_data:
        writer.writerow(csv_data)

    csv_file = InMemoryUploadedFile(writer, None, file_name, 'text/csv',
                              None, 'utf-8')
    return csv_file

Answer 1

您可以尝试使用ContentFile吗？ Original tutorial

from django.core.files.base import ContentFile
def save_file(csv_output):
    mymodel = MyModel.objects.get(id=1)
    file_content = ContentFile(csv_output.getvalue())
    mymodel.myfilefield.save(request.FILES['yourupload'].name, file_content)

将xls转换为csv并创建/更新InMemoryUploadedFile Django

1 个答案: