利用django如何解析用戶上傳的excel文件

2020-01-04 17:03:59

字體：大中小

來源：轉載

供稿：網友

前言

我們在工作中的時候，會有這種需求：用戶上傳一個格式固定excel表格到網站上，然后程序負債解析內容并進行處理。我最近在工作中就遇到了，所以想著將解決的過程總結分享出來，方便大家參考學習，下面話不多說，來一起看看詳細的介紹：

舉一個簡單的栗子，比如我們有這樣一個HTML：

<!DOCTYPE html><html> <head> <meta charset="utf-8"> </head> <body> <p>上傳EXCEL表格</p> <form class="" action="" method="post" enctype="multipart/form-data" >  {% csrf_token %}  <input type="file" name="excel">  <input type="submit" value="上傳"> </form> </body></html>

forms.py文件內容如下，編寫一個簡單的判斷后綴的驗證：

# coding=utf-8from django import formsfrom django.utils.translation import gettext as _from django.core.exceptions import ValidationErrordef validate_excel(value): if value.name.split('.')[-1] not in ['xls','xlsx']: raise ValidationError(_('Invalid File Type: %(value)s'),params={'value': value},)class UploadExcelForm(forms.Form): excel = forms.FileField(validators=[validate_excel]) #這里使用自定義的驗證

處理excel表格我這里使用xlrd庫，使用pip安裝即可。此時處理POST請求時有2種方法：

將用戶上傳的excel存儲到磁盤中再讀取交給xlrd處理。
直接在內存中讀取用戶上傳的excel讀取交給xlrd處理。

這里我使用第二個辦法——在不修改django默認settings.py配置情況下，用戶上傳的文件其實是InMemoryUploadedFile類型，這個類型有一個read()方法，所以views.py中可以內存直接讀取內容而不用寫磁盤再讀取了：

def post(self, request, *args, **kwargs): form = UploadExcelForm(request.POST, request.FILES) if form.is_valid(): wb = xlrd.open_workbook(  filename=None, file_contents=request.FILES['excel'].read()) # 關鍵點在于這里 table = wb.sheets()[0] row = table.nrows for i in xrange(1, row):  col = table.row_values(i)  print col return HttpResponse("ok")

其他文件類型同理，如果不需要保存用戶上傳的文件到硬盤其實都可以這么處理。這里記錄2個和django處理excel有關的資源：