diff options
Diffstat (limited to 'subcmds/status.py')
-rw-r--r-- | subcmds/status.py | 16 |
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 | ||
15 | import functools | 15 | import functools |
16 | import glob | 16 | import glob |
17 | import io | ||
17 | import multiprocessing | 18 | import multiprocessing |
18 | import os | 19 | import os |
19 | 20 | ||
20 | from command import DEFAULT_LOCAL_JOBS, PagedCommand | 21 | from command import DEFAULT_LOCAL_JOBS, PagedCommand, WORKER_BATCH_SIZE |
21 | 22 | ||
22 | from color import Coloring | 23 | from color import Coloring |
23 | import platform_utils | 24 | import 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 | ||