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

Posted on by

前言：

最近公司領導要統計技術部門在各個業務條線花費的工時百分比，而 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表格內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！

推薦閱讀：