1 files changed, 7 insertions, 6 deletions
diff --git a/upgrade.sh b/upgrade.sh
index aa5e693c..48849b82 100755
--- a/upgrade.sh
+++ b/upgrade.sh
@@ -1,12 +1,13 @@
 #!/bin/bash
 
-if [ -z "$1" ]; then
-  echo "$0 <host(s)>"
+if [ -z "$1" ] || [ -z "$2" ]; then
+  echo "$0 <environment> <host(s)>"
   exit 1
 fi
-
-host="$1"
+env="$1"
+shift
+hosts="$1"
 shift
 
-echo "######## upgrading host(s) '$host' ########"
-exec ansible-playbook -e "myname=$host" -e "myrole=upgrade" $@ generic.yaml
+echo "######## upgrading host(s) '$hosts' in environment '$env' ########"
+exec ansible-playbook -e "myhosts=$hosts" -e "myrole=upgrade" $@ "$env/generic.yaml"