comparison piecrust/baking/baker.py @ 462:04abc97dd3b6

bake: Add CLI argument to specify job batch size.
author Ludovic Chabant <ludovic@chabant.com>
date Sat, 11 Jul 2015 18:49:50 -0700
parents 5e902e228053
children 456db44dcc53
comparison
equal deleted inserted replaced
461:b015e38d4ee1 462:04abc97dd3b6
539 def _createWorkerPool(self, previous_record_path): 539 def _createWorkerPool(self, previous_record_path):
540 from piecrust.workerpool import WorkerPool 540 from piecrust.workerpool import WorkerPool
541 from piecrust.baking.worker import BakeWorkerContext, BakeWorker 541 from piecrust.baking.worker import BakeWorkerContext, BakeWorker
542 542
543 worker_count = self.app.config.get('baker/workers') 543 worker_count = self.app.config.get('baker/workers')
544 batch_size = self.app.config.get('baker/batch_size')
544 545
545 ctx = BakeWorkerContext( 546 ctx = BakeWorkerContext(
546 self.app.root_dir, self.app.cache.base_dir, self.out_dir, 547 self.app.root_dir, self.app.cache.base_dir, self.out_dir,
547 previous_record_path=previous_record_path, 548 previous_record_path=previous_record_path,
548 force=self.force, debug=self.app.debug) 549 force=self.force, debug=self.app.debug)
549 pool = WorkerPool( 550 pool = WorkerPool(
550 worker_count=worker_count, 551 worker_count=worker_count,
552 batch_size=batch_size,
551 worker_class=BakeWorker, 553 worker_class=BakeWorker,
552 initargs=(ctx,)) 554 initargs=(ctx,))
553 return pool 555 return pool
554 556
555 557