diff options
Diffstat (limited to '')
| -rw-r--r-- | patternutils/commands/targs.py | 3 | ||||
| -rw-r--r-- | patternutils/utils.py | 6 |
2 files changed, 8 insertions, 1 deletions
diff --git a/patternutils/commands/targs.py b/patternutils/commands/targs.py index c1db5e4..3344e69 100644 --- a/patternutils/commands/targs.py +++ b/patternutils/commands/targs.py @@ -16,6 +16,7 @@ import click from .. import template from .. import config +from .. import utils def run_command( @@ -77,7 +78,7 @@ def targs( click.secho('jinja2 not installed!', fg='red', err=True) raise SystemExit - max_procs = os.cpu_count() * 2 if max_procs == 0 else max_procs + max_procs = utils.max_procs() if max_procs == 0 else max_procs stdout = subprocess.DEVNULL if redirect_stdout else None stderr = subprocess.DEVNULL if redirect_stderr else None diff --git a/patternutils/utils.py b/patternutils/utils.py new file mode 100644 index 0000000..da64edb --- /dev/null +++ b/patternutils/utils.py @@ -0,0 +1,6 @@ +import os + + +def max_procs() -> int: + cpu_count = os.cpu_count() + return 1 if cpu_count is None else cpu_count * 2 |