rattail/rattail
2
0
Fork 0
Code Issues 7 Pull requests Actions Packages Projects Releases Wiki Activity

Return new batches from ImportHandler.make_batches()

Browse source
This commit is contained in:
Lance Edgar 2018-02-19 17:08:40 -06:00
parent 4107a6d34f
commit 752de8830a
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
rattail/importing/handlers.py
View file

@ -121,6 +121,7 @@ class ImportHandler(object):
self.setup() self.setup()
self.begin_transaction() self.begin_transaction()
batches = []
for key in keys: for key in keys:
@ -151,6 +152,7 @@ class ImportHandler(object):
row_table = self.make_row_table(metadata, importer, batch) row_table = self.make_row_table(metadata, importer, batch)
self.populate_row_table(session, importer, batch, row_table) self.populate_row_table(session, importer, batch, row_table)
batches.append(batch)
elif importer: elif importer:
log.info("batches not supported for importer: %s", key) log.info("batches not supported for importer: %s", key)
@ -165,6 +167,7 @@ class ImportHandler(object):
session.commit() session.commit()
session.close() session.close()
return batches
def make_row_table(self, metadata, importer, batch): def make_row_table(self, metadata, importer, batch):
columns = [ columns = [