django解析用户上传的excel文件

Roy
编程技术

有些时候我们会有这种需求：用户上传一个格式固定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-8


from django import forms
from django.utils.translation import gettext as _
from django.core.exceptions import ValidationError

def 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种方法：

  1. 将用户上传的excel存储到磁盘中再读取交给xlrd处理。
  2. 直接在内存中读取用户上传的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有关的资源：

  1. django-excel 判断用户excel格式的三方库
  2. https://assist-software.net/blog/how-export-excel-files-python-django-application 讲解如何导出excel的文章