cleanup of yaml print output if yaml package isnt installed

markllama · Dec 10, 2016 · 8c66d32 · 8c66d32
1 parent 894ae11
commit 8c66d32
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/cli4/cli4.py b/cli4/cli4.py
@@ -194,6 +194,8 @@ def cli4(args):
         elif opt in ('-j', '--json'):
             output = 'json'
         elif opt in ('-y', '--yaml'):
+            if yaml is None:
+                exit('cli4: install yaml support')
             output = 'yaml'
         elif opt in ('-r', '--raw'):
             raw = True
@@ -365,6 +367,6 @@ def cli4(args):
 
     if output == 'json':
         print(json.dumps(results, indent=4, sort_keys=True))
-    if output == 'yaml' and yaml is not None:
+    if output == 'yaml':
         print(yaml.safe_dump(results))