Django上傳excel表格並將數據寫入數據庫

時間 2020-07-07

標籤 django 上傳 excel 表格數據寫入數據庫欄目 Python 简体版

原文原文鏈接

前言：

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

1、上傳文件：

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

1.配置 setting.py mysql

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

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

{% 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('上傳成功')

2、解析 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('導入成功')