Add better support for "make import batch from file" pattern

This commit is contained in:
Lance Edgar 2018-11-22 20:26:28 -06:00
parent 0375d66b91
commit 4fa9ab3c6e
6 changed files with 80 additions and 6 deletions

View file

@ -103,15 +103,23 @@ class ImporterBatchView(BatchMasterView):
f.set_readonly('importer_key')
f.set_readonly('row_table')
def make_status_breakdown(self, batch):
# TODO: should implement this, just can't use batch.data_rows apparently
pass
def delete_instance(self, batch):
self.make_row_table(batch.row_table)
self.current_row_table.drop()
if self.current_row_table is not None:
self.current_row_table.drop()
super(ImporterBatchView, self).delete_instance(batch)
def make_row_table(self, name):
if not hasattr(self, 'current_row_table'):
metadata = sa.MetaData(schema='batch', bind=self.Session.bind)
self.current_row_table = sa.Table(name, metadata, autoload=True)
try:
self.current_row_table = sa.Table(name, metadata, autoload=True)
except sa.exc.NoSuchTableError:
self.current_row_table = None
def get_row_data(self, batch):
self.make_row_table(batch.row_table)