if /snacks/wireguard/scripts/is_root_namespace.py
then
    export PS1='%n@%m % %c %#'
else
    export PS1='%F{red}<vpn>%f %n@%m % %c %#'
fi