打开电子表格返回InMemoryUploadedFile

时间:2012-10-14 21:45:59

标签: python django

我有一个用户将文件上传到网站,我需要解析电子表格。这是我的代码:

input_file = request.FILES.get('file-upload')
wb = xlrd.open_workbook(input_file)

我不断得到的错误是:

TypeError at /upload_spreadsheet/
coercing to Unicode: need string or buffer, InMemoryUploadedFile found

为什么会发生这种情况,我需要做些什么来解决它?谢谢。

作为参考,这是我在shell中打开文件的方式

>>> import xlrd
>>> xlrd.open_workbook('/Users/me/dave_example.xls')
<xlrd.Book object at 0x10d9f7390>

1 个答案:

答案 0 :(得分:13)

在使用xlrd打开之前,您可以将InMemoryUploadedFile转储到临时文件。

try:
    fd, tmp = tempfile.mkstemp()
    with os.fdopen(fd, 'w') as out:
        out.write(input_file.read())
    wb = xlrd.open_workbook(tmp)
    ...  # do what you have to do
finally:
    os.unlink(tmp)  # delete the temp file no matter what

如果您想将所有内容保存在内存中,请尝试:

wb = xlrd.open_workbook(filename=None, file_contents=input_file.read())