Minor changes

dhondta · dhondta · commit b13c842f533a · 2020-07-20T17:04:49.000+02:00
diff --git a/codext/__common__.py b/codext/__common__.py
@@ -423,7 +423,7 @@ def list_encodings():
     """ Get a list of codext's added encodings from the local registry. """
     enc = []
     for search_function in __codecs_registry:
-        enc.append(search_function.__name__)
+        enc.append(search_function.__name__.replace("_", "-"))
     return enc
 
 
diff --git a/codext/binary/baudot.py b/codext/binary/baudot.py
@@ -190,7 +190,7 @@ def baudot_encode(alphabet=None, spaced=False, tape=False):
     ename = "baudot" + ("-spaced" if spaced else "-tape" if tape else "")
     alphabet, states, func = _handle_alphabet(alphabet)
     def encode(text, errors="strict"):
-        s, l, state, seen_states = "", len(text), None, []
+        s, l, state, seen_states = "", len(b(text)), None, []
         for i, c in enumerate(text):
             # if the state is undefined yet, find the relevant alphabet
             if state is None:
@@ -239,7 +239,7 @@ def baudot_decode(alphabet=None, spaced=False, tape=False):
     alphabet = {st: {v: k for k, v in alph.items()} for st, alph in alphabet.items()}
     states = {v: k for k, v in states.items()}
     def decode(text, errors="strict"):
-        s, l = "", len(text)
+        s, l = "", len(b(text))
         if spaced:
             text = text.replace(" ", "")
         elif tape:
diff --git a/codext/binary/excess3.py b/codext/binary/excess3.py
@@ -25,16 +25,16 @@
 
 
 def excess3_encode(text, errors="strict"):
-    r, b = "", ""
+    r, bits = "", ""
     for c in text:
         for i in str(ord(c)).zfill(3):
-            b += CODE[i]
-            if len(b) == 8:
-                r += chr(int(b, 2))
-                b = ""
-    if len(b) > 0:
-        r += chr(int(b + "0000", 2))
-    return r, len(text)
+            bits += CODE[i]
+            if len(bits) == 8:
+                r += chr(int(bits, 2))
+                bits = ""
+    if len(bits) > 0:
+        r += chr(int(bits + "0000", 2))
+    return r, len(b(text))
 
 
 def excess3_decode(text, errors="strict"):
