diff options
| author | Dennis Fink | 2018-04-07 10:30:35 +0200 |
|---|---|---|
| committer | Dennis Fink | 2018-04-07 10:30:35 +0200 |
| commit | 2683d479c8146cafaa5baa1baa3d3bf7df140151 (patch) | |
| tree | 2753e6c895dad2dd1985d0d709a43d26fba38351 | |
| parent | 285f39ea15261a3e2f0410373c8d66726d6f32c1 (diff) | |
| download | patternutils-2683d479c8146cafaa5baa1baa3d3bf7df140151.tar.gz patternutils-2683d479c8146cafaa5baa1baa3d3bf7df140151.zip | |
Add the utils module
| -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 |