1. 程式人生 > 實用技巧 >Django-Excel匯入匯出

Django-Excel匯入匯出

  • 利用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