Django上傳excel表格并將數據寫入數據庫的詳細步驟

更新時間：2022年06月30日 16:57:09 作者：亂糟糟

這篇文章主要介紹了Django上傳excel表格并將數據寫入數據庫,將文件上傳到服務器指定路徑，其實很簡單，本文分三個步驟給大家詳細介紹，需要的朋友可以參考下

前言：

最近公司領導要統(tǒng)計技術部門在各個業(yè)務條線花費的工時百分比，而 jira 當前的 Tempo 插件只能統(tǒng)計個人工時。于是就寫了個報表工具，將 jira 中導出的個人工時excel表格導入數據庫，在后端處理各個業(yè)務工時占比。后來研究了 jira 的 API 文檔，放棄了之前的思路，直接調用 jira API 處理數據，這個先不談。這篇博客主要介紹 Django 上傳文件，然后解析 excel 導入數據庫。

一、上傳文件：

將文件上傳到服務器指定路徑，其實很簡單，一共有三個步驟：

1.配置 setting.py

# 文件上傳配置
UPLOAD_ROOT = os.path.join(BASE_DIR,'upload')

2.前端代碼如下，使用 <form> 表單提交，"/upload/" 路由配置在 urls 中，這個就不再多說了。

{% extends 'base.html' %}

{% block content %}
<body>
     <form id="form"  enctype="multipart/form-data" action="/upload/" method="post">
      <p><input type="file"  name="file"></p>
         <input type="submit" name="提交">
    </form>
</body>
{% endblock %}

3.后端代碼如下，這段代碼可以上傳任意格式的文件，沒有校驗文件類型。

@csrf_exempt
def upload(request):
    # 根name取 file 的值
    file = request.FILES.get('file')
    logger.log().info('uplaod:%s'% file)
    # 創(chuàng)建upload文件夾
    if not os.path.exists(settings.UPLOAD_ROOT):
        os.makedirs(settings.UPLOAD_ROOT)
    try:
        if file is None:
            return HttpResponse('請選擇要上傳的文件')
        # 循環(huán)二進制寫入
        with open(settings.UPLOAD_ROOT + "/" + file.name, 'wb') as f:
            for i in file.readlines():
                f.write(i)
    except Exception as e:
        return HttpResponse(e)
    return HttpResponse('上傳成功')

二、解析 excel 導入數據庫

1.文件上傳結束后，接下來讀取剛上傳到服務器的 excel 表格，然后寫入數據庫。所以整個后端代碼是這樣的：

# 將excel數據寫入mysql
def wrdb(filename):
    # 打開上傳 excel 表格
    readboot = xlrd.open_workbook(settings.UPLOAD_ROOT + "/" + filename)
    sheet = readboot.sheet_by_index(0)
    #獲取excel的行和列
    nrows = sheet.nrows
    ncols = sheet.ncols
    print(ncols,nrows)
    sql = "insert into working_hours (jobnum,name,workingtime,category,project,date,createtime) VALUES"
    for i in range(1,nrows):
        row = sheet.row_values(i)
        jobnum = row[4]
        name = row[5]
        workingtime = row[2]
        category = row[8]
        project = row[1]
        date = xldate_as_datetime(row[3],0).strftime('%Y/%m/%d')
        values = "('%s','%s','%s','%s','%s','%s','%s')"%(jobnum,name,workingtime,category,project,date,datetime.datetime.now())
        sql = sql + values +","
　   # 為了提高運行效率，一次性把數據 insert 進數據庫  　
    sql = sql[:-1]
    # 寫入數據庫  
    # DataConnection 是自定義的公共模塊，用的是第三方庫，用來操作數據庫。沒有用 ORM ，后續(xù)有 group by 等復雜 sql 不好操作。
    DataConnection.MysqlConnection().insert('work',sql)
@csrf_exempt
def upload(request):
    # 根name取 file 的值
    file = request.FILES.get('file')
    print('uplaod:%s'% file)
    # 創(chuàng)建upload文件夾
    if not os.path.exists(settings.UPLOAD_ROOT):
        os.makedirs(settings.UPLOAD_ROOT)
    try:
        if file is None:
            return HttpResponse('請選擇要上傳的文件')
        # 循環(huán)二進制寫入
        with open(settings.UPLOAD_ROOT + "/" + file.name, 'wb') as f:
            for i in file.readlines():
                f.write(i)

        # 寫入 mysql
        wrdb(file.name)
    except Exception as e:
        return HttpResponse(e)

    return HttpResponse('導入成功')

2.數據導入后，通過一些處理就得到了我們想要的數據。報表其中之一的餅圖：