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)