利用django如何解析用户上传的excel文件

最新推荐文章于 2023-12-27 11:21:06 发布

weixin_30824479

最新推荐文章于 2023-12-27 11:21:06 发布

阅读量698

点赞数

文章标签： python 数据库开发工具

原文链接：http://www.cnblogs.com/linkenpark/p/9607402.html

版权

https://www.jb51.net/article/119452.htm

前言

我们在工作中的时候，会有这种需求：用户上传一个格式固定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种方法：

将用户上传的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" 
          ) 
         