def setUpClass(cls): super().setUpClass() mb_factories.MeetboutFactory( id=123456, bouwbloknummer='AB10' ) mb_factories.MeetboutFactory( id=123451, bouwbloknummer='AB10' ) mb_factories.MeetboutFactory( id=12346, bouwbloknummer='AB11' ) batch.execute( datasets.meetbouten.batch.IndexMeetboutenJob())
def handle(self, *args, **options): dataset = options['dataset'] for ds in dataset: if ds not in self.indexes.keys(): self.stderr.write("Unkown dataset: {}".format(ds)) return sets = [ds for ds in self.ordered if ds in dataset] # enforce order self.stdout.write("Working on {}".format(", ".join(sets))) if options['batch_size'] > 0: settings.BATCH_SETTINGS['batch_size'] = options['batch_size'] for ds in sets: if options['backup_indexes_es']: for job_class in self.backup_indexes[ds]: batch.execute(job_class()) # we do not run the other tasks continue # to next dataset please.. if options['restore_indexes_es']: for job_class in self.restore_indexes[ds]: batch.execute(job_class()) # we do not run the other tasks continue # to next dataset please.. if options['build_index']: for job_class in self.indexes[ds]: batch.execute(job_class())