updated binary conversion module

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