mirror of
https://github.com/ohmyzsh/ohmyzsh.git
synced 2024-11-30 09:40:09 +00:00
➕ Add viper-env plugin.
This commit is contained in:
parent
69507c9518
commit
78f8066681
3 changed files with 179 additions and 0 deletions
65
plugins/viper-env/README.md
Normal file
65
plugins/viper-env/README.md
Normal file
|
@ -0,0 +1,65 @@
|
|||
# Viper Env Z-Shel plugin
|
||||
|
||||
Automatically activates and deactivates python virtualenv upon cd in and out.
|
||||
|
||||
## Inspiration
|
||||
|
||||
Based on [blueray](https://stackoverflow.com/users/1772898/blueray)'s [answer](https://stackoverflow.com/a/63955939/11685534), I decided to go one step further and implement it as a Z-Shell plugin.
|
||||
|
||||
## Usage
|
||||
<!-- [![asciicast](https://asciinema.org/a/4iMwcKfBS1dc1EgI1FihrDVxT.svg)](https://asciinema.org/a/4iMwcKfBS1dc1EgI1FihrDVxT) -->
|
||||
|
||||
![Alt text](./make_animation/assets/final.svg)
|
||||
|
||||
## Example
|
||||
```zsh
|
||||
> viper-env help
|
||||
|
||||
Description:
|
||||
Automatically activates and deactivates python virtualenv upon cd in and out.
|
||||
|
||||
Dependencies:
|
||||
- zsh
|
||||
- python
|
||||
|
||||
Example usage:
|
||||
# Create virtual environment
|
||||
python -m venv .venv
|
||||
# Save current dir
|
||||
current_dir=$(basename $PWD)
|
||||
# Exit current directory
|
||||
cd ..
|
||||
# Reenter it
|
||||
cd $current_dir
|
||||
```
|
||||
|
||||
## Instalation
|
||||
|
||||
### Oh my Zsh
|
||||
|
||||
Git clone this repository to the Oh my Zsh custom plugins folder.
|
||||
|
||||
Add plugin to plugins directive in `~/.zshrc`
|
||||
```zsh
|
||||
plugins=(
|
||||
# put local oh-my-zsh plugins here
|
||||
viper-env
|
||||
)
|
||||
|
||||
source $HOME/.oh-my-zsh/oh-my-zsh.sh
|
||||
```
|
||||
|
||||
### Antigen
|
||||
It is recommended to use a `.antigenrc` file. Then add the following to it:
|
||||
|
||||
```zsh
|
||||
antigen bundle DanielAtKrypton/viper-env
|
||||
|
||||
# Apply changes
|
||||
antigen apply
|
||||
```
|
||||
|
||||
Make sure in your `.zshrc` Antigen is loading the `.antigenrc` file as follows:
|
||||
```zsh
|
||||
antigen init ~/.antigenrc
|
||||
```
|
1
plugins/viper-env/make_animation/assets/final.svg
Normal file
1
plugins/viper-env/make_animation/assets/final.svg
Normal file
File diff suppressed because one or more lines are too long
After Width: | Height: | Size: 1.7 MiB |
113
plugins/viper-env/viper-env.plugin.zsh
Normal file
113
plugins/viper-env/viper-env.plugin.zsh
Normal file
|
@ -0,0 +1,113 @@
|
|||
# Colors based on:
|
||||
# https://www.lihaoyi.com/post/BuildyourownCommandLinewithANSIescapecodes.html#16-colors
|
||||
export COLOR_BLACK='\033[0;30m'
|
||||
export COLOR_BRIGHT_BLACK='\u001b[30;1m'
|
||||
export COLOR_RED='\033[0;31m'
|
||||
export COLOR_BRIGHT_RED='\u001b[31;1m'
|
||||
export COLOR_GREEN='\033[0;32m'
|
||||
export COLOR_BRIGHT_GREEN='\u001b[32;1m'
|
||||
export COLOR_YELLOW='\033[0;33m'
|
||||
export COLOR_BRIGHT_YELLOW='\u001b[33;1m'
|
||||
export COLOR_BLUE='\033[0;34m'
|
||||
export COLOR_BRIGHT_BLUE='\u001b[34;1m'
|
||||
export COLOR_VIOLET='\033[0;35m'
|
||||
export COLOR_BRIGHT_VIOLET='\u001b[35;1m'
|
||||
export COLOR_CYAN='\033[0;36m'
|
||||
export COLOR_BRIGHT_CYAN='\u001b[36;1m'
|
||||
export COLOR_WHITE='\033[0;37m'
|
||||
export COLOR_NC='\033[0m' # No Color
|
||||
|
||||
function activate_env_execution() {
|
||||
local virtualenv_directory=$1
|
||||
local d=$2
|
||||
local relative_env_path=$3
|
||||
local full_virtualenv_directory=$d/$virtualenv_directory
|
||||
|
||||
echo Activating virtual environment ${COLOR_BRIGHT_VIOLET}$relative_env_path${COLOR_NC}
|
||||
source $full_virtualenv_directory/bin/activate
|
||||
}
|
||||
|
||||
function get_env_path(){
|
||||
echo "$(basename "$1")/$2"
|
||||
}
|
||||
|
||||
function get_envs(){
|
||||
local output="$(ls (.*|*)/bin/pip(.x))" &> /dev/null
|
||||
local candidate_envs_found
|
||||
if [ $? -eq 0 ]; then
|
||||
candidate_envs_found=($output)
|
||||
else
|
||||
candidate_envs_found=()
|
||||
fi
|
||||
envs_found=()
|
||||
for env in "${candidate_envs_found[@]}"
|
||||
do
|
||||
local env_name="$(dirname $env)"
|
||||
ls $env_name/activate &> /dev/null
|
||||
if [ $? -eq 0 ]; then
|
||||
envs_found+=("$(dirname $env_name)")
|
||||
fi
|
||||
done
|
||||
}
|
||||
|
||||
function activate_env(){
|
||||
local current_dir=$1
|
||||
get_envs
|
||||
# echo "Envs found: $envs_found"
|
||||
if [ ${#envs_found[@]} -gt 0 ]; then
|
||||
# Use first found only!
|
||||
local env_name="${envs_found[1]}"
|
||||
# local env_name="$(dirname "$(dirname $env_to_use)")"
|
||||
local relative_activating_env_path="$(get_env_path $current_dir $env_name)"
|
||||
activate_env_execution $env_name $current_dir $relative_activating_env_path
|
||||
fi
|
||||
}
|
||||
|
||||
function automatically_activate_python_env() {
|
||||
local current_dir="$PWD"
|
||||
local env_var="$VIRTUAL_ENV"
|
||||
if [[ -z $env_var ]] ; then
|
||||
activate_env $current_dir
|
||||
else
|
||||
parentdir="$(dirname $env_var)"
|
||||
if [[ $current_dir/ != $parentdir/* ]] ; then
|
||||
local deactivating_relative_env_path="$(realpath --relative-to=$current_dir $env_var)"
|
||||
echo Deactivating virtual environment ${COLOR_BRIGHT_VIOLET}$deactivating_relative_env_path${COLOR_NC}
|
||||
deactivate
|
||||
activate_env $current_dir
|
||||
fi
|
||||
fi
|
||||
}
|
||||
autoload -Uz add-zsh-hook
|
||||
add-zsh-hook precmd automatically_activate_python_env
|
||||
|
||||
__viper-env_help () {
|
||||
printf "Description:
|
||||
${COLOR_BRIGHT_BLACK}Automatically activates and deactivates python virtualenv upon cd in and out.${COLOR_NC}
|
||||
|
||||
"
|
||||
printf "Dependencies:
|
||||
${COLOR_BRIGHT_BLACK}- zsh
|
||||
- python${COLOR_NC}
|
||||
|
||||
"
|
||||
printf "Example usage:
|
||||
${COLOR_BRIGHT_BLACK}# Create virtual environment${COLOR_NC}
|
||||
${COLOR_GREEN}python${COLOR_NC} -m env .env
|
||||
${COLOR_BRIGHT_BLACK}# Save current dir${COLOR_NC}
|
||||
current_dir=${COLOR_VIOLET}\$(${COLOR_GREEN}basename ${COLOR_YELLOW}"${COLOR_NC}\$PWD${COLOR_YELLOW}"${COLOR_VIOLET})${COLOR_NC}
|
||||
${COLOR_BRIGHT_BLACK}# Exit current directory${COLOR_NC}
|
||||
${COLOR_GREEN}cd${COLOR_NC} ..
|
||||
${COLOR_BRIGHT_BLACK}# Reenter it${COLOR_NC}
|
||||
${COLOR_GREEN}cd${COLOR_NC} \$current_dir
|
||||
"
|
||||
}
|
||||
|
||||
viper-env_runner() {
|
||||
if [[ $@ == "help" || $@ == "h" ]]; then
|
||||
__viper-env_help
|
||||
#elif [[ $@ == "something" || $@ == "alias" ]]; then
|
||||
fi
|
||||
}
|
||||
|
||||
alias viper-env='viper-env_runner'
|
Loading…
Reference in a new issue