From 2683d479c8146cafaa5baa1baa3d3bf7df140151 Mon Sep 17 00:00:00 2001 From: Dennis Fink Date: Sat, 7 Apr 2018 10:30:35 +0200 Subject: Add the utils module --- patternutils/commands/targs.py | 3 ++- patternutils/utils.py | 6 ++++++ 2 files changed, 8 insertions(+), 1 deletion(-) create mode 100644 patternutils/utils.py 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 -- cgit v1.3.1