summaryrefslogtreecommitdiffstats
path: root/subcmds/status.py
diff options
context:
space:
mode:
Diffstat (limited to 'subcmds/status.py')
-rw-r--r--subcmds/status.py16
1 files changed, 12 insertions, 4 deletions
diff --git a/subcmds/status.py b/subcmds/status.py
index f0f2e034..6c8e22e5 100644
--- a/subcmds/status.py
+++ b/subcmds/status.py
@@ -14,10 +14,11 @@
14 14
15import functools 15import functools
16import glob 16import glob
17import io
17import multiprocessing 18import multiprocessing
18import os 19import os
19 20
20from command import DEFAULT_LOCAL_JOBS, PagedCommand 21from command import DEFAULT_LOCAL_JOBS, PagedCommand, WORKER_BATCH_SIZE
21 22
22from color import Coloring 23from color import Coloring
23import platform_utils 24import platform_utils
@@ -99,7 +100,9 @@ the following meanings:
99 Returns: 100 Returns:
100 The status of the project. 101 The status of the project.
101 """ 102 """
102 return project.PrintWorkTreeStatus(quiet=quiet) 103 buf = io.StringIO()
104 ret = project.PrintWorkTreeStatus(quiet=quiet, output_redir=buf)
105 return (ret, buf.getvalue())
103 106
104 def _FindOrphans(self, dirs, proj_dirs, proj_dirs_parents, outstring): 107 def _FindOrphans(self, dirs, proj_dirs, proj_dirs_parents, outstring):
105 """find 'dirs' that are present in 'proj_dirs_parents' but not in 'proj_dirs'""" 108 """find 'dirs' that are present in 'proj_dirs_parents' but not in 'proj_dirs'"""
@@ -128,8 +131,13 @@ the following meanings:
128 counter += 1 131 counter += 1
129 else: 132 else:
130 with multiprocessing.Pool(opt.jobs) as pool: 133 with multiprocessing.Pool(opt.jobs) as pool:
131 states = pool.map(functools.partial(self._StatusHelper, opt.quiet), all_projects) 134 states = pool.imap(functools.partial(self._StatusHelper, opt.quiet),
132 counter += states.count('CLEAN') 135 all_projects, chunksize=WORKER_BATCH_SIZE)
136 for (state, output) in states:
137 if output:
138 print(output, end='')
139 if state == 'CLEAN':
140 counter += 1
133 if not opt.quiet and len(all_projects) == counter: 141 if not opt.quiet and len(all_projects) == counter:
134 print('nothing to commit (working directory clean)') 142 print('nothing to commit (working directory clean)')
135 143