cubicweb/dataimport/massive_store.py
changeset 11786 f5d26d3648d4
parent 11785 0cea67f41d0c
child 11787 05b3e44c135f
equal deleted inserted replaced
11785:0cea67f41d0c 11786:f5d26d3648d4
   227         self.on_commit()
   227         self.on_commit()
   228         super(MassiveObjectStore, self).commit()
   228         super(MassiveObjectStore, self).commit()
   229 
   229 
   230     def finish(self):
   230     def finish(self):
   231         """Remove temporary tables and columns."""
   231         """Remove temporary tables and columns."""
   232         if self.slave_mode:
   232         assert not self.slave_mode, 'finish method should only be called by the master store'
   233             raise RuntimeError('Store cleanup is not allowed in slave mode')
       
   234         self.logger.info("Start cleaning")
   233         self.logger.info("Start cleaning")
   235         # Get all the initialized etypes/rtypes
   234         # Get all the initialized etypes/rtypes
   236         if self._dbh.table_exists('cwmassive_initialized'):
   235         if self._dbh.table_exists('cwmassive_initialized'):
   237             cu = self.sql('SELECT retype, type, uuid FROM cwmassive_initialized')
   236             cu = self.sql('SELECT retype, type, uuid FROM cwmassive_initialized')
   238             relations = defaultdict(list)
   237             relations = defaultdict(list)