From 45e26fda237441777d76f5b7362c84ce9f757c89 Mon Sep 17 00:00:00 2001 From: AnilDaoud Date: Thu, 23 May 2019 21:13:14 +0800 Subject: fixed learnfile switch and python3 compatibility (#247) --- cli/broadlink_cli | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) (limited to 'cli/broadlink_cli') diff --git a/cli/broadlink_cli b/cli/broadlink_cli index 1c3e231d69e5..1a8fe5703468 100755 --- a/cli/broadlink_cli +++ b/cli/broadlink_cli @@ -4,6 +4,8 @@ import broadlink import sys import argparse import time +import base64 +import codecs TICK = 32.84 IR_TOKEN = 0x26 @@ -120,7 +122,7 @@ if args.send: data = durations_to_broadlink(parse_durations(' '.join(args.data))) \ if args.durations else bytearray.fromhex(''.join(args.data)) dev.send_data(data) -if args.learn: +if args.learn or args.learnfile: dev.enter_learning() data = None print("Learning...") @@ -135,7 +137,8 @@ if args.learn: else ''.join(format(x, '02x') for x in bytearray(data)) if args.learn: print(learned) - print("Base64: " + base64.b64encode(learned.decode("hex"))) + decode_hex = codecs.getdecoder("hex_codec") + print("Base64: " + str(base64.b64encode(decode_hex(learned)[0]))) if args.learnfile: print("Saving to {}".format(args.learnfile)) with open(args.learnfile, "w") as text_file: -- cgit 1.4.1