Migrate miplug cli to the new ChuangmiPlug class (Fixes: rytilahti#296)

syssi · May 20, 2018 · cb187f7 · cb187f7
1 parent b5738af
commit cb187f7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/miio/plug_cli.py b/miio/plug_cli.py
@@ -11,7 +11,7 @@
                                validate_token, )
 
 _LOGGER = logging.getLogger(__name__)
-pass_dev = click.make_pass_decorator(miio.Plug)
+pass_dev = click.make_pass_decorator(miio.ChuangmiPlug)
 
 
 @click.group(invoke_without_command=True, cls=ExceptionHandlerGroup)
@@ -35,7 +35,7 @@ def cli(ctx, ip: str, token: str, debug: int):
         click.echo("You have to give ip and token!")
         sys.exit(-1)
 
-    dev = miio.Plug(ip, token, debug)
+    dev = miio.ChuangmiPlug(ip, token, debug)
     _LOGGER.debug("Connecting to %s with token %s", ip, token)
 
     ctx.obj = dev