refactor & fix: command not found: pipenv

Author: TTWShell

hobbit/devtools.py

@@ -1,9 +1,7 @@
-import os
 import click
-from subprocess import run
 
-from .handlers import echo
-from . import HobbitGroup, CONTEXT_SETTINGS, ROOT_PATH
+from . import HobbitGroup, CONTEXT_SETTINGS
+from .handlers.devtools import dev_init
 
 
 class BootstrapGroup(HobbitGroup):
@@ -30,20 +28,4 @@ def dev():
 def init(ctx, all_, hooks, pipenv):
     """Init dev env: git hooks, pyenv install etc.
     """
-    run('git init', shell=True)
-
-    if all_ or hooks:
-        HOOKS_PATH = os.path.join(ROOT_PATH, 'static', 'hooks')
-        run(f'cp -r {HOOKS_PATH}/* .git/hooks', shell=True)
-
-    pipenv_cmds = [
-        'pipenv install --dev pytest pytest-cov pytest-env flake8',
-    ]
-    if 'requirements.txt' in os.listdir():
-        pipenv_cmds.insert(0, 'pipenv install -r requirements.txt --pre')
-
-    cmd = ' && '.join(pipenv_cmds)
-    # force pipenv to ignore that environment and create its own instead
-    if all_ or pipenv:
-        run(cmd, shell=True)
-        echo('Please run: pipenv shell')
+    dev_init(all_, hooks, pipenv)

hobbit/handlers/devtools.py

@@ -0,0 +1,36 @@
+import os
+import sys
+from subprocess import run, PIPE, STDOUT
+
+import click
+
+from hobbit import ROOT_PATH
+
+
+def dev_init(all_, hooks, pipenv):
+    run('git init', shell=True)
+
+    if all_ or hooks:
+        HOOKS_PATH = os.path.join(ROOT_PATH, 'static', 'hooks')
+        run(f'cp -r {HOOKS_PATH}/* .git/hooks', shell=True)
+
+    if all_ or pipenv:
+        sub = run('which pipenv', shell=True, stdout=PIPE, stderr=STDOUT)
+        if sub.returncode != 0:
+            click.echo(click.style('cmd pipenv not exist.', fg='red'))
+            sys.exit(sub.returncode)
+        pipenv_path = sub.stdout.strip().decode('utf8')
+
+        pipenv_cmds = [
+            f'{pipenv_path} install --dev pytest pytest-cov pytest-env flake8',
+        ]
+        if 'requirements.txt' in os.listdir():
+            pipenv_cmds.insert(
+                0, f'{pipenv_path} install -r requirements.txt --pre')
+
+        cmd = ' && '.join(pipenv_cmds)
+        click.echo(click.style(cmd, fg='green'))
+        # force pipenv to ignore that environment and create its own instead
+        env = os.environ.copy()
+        env.update({'PIPENV_IGNORE_VIRTUALENVS': '1'})
+        run(cmd, shell=True, env=env)

tox.ini

@@ -30,6 +30,7 @@ deps =
     pytest-cov
     pytest-env
     flake8
+    pipenv
 commands =
     flake8 .
     mypy hobbit hobbit_core tests