fix parse error on extra-vars

fixes gh-33
spencergibb · Jun 8, 2015 · 49d8c12 · 49d8c12
1 parent 3542b1f
commit 49d8c12
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/lib/battleschool/__init__.py b/lib/battleschool/__init__.py
@@ -1,2 +1,2 @@
-__version__ = '0.6.0'
+__version__ = '0.6.1'
 __author__ = 'Spencer Gibb'
diff --git a/lib/battleschool/main.py b/lib/battleschool/main.py
@@ -124,22 +124,20 @@ def main(args, battleschool_dir=None):
     # setup default options
     sshpass = None
     sudopass = None
+    vault_pass = None
     options.remote_user = AC.DEFAULT_REMOTE_USER
     if not options.listhosts and not options.syntax and not options.listtasks:
         options.ask_pass = AC.DEFAULT_ASK_PASS
         options.ask_sudo_pass = options.ask_sudo_pass or AC.DEFAULT_ASK_SUDO_PASS
         passwds = utils.ask_passwords(ask_pass=options.ask_pass, become_ask_pass=options.ask_sudo_pass)
         sshpass = passwds[0]
         sudopass = passwds[1]
+        vault_pass = passwds[2]
         # if options.sudo_user or options.ask_sudo_pass:
         #     options.sudo = True
         options.sudo_user = AC.DEFAULT_SUDO_USER
-    if options.extra_vars and options.extra_vars[0] in '[{':
-        extra_vars = utils.json_loads(options.extra_vars)
-    elif options.extra_vars:
-        extra_vars = utils.parse_kv(options.extra_vars)
-    else:
-        extra_vars = None
+
+    extra_vars = utils.parse_extra_vars(options.extra_vars, vault_pass)
     only_tags = None  # options.tags.split(",")
 
     #-----------------------------------------------------------