PythonでBase64を車輪の再開発してみた

Pythonの練習も兼ねて。
as3cryptoを参考に。
Pythonの場合StaticMethodにせず、直にdefを定義するべきだったのかな。
　

Base64

encode

3バイト入力して、6ビットずつに分割して、4バイトで出力する。
これで0〜63で表せる。
余ったら=を入れとく。
　

decode

4バイト入力して、8ビットずつに戻して、3バイトで出力する。
=は気にしない。
　

ソース

#!/usr/local/bin/python
#set encoding=utf8

class Base64(object):
	__CHARS = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/='

	@staticmethod
	def encode(input):
		i = 0
		output = ''
		while i<len(input):
			input_buf = [0, 0, 0]
			output_buf = [0, 0, 0, 0]
			j = 0
			while j<3 and i+j<len(input):
				input_buf[j] = ord(input[i+j])
				j += 1

			output_buf[0] = (input_buf[0] & 0xfc) >> 2
			output_buf[1] = ((input_buf[0] & 0x03) << 4) | ((input_buf[1]) >> 4)
			output_buf[2] = ((input_buf[1] & 0x0f) << 2) | ((input_buf[2]) >> 6)
			output_buf[3] = input_buf[2] & 0x3f

			for k in range(j, 3):
				output_buf[k+1] = 64

			for l in output_buf:
				output += Base64.__CHARS[l]

			i += 3
		return output

	@staticmethod
	def decode(input):
		i = 0
		output = ''
		while i<len(input):
			input_buf = [0, 0, 0, 0]
			output_buf = [0, 0, 0]
			j = 0
			while j<4 and i+j<len(input):
				input_buf[j] = Base64.__CHARS.index(input[i+j])
				j += 1

			output_buf[0] = (input_buf[0] << 2) | ((input_buf[1] & 0x30) >> 4)
			output_buf[1] = ((input_buf[1] & 0x0f) << 4) | ((input_buf[2] & 0x3c) >> 2)
			output_buf[2] = ((input_buf[2] & 0x03) << 6) | input_buf[3]

			for k in output_buf:
				if k==64:
					break
				output += chr(k)

			i += 4
		return output

s = 'あいうえお'
enc = Base64.encode(s)
dec = Base64.decode(enc)

print 's: %s' % s
print 'enc: %s' % enc
print 'dec: %s' % dec

実行結果

$ python base64.py 
s: あいうえお
enc: 44GC44GE44GG44GI44GK
dec: あいうえお