diff options
Diffstat (limited to 'dialoguer')
| -rw-r--r-- | dialoguer/binary_conversion.py | 20 |
1 files changed, 10 insertions, 10 deletions
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 |