miiocli: Handle unknown commands (Closes: #327) (#329)

rytilahti · Jun 2, 2018 · af2417a · af2417a
1 parent a639b12
commit af2417a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/miio/click_common.py b/miio/click_common.py
@@ -218,6 +218,9 @@ def command_callback(self, command, device, *args, **kwargs):
         return command.call(device, *args, **kwargs)
 
     def get_command(self, ctx, cmd_name):
+        if cmd_name not in self.commands:
+            ctx.fail('Unknown command (%s)' % cmd_name)
+
         cmd = self.commands[cmd_name]
         return self.commands[cmd_name].wrap(ctx, self.device_pass(partial(
             self.command_callback, cmd