Fix check_beta error if cmd is not in beta args dict.

This commit is contained in:
Darko Poljak 2016-08-15 18:04:57 +02:00
parent c5e6ed041c
commit 5f825d1da9
1 changed files with 4 additions and 3 deletions

View File

@ -49,9 +49,10 @@ def check_beta(args_dict):
# raise error.
if not args_dict['beta']:
cmd = args_dict['command']
for arg in BETA_ARGS[cmd]:
if arg in args_dict:
raise cdist.CdistBetaRequired(cmd, arg)
if cmd in BETA_ARGS:
for arg in BETA_ARGS[cmd]:
if arg in args_dict:
raise cdist.CdistBetaRequired(cmd, arg)
def commandline():