Django上傳excel表格並將資料寫入資料庫的詳細步驟

2022-06-30 22:00:34

前言：

最近公司領導要統計技術部門在各個業務條線花費的工時百分比，而 jira 當前的 Tempo 外掛只能統計個人工時。於是就寫了個報表工具，將 jira 中匯出的個人工時excel表格匯入資料庫，在後端處理各個業務工時佔比。後來研究了 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)
    # 建立upload資料夾
    if not os.path.exists(settings.UPLOAD_ROOT):
        os.makedirs(settings.UPLOAD_ROOT)
    try:
        if file is None:
            return HttpResponse('請選擇要上傳的檔案')
        # 迴圈二進位制寫入
        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 ，後續有 group by 等複雜 sql 不好操作。
    DataConnection.MysqlConnection().insert('work',sql)
@csrf_exempt
def upload(request):
    # 根name取 file 的值
    file = request.FILES.get('file')
    print('uplaod:%s'% file)
    # 建立upload資料夾
    if not os.path.exists(settings.UPLOAD_ROOT):
        os.makedirs(settings.UPLOAD_ROOT)
    try:
        if file is None:
            return HttpResponse('請選擇要上傳的檔案')
        # 迴圈二進位制寫入
        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.資料匯入後，通過一些處理就得到了我們想要的資料。報表其中之一的餅圖：

到此這篇關於Django上傳excel表格並將資料寫入資料庫的文章就介紹到這了,更多相關Django上傳excel表格內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com！

Django上傳excel表格並將資料寫入資料庫的詳細步驟

目錄

前言：

一、上傳檔案：

二、解析 excel 匯入資料庫

熱門文章