Add helper to setup favorite virtual environments

diff --git a/invoke/setup-linux/tasks.py b/invoke/setup-linux/tasks.py
index 067c8115b2c680554b5e59484fc7c51dca02afb6..e5fd14c1ce6cc3bfca2a993d6fc0252c8107a359 100644 (file)
--- a/invoke/setup-linux/tasks.py
+++ b/invoke/setup-linux/tasks.py
@@ -100,3 +100,22 @@ 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_jupyter(c):
+    ensure_env(c, 'jupyter')
+

diff --git a/reqs/jupyter.txt b/reqs/jupyter.txt
new file mode 100644 (file)
index 0000000..647370f
--- /dev/null
+++ b/reqs/jupyter.txt
@@ -0,0 +1,4 @@
+jupyterlab
+notebook
+jupyter_console
+pandas_ods_reader