Mercurial > piecrust2
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 |