快捷導(dǎo)航

Django上傳excel表格并將數(shù)據(jù)寫入數(shù)據(jù)庫的詳細步驟

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

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

前言：

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

一、上傳文件：

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

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 導(dǎo)入數(shù)據(jù)庫

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

# 將excel數(shù)據(jù)寫入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 +","
　   # 為了提高運行效率，一次性把數(shù)據(jù) insert 進數(shù)據(jù)庫  　
    sql = sql[:-1]
    # 寫入數(shù)據(jù)庫  
    # DataConnection 是自定義的公共模塊，用的是第三方庫，用來操作數(shù)據(jù)庫。沒有用 ORM ，后續(xù)有 group by 等復(fù)雜 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('導(dǎo)入成功')

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