Organize error checking in virtualenwrapper plugin

plugins/virtualenvwrapper/virtualenvwrapper.plugin.zsh

@@ -6,14 +6,21 @@ elif [[ -f "/etc/bash_completion.d/virtualenvwrapper" ]]; then
   virtualenvwrapper="/etc/bash_completion.d/virtualenvwrapper"
   source "/etc/bash_completion.d/virtualenvwrapper"
 else
-  print "zsh virtualenvwrapper plugin: Cannot find ${virtualenvwrapper}. Please install with \`pip install virtualenvwrapper\`."
+  print "zsh virtualenvwrapper plugin: Cannot find ${virtualenvwrapper}.\n"\
+        "Please install with \`pip install virtualenvwrapper\`" >&2
+  return
+fi
+if ! type workon &>/dev/null; then
+  print "zsh virtualenvwrapper plugin: shell function 'workon' not defined.\n"\
+        "Please check ${virtualenvwrapper}" >&2
   return
 fi
 
-if type workon 2>&1 >/dev/null; then
 if [[ "$WORKON_HOME" == "" ]]; then
-    echo "\$WORKON_HOME is not defined so ZSH plugin virtualenvwrapper will not work"
+  print "\$WORKON_HOME is not defined so ZSH plugin virtualenvwrapper will not work" >&2
-  else
+  return
+fi
+
 if [[ ! $DISABLE_VENV_CD -eq 1 ]]; then
   # Automatically activate Git projects's virtual environments based on the
   # directory name of the project. Virtual environment name can be overridden
@@ -66,7 +73,3 @@ if type workon 2>&1 >/dev/null; then
     set -A chpwd_functions workon_cwd
   fi
 fi
-  fi
-else
-  print "zsh virtualenvwrapper plugin: shell function 'workon' not defined. Please check ${virtualenvwrapper}." >&2
-fi