Django-Excel匯入匯出
阿新 • • 發佈:2020-07-28
- 利用pandas模組實現Excel與MySQL的互通
- 程式碼實現
- Excel資料匯入MySQL(方式一)
# 批量匯入資料(速度快) def importdata(localpath: str, db: str, foreignkey): data = pandas.read_excel(localpath, dtype='str', keep_default_na=False) # 讀取本地上傳的excel表內容, dtype='str'防止讀取001為1這種情況,keep_default_na=False是防止空值為NULL。 verbosename = db._meta.fields #獲取被匯入資料庫的欄位 rows = data.shape[0] # 獲取excel表的行數 columns = data.shape[1] # 獲取excel表的列數 querysetlist = [] # 用於儲存資料 eachdata = {} # 用於儲存每一行資料 # 外來鍵需要新增_id,與資料庫中的欄位保持一致 for i in range(len(verbosename)): if verbosename[i].name in foreignkey: # 判斷是否為外來鍵 verbosename[i].name = verbosename[i].name + '_id' if len(verbosename) == columns: # 必須保證二者長度一致,順序一致。 for i in range(rows): for j in range(len(verbosename)): eachdata[verbosename[j].name] = data.iloc[i, j] # 不會包括表頭 querysetlist.append(db(**eachdata)) try: db.objects.bulk_create(querysetlist)# 批量匯入資料 msg = '資料匯入成功,匯入資料共計{}條'.format(rows) except Exception as e: print('報錯資訊:{}'.format(e)) msg = '資料匯入失敗' return msg else: msg = '資料匯入失敗!請檢查excel表。' return msg
-
- Excel資料匯入MySQL(方式二)
# 逐個匯入資料(速度慢) def importdata2(localpath: str, db: str, foreignkey): data = pandas.read_excel(localpath, dtype='str', keep_default_na=False) # 讀取本地上傳的excel表內容, dtype='str'防止讀取001為1這種情況。 verbosename = db._meta.fields # 獲取被匯入資料庫的欄位 # for i in range(0, len(verbosename)): # print(verbosename[i].name) rows = data.shape[0] # 獲取excel表的行數 columns = data.shape[1] # 獲取excel表的列數 eachdata = {} # 用於儲存每一行資料 # 外來鍵需要新增_id,與資料庫中的欄位保持一致 for i in range(len(verbosename)): if verbosename[i].name in foreignkey: # 判斷是否為外來鍵 verbosename[i].name = verbosename[i].name + '_id' if len(verbosename) == columns: # 必須保證二者長度一致,順序一致。 for i in range(rows): for j in range(len(verbosename)): eachdata[verbosename[j].name] = data.iloc[i, j] # 不會包括表頭 try: db.objects.create(**eachdata) msg = '資料匯入成功,共計{}條'.format(rows) except Exception as e: print('報錯資訊:{}'.format(e)) msg = '資料匯入失敗' # return msg else: msg = '資料匯入失敗!請檢查excel表。' return msg
-
- 呼叫部分
def import_ceping(request): """匯入測評人員""" foreignkey = ['department', 'rank_id', 'post_id'] # 記錄外來鍵 # 接收excel檔案儲存到Media資料夾 rev_file = request.FILES.get('excel') # 判斷是否有檔案 if not rev_file: return JsonResponse({'code': 0, 'msg': 'Excel檔案不存在'}) # 獲得一個唯一的名字:uuid+hash new_name = get_random_str() # 準備寫入的URL file_path = os.path.join(settings.MEDIA_ROOT, new_name + os.path.splitext(rev_file.name)[1]) # 開始寫入磁碟 try: f = open(file_path, 'wb') # 分多次寫入 for i in rev_file.chunks(): f.write(i) f.close() except Exception as e: return JsonResponse({'code': 0, 'msg': '寫入失敗,請聯絡管理員'}) # 讀取儲存在Media資料夾的資料 ex_data = importdata(file_path, Ceping, foreignkey) return JsonResponse({'code': 1, 'msg': ex_data})
-
- MySQL資料匯出Excel
# 匯出資料 def exportdata(localpath, db): verbosename = db._meta.fields columnname = [] # 儲存欄位名 for i in range(len(verbosename)): columnname.append(verbosename[i].name) info = db.objects.values_list() # 查詢資料庫資料 data = pandas.DataFrame(info) try: data.to_excel(localpath, na_rep='NULL', header=columnname, index=False) msg = '資料匯出成功' except Exception as e: msg = '資料匯出失敗' return msg