base91.py

   1 import struct
   2
   3 base91_alphabet = ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M',
   4         'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z',
   5         'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm',
   6         'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z',
   7         '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '!', '#', '$',
   8         '%', '&', '(', ')', '*', '+', ',', '.', '/', ':', ';', '<', '=',
   9         '>', '?', '@', '[', ']', '^', '_', '`', '{', '|', '}', '~', '"']
  10
  11 decode_table = dict((v,k) for k,v in enumerate(base91_alphabet))
  12
  13 def decode(encoded_str):
  14     ''' Decode Base91 string to a bytearray '''
  15     v = -1
  16     b = 0
  17     n = 0
  18     out = bytearray()
  19     for strletter in encoded_str:
  20         if not strletter in decode_table:
  21             continue
  22         c = decode_table[strletter]
  23         if(v < 0):
  24             v = c
  25         else:
  26             v += c*91
  27             b |= v << n
  28             n += 13 if (v & 8191)>88 else 14
  29             while True:
  30                 out += struct.pack('B', b&255)
  31                 b >>= 8
  32                 n -= 8
  33                 if not n>7:
  34                     break
  35             v = -1
  36     if v+1:
  37         out += struct.pack('B', (b | v << n) & 255 )
  38     return out
  39
  40 def encode(bindata):
  41     ''' Encode a bytearray to a Base91 string '''
  42     l = len(bindata)
  43     b = 0
  44     n = 0
  45     out = ''
  46     for byte in bindata:
  47         b |= struct.unpack('B', byte)[0] << n
  48         n += 8
  49         if n>13:
  50             v = b & 8191
  51             if v > 88:
  52                 b >>= 13
  53                 n -= 13
  54             else:
  55                 v = b & 16383
  56                 b >>= 14
  57                 n -= 14
  58             out += base91_alphabet[v % 91] + base91_alphabet[v / 91]
  59     if n:
  60         out += base91_alphabet[b % 91]
  61         if n>7 or b>90:
  62             out += base91_alphabet[b / 91]
  63     return out
  64
  65
  66
  67
  68
  69