Introduce new lick parameter type for hexadecimal strings

Author: syssi

miio/airconditioningcompanion.py

@@ -4,7 +4,7 @@
 
 import click
 
-from .click_common import command, format_output, EnumType
+from .click_common import command, format_output, EnumType, HexStringParamType
 from .device import Device, DeviceException
 
 _LOGGER = logging.getLogger(__name__)
@@ -310,8 +310,8 @@ def learn_stop(self, slot: int=STORAGE_SLOT_ID):
         return self.send("end_ir_learn", [slot])
 
     @command(
-        click.argument("model", type=bytes),
-        click.argument("code", type=bytes),
+        click.argument("model", type=HexStringParamType),
+        click.argument("code", type=HexStringParamType),
         default_output=format_output("Sending the supplied infrared command")
     )
     def send_ir_code(self, model: bytes, code: bytes, slot: int=0):
@@ -346,7 +346,7 @@ def send_command(self, command: str):
         return self.send("send_cmd", [str(command)])
 
     @command(
-        click.argument("model", type=str),
+        click.argument("model", type=HexStringParamType),
         click.argument("power", type=EnumType(Power, False)),
         click.argument("operation_mode", type=EnumType(OperationMode, False)),
         click.argument("target_temperature", type=int),

miio/click_common.py

@@ -93,6 +93,14 @@ def get_metavar(self, param):
         return ("_".join(word)).upper()
 
 
+class HexStringParamType(click.ParamType):
+    def convert(self, value, param, ctx):
+        try:
+            return bytes.fromhex(value)
+        except ValueError:
+            self.fail('%s is not a valid hexadecimal string' % value, param, ctx)
+
+
 class GlobalContextObject:
     def __init__(self, debug: int=0, output: callable=None):
         self.debug = debug