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

前言:

最近公司領導要統計技術部門在各個業務條線花費的工時百分比,而 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!

推薦閱讀: