Migrated to dedicated minion repo

[dotfiles.git] / invoke / setup-linux / tasks.py

diff --git a/invoke/setup-linux/tasks.py b/invoke/setup-linux/tasks.py
index 067c8115b2c680554b5e59484fc7c51dca02afb6..1e20b05edd21d876061939f87802c22f4ad23250 100644 (file)
--- a/invoke/setup-linux/tasks.py
+++ b/invoke/setup-linux/tasks.py
@@ -100,3 +100,26 @@ def add_vim_plugin(c, git_path, folder):
     c.run(f'ls -l {HOME}/.vim/pack/vendor/opt')
     print(f"Use `packadd {folder}` in .vimrc to enable")
 
+@task
+def ensure_env(c, env):
+    envs = f'{HOME}/envs'
+    path = f'{envs}/{env}'
+    if(not os.path.exists(envs)):
+        c.run(f'mkdir -p {envs}')
+
+    if(not os.path.exists(path)):
+       c.run(f'python3 -m venv {path}')
+       reqs = f'{DOTFILES}/reqs/{env}.txt'
+       c.run(f'{path}/bin/pip install -r {reqs}')
+
+    c.run(f'{path}/bin/pip freeze')
+
+@task
+def env_ansible(c):
+    ensure_env(c, 'ansible')
+
+
+@task
+def env_jupyter(c):
+    ensure_env(c, 'jupyter')
+