Django operates Excel files through openpyxl to import data into the database in batches.

1、 In some scenarios, administrators often need to import data in batches , First write the data to excel In the table , Upload to the system , The final will be excel The data in is imported into the database of the system in batches .

1.1、test.xlsx File data format

The first line's name ： It must be the field name corresponding to the database , So that we can use ** Break up the dictionary , Achieve rapid creation .

 

2、 How to implement the view class

class TestExcelView(ViewSet):
    authentication_classes = []
    permission_classes = []
    throttle_classes = []
    @action(methods=['get'],detail=False)
    def testexcel(self,request):
        path=os.path.join(settings.MEDIA_ROOT,'students_excels','test.xlsx')
        wb = openpyxl.load_workbook(path)
        ws = wb.active
        max_row = ws.max_row
        max_column = ws.max_column
        all_lis=[]
        lis=[]
        for row in range(1,max_row+1):
            dic={}
            for column in range(1,max_column+1) :
                if row==1:
                    # Add the first line to the list 
                    lis.append(ws.cell(row,column).value)
                    continue
                # Take the line header as key, The remaining rows correspond to columns as value, Get dict, One in a row dict
                dic.setdefault(lis[column-1],ws.cell(row,column).value)
            if row>1:
                # First line dict It's empty , There is no need to join 
                all_lis.append(dic)
        try:
            for user_dic in all_lis:
                models.User.objects.create_user(**user_dic)
            return ApiResponse(msg=' User created successfully ',data=all_lis)
        except:
            return ApiResponse(data={'erroe':'excel There is something wrong with the document. , First check the file format '})

3、all_list Data format

A row is a database record , Directly when creating users **dic, You can create records .

The principle of other batch generated data is the same .