Mở bảng tính trả về InMemoryUploadedFile

Question

Tôi có một người dùng tải tệp lên trang web và tôi cần phải phân tích cú pháp bảng tính. Đây là mã của tôi:

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

Các lỗi tôi tiếp tục nhận được là:

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

Tại sao điều này xảy ra và những gì tôi cần phải làm gì để khắc phục nó? Cảm ơn bạn.

Để tham khảo, đây là cách tôi mở file trong vỏ

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

Answer 1

Bạn có thể đổ InMemoryUploadedFile vào một tập tin tạm thời trước khi mở với xlrd.

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 

Nếu bạn muốn giữ lại tất cả mọi thứ trong bộ nhớ, hãy thử:

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