BLD: add -n flag to control parallel doc build

tacaswell · tacaswell · commit 08d4b1998b6f · 2016-08-28T14:10:01.000-04:00
diff --git a/.travis.yml b/.travis.yml
@@ -142,7 +142,7 @@ script:
       gdb -return-child-result -batch -ex r -ex bt --args python tests.py -s --processes=$NPROC --process-timeout=300 $TEST_ARGS
     else
       cd doc
-      python make.py html
+      python make.py html -n 2
       # We don't build the LaTeX docs here, so linkchecker will complain
       touch build/html/Matplotlib.pdf
       linkchecker build/html/index.html
diff --git a/doc/make.py b/doc/make.py
@@ -53,7 +53,7 @@ def html(buildername='html'):
     if warnings_as_errors:
         options = options + ' -W'
     if os.system('sphinx-build -j %d %s -b %s -d build/doctrees . build/%s' % (
-            2, options, buildername, buildername)):
+            n_proc, options, buildername, buildername)):
         raise SystemExit("Building HTML failed.")
 
     # Clean out PDF files from the _images directory
@@ -147,6 +147,7 @@ def all():
 
 small_docs = False
 warnings_as_errors = True
+n_proc = 1
 
 # Change directory to the one containing this file
 current_dir = os.getcwd()
@@ -195,11 +196,16 @@ def all():
 parser.add_argument("--allowsphinxwarnings",
                     help="Don't turn Sphinx warnings into errors",
                     action="store_true")
+parser.add_argument("-n",
+                    help="Number of parallel workers to use")
+
 args = parser.parse_args()
 if args.small:
     small_docs = True
 if args.allowsphinxwarnings:
     warnings_as_errors = False
+if args.n is not None:
+    n_proc = int(args.n)
 
 if args.cmd:
     for command in args.cmd:
