diff 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
line wrap: on
line diff
--- a/piecrust/baking/baker.py	Sat Jul 11 17:51:56 2015 -0700
+++ b/piecrust/baking/baker.py	Sat Jul 11 18:49:50 2015 -0700
@@ -541,6 +541,7 @@
         from piecrust.baking.worker import BakeWorkerContext, BakeWorker
 
         worker_count = self.app.config.get('baker/workers')
+        batch_size = self.app.config.get('baker/batch_size')
 
         ctx = BakeWorkerContext(
                 self.app.root_dir, self.app.cache.base_dir, self.out_dir,
@@ -548,6 +549,7 @@
                 force=self.force, debug=self.app.debug)
         pool = WorkerPool(
                 worker_count=worker_count,
+                batch_size=batch_size,
                 worker_class=BakeWorker,
                 initargs=(ctx,))
         return pool