From 2667e44ca957ec75f61266305e375c011fc52cd8 Mon Sep 17 00:00:00 2001
From: Patrick Spek <p.spek@tyil.nl>
Date: Tue, 24 Oct 2023 15:31:45 +0200
Subject: Add shell functions for Python venvs

It is an ugly system, and it seems it is practically inescapable. Truly
an homage to the quality of the programming language and the devs who
love it without question.
---
 .config/shell/functions.d/venv | 48 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)
 create mode 100644 .config/shell/functions.d/venv

(limited to '.config')

diff --git a/.config/shell/functions.d/venv b/.config/shell/functions.d/venv
new file mode 100644
index 0000000..50a91e9
--- /dev/null
+++ b/.config/shell/functions.d/venv
@@ -0,0 +1,48 @@
+#!/bin/sh
+
+venv() {
+	if ! command -v "venv_$1" | grep -q " function "
+	then
+		cat <<EOF
+Usage:
+	venv on
+	venv off
+
+Manage the python venv for a given project or directory.
+
+Commands:
+	on	Enable the venv for \$PWD
+	off	Deactivate the current venv
+EOF
+
+		return 1
+	fi
+
+	"venv_$1"
+}
+
+venv_on() {
+	export PYTHON_VENV_DIR="$PWD/.venv"
+	export VIRTUAL_ENV_DISABLE_PROMPT=1
+
+	if [ ! -d "$PYTHON_VENV_DIR" ]
+	then
+		printf "Initializing venv directory at %s\n" "$PYTHON_VENV_DIR"
+		python3 -m venv "$PYTHON_VENV_DIR"
+	fi
+
+	. "$PYTHON_VENV_DIR/bin/activate"
+}
+
+venv_off() {
+	if [ -z "$PYTHON_VENV_DIR" ]
+	then
+		printf "No venv active?\n"
+		return 1
+	fi
+
+	deactivate
+
+	unset PYTHON_VENV_DIR
+	unset VIRTUAL_ENV_DISABLE_PROMPT
+}
-- 
cgit v1.1