base64: Reduce module size and improve base32 decoding speed.

python-stdlib/base64/base64.py

@@ -171,43 +171,25 @@ def urlsafe_b64decode(s):
 
 
 # Base32 encoding/decoding must be done in Python
-_b32alphabet = {
-    0: b"A",
-    9: b"J",
-    18: b"S",
-    27: b"3",
-    1: b"B",
-    10: b"K",
-    19: b"T",
-    28: b"4",
-    2: b"C",
-    11: b"L",
-    20: b"U",
-    29: b"5",
-    3: b"D",
-    12: b"M",
-    21: b"V",
-    30: b"6",
-    4: b"E",
-    13: b"N",
-    22: b"W",
-    31: b"7",
-    5: b"F",
-    14: b"O",
-    23: b"X",
-    6: b"G",
-    15: b"P",
-    24: b"Y",
-    7: b"H",
-    16: b"Q",
-    25: b"Z",
-    8: b"I",
-    17: b"R",
-    26: b"2",
-}
-
-_b32tab = [v[0] for k, v in sorted(_b32alphabet.items())]
-_b32rev = dict([(v[0], k) for k, v in _b32alphabet.items()])
+_b32tab = b"ABCDEFGHIJKLMNOPQRSTUVWXYZ234567"
+# _b32rev = bytearray('\xFF' * 256); for i, j in enumerate(_b32tab): _b32rev[j] = i
+_b32rev = (
+    b"\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff"
+    b"\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff"
+    b"\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\x1a\x1b\x1c\x1d"
+    b"\x1e\x1f\xff\xff\xff\xff\xff\xff\xff\xff\xff\x00\x01\x02\x03\x04\x05\x06"
+    b"\x07\x08\x09\x0a\x0b\x0c\x0d\x0e\x0f\x10\x11\x12\x13\x14\x15\x16\x17\x18"
+    b"\x19\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff"
+    b"\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff"
+    b"\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff"
+    b"\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff"
+    b"\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff"
+    b"\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff"
+    b"\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff"
+    b"\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff"
+    b"\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff\xff"
+    b"\xff\xff\xff\xff"
+)
 
 
 def b32encode(s):
@@ -303,10 +285,10 @@ def b32decode(s, casefold=False, map01=None):
     acc = 0
     shift = 35
     for c in s:
-        val = _b32rev.get(c)
-        if val is None:
+        val = _b32rev[c]
+        if val == 0xFF:
             raise binascii.Error("Non-base32 digit found")
-        acc += _b32rev[c] << shift
+        acc += val << shift
         shift -= 5
         if shift < 0:
             parts.append(binascii.unhexlify(bytes("%010x" % acc, "ascii")))
@@ -441,11 +423,9 @@ def main():
         sys.stdout = sys.stderr
         print(msg)
         print(
-            """usage: %s [-d|-e|-u|-t] [file|-]
+            """usage: %s [-d|-e|-u] [file|-]
         -d, -u: decode
-        -e: encode (default)
-        -t: encode and decode string 'Aladdin:open sesame'"""
-            % sys.argv[0]
+        -e: encode (default)"""
         )
         sys.exit(2)
     func = encode
@@ -456,25 +436,12 @@ def main():
             func = decode
         if o == "-u":
             func = decode
-        if o == "-t":
-            test()
-            return
     if args and args[0] != "-":
         with open(args[0], "rb") as f:
             func(f, sys.stdout.buffer)
     else:
         func(sys.stdin.buffer, sys.stdout.buffer)
 
 
-def test():
-    s0 = b"Aladdin:open sesame"
-    print(repr(s0))
-    s1 = encodebytes(s0)
-    print(repr(s1))
-    s2 = decodebytes(s1)
-    print(repr(s2))
-    assert s0 == s2
-
-
 if __name__ == "__main__":
     main()

python-stdlib/base64/manifest.py

@@ -1,4 +1,4 @@
-metadata(version="3.3.6")
+metadata(version="3.3.7")
 
 require("binascii")
 

python-stdlib/base64/test_base64.py

@@ -12,7 +12,14 @@
 if d != b"zlutoucky kun upel dabelske ody":
     raise Exception("Error")
 
-base64.test()
+s0 = b"Aladdin:open sesame"
+s1 = base64.encodebytes(s0)
+if s1 != b"QWxhZGRpbjpvcGVuIHNlc2FtZQ==\n":
+    raise Exception("Error")
+print(s1)
+s2 = base64.decodebytes(s1)
+if s0 != s2:
+    raise Exception("Error")
 
 binary = b"\x99\x10\xaa"
 b = base64.b64encode(binary)
@@ -33,4 +40,15 @@
 if b != b"zlutoucky kun upel dabelske ody":
     raise Exception("Error")
 
+binary = b"\x99\x10\xaa"
+b = base64.b32encode(binary)
+if b != b"TEIKU===":
+    raise Exception("Error")
+
+d = base64.b32decode(b)
+print(d)
+if d != binary:
+    raise Exception("Error")
+
+
 print("OK")