def write_data(self, thebook): from pyexcel.book import to_book book = thebook if isinstance(thebook, BookStream): book = to_book(thebook) new_models = [model for model in self.__models if model is not None] batch_size = self.__keywords.get(params.BATCH_SIZE, None) initializers = self.__keywords.get(params.INITIALIZERS, None) if initializers is None: initializers = [None] * len(new_models) mapdicts = self.__keywords.get(params.MAPDICTS, None) if mapdicts is None: mapdicts = [None] * len(new_models) for sheet in book: if len(sheet.colnames) == 0: sheet.name_columns_by_row(0) colnames_array = [sheet.colnames for sheet in book] scattered = zip(new_models, colnames_array, mapdicts, initializers) importer = django.DjangoModelImporter() for each_model in scattered: adapter = django.DjangoModelImportAdapter(each_model[0]) adapter.column_names = each_model[1] adapter.column_name_mapping_dict = each_model[2] adapter.row_initializer = each_model[3] importer.append(adapter) to_store = OrderedDict() for sheet_name in book.sheet_names(): # due book.to_dict() brings in column_names # which corrupts the data to_store[sheet_name] = book[sheet_name].get_internal_array() save_data(importer, to_store, file_type=DB_DJANGO, batch_size=batch_size)
def write_data(self, sheet): headers = sheet.colnames if len(headers) == 0: headers = sheet.rownames importer = django.DjangoModelImporter() adapter = django.DjangoModelImportAdapter(self.__model) adapter.column_names = headers adapter.column_name_mapping_dict = self.__keywords.get( params.MAPDICT, None) adapter.row_initializer = self.__keywords.get(params.INITIALIZER, None) importer.append(adapter) save_data(importer, {adapter.get_name(): sheet.get_internal_array()}, file_type=DB_DJANGO, **self.__keywords)