From d1da57a2b981cd955c858069a919616af57af3a0 Mon Sep 17 00:00:00 2001 From: TinWoodman92 Date: Mon, 19 Jun 2023 22:14:00 -0500 Subject: updated binary conversion module --- dialoguer/binary_conversion.py | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) (limited to 'dialoguer') diff --git a/dialoguer/binary_conversion.py b/dialoguer/binary_conversion.py index dba0ea9..c39b78e 100644 --- a/dialoguer/binary_conversion.py +++ b/dialoguer/binary_conversion.py @@ -1,12 +1,10 @@ -cls_str = type("string") -cls_int = type(1) -cls_byt = type(b'\x00') - def convert_to_binary(data): bin_data = None - if type(data) == cls_str: + if type(data) == str: + bin_data = reversed(data) + bin_data = "".join(bin_data) n = '' - for s in data: + for s in bin_data: m = ord(s) m = hex(m) n += m.lstrip('0x') @@ -24,7 +22,7 @@ def convert_to_binary(data): m = bytes(m, 'utf-8') n += m bin_data = n - elif type(data) == cls_int: + elif type(data) == int: bin_data = bin(data) bin_data = bin_data.lstrip('0b') bin_data = bin_data.zfill(8) @@ -40,7 +38,7 @@ def convert_to_binary(data): def convert_from_binary(bin_data, cls_type): - if cls_type == cls_str: + if cls_type == str: n = '' for b in bin_data: n = f'{b}{n}' #for b in bin_data: n = f'{n}{b}' @@ -51,10 +49,12 @@ def convert_from_binary(bin_data, cls_type): data = bytearray.fromhex(data) data = data.decode() data = ''.join(reversed(data)) - elif cls_type == cls_int: + elif cls_type == int: n = '' - for b in bin_data: n = f'{n}{b}' + for b in bin_data: n = f'{b}{n}' data = n data = int(data, 2) + elif cls_type == bool: + data = bool(bin_data[0]) return data -- cgit v1.2.3