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

9 years ago · 5f825d1da9
parent c5e6ed041c
commit 5f825d1da9
1 changed files with 4 additions and 3 deletions
--- a/scripts/cdist
+++ b/scripts/cdist
@ -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():