Django中從本地上傳excel文件並將數據存儲到數據庫

Django中從本地上傳excel文件並將數據存儲到數據庫

一、前端界面

<div class="page-container">
    <form action="{% url 'basic:uploadGrade' %}" method="post"
          enctype="multipart/form-data">
        {% csrf_token %}
        <input type="file" style="width: 50%" name="file" id="filename">
        <button class="btn btn-primary-outline radius size-S" type="submit">提交</button>
    </form>
</div>

二、views中

def uploadGrade(request):
    '''
    班級信息導入
    :param request:
    :return:
    '''
    if request.method == 'POST':
        f = request.FILES.get('file')
        excel_type = f.name.split('.')[1]
        if excel_type in ['xlsx','xls']:
            # 開始解析上傳的excel表格
            wb = xlrd.open_workbook(filename=None,file_contents=f.read())
            table = wb.sheets()[0]
            rows = table.nrows  # 總行數
            try:
                with transaction.atomic():  # 控制數據庫事務交易
                    for i in range(1,rows):
                        rowVlaues = table.row_values(i)
                        major = models.TMajor.objects.filter(majorid=rowVlaues[1]).first()
                        models.TGrade.objects.create(gradeid=rowVlaues[0],major=major,gradename=rowVlaues[2],memo=rowVlaues[3])
            except:
                logger.error('解析excel文件或者數據插入錯誤')
            return render(request,'bg/success.html',{'message':'導入成功'})
        else:
            logger.error('上傳文件類型錯誤！')
            return render(request,'bg/failed.html',{'message':'導入失敗'})