add error handling around cbapi-{response,defense,protection} scripts

jgarman · jgarman · commit d92f2c58cc4f · 2017-06-22T23:00:49.000-04:00
diff --git a/src/cbapi/defense/cli.py b/src/cbapi/defense/cli.py
@@ -80,6 +80,12 @@ def main(args):
 
     opts = parser.parse_args(args)
     command = command_map.get(opts.command_name)
+    if not command:
+        parser.print_usage()
+        return
+
     command_method = command.get("method", None)
     if command_method:
         return command_method(opts)
+    else:
+        parser.print_usage()
diff --git a/src/cbapi/protection/cli.py b/src/cbapi/protection/cli.py
@@ -87,6 +87,12 @@ def main(args):
 
     opts = parser.parse_args(args)
     command = command_map.get(opts.command_name)
+    if not command:
+        parser.print_usage()
+        return
+
     command_method = command.get("method", None)
     if command_method:
         return command_method(opts)
+    else:
+        parser.print_usage()
diff --git a/src/cbapi/response/cli.py b/src/cbapi/response/cli.py
@@ -101,6 +101,12 @@ def main(args):
 
     opts = parser.parse_args(args)
     command = command_map.get(opts.command_name)
+    if not command:
+        parser.print_usage()
+        return
+
     command_method = command.get("method", None)
     if command_method:
         return command_method(opts)
+    else:
+        parser.print_usage()
