zenhan source code

#!/usr/bin/python
# -*- coding: euc-jp -*-
import sys

ASCII = 1
DIGIT = 2
KANA  = 4
ALL = ASCII | DIGIT | KANA

# list of ZENKAKU characters
z_ascii = [u"ａ", u"ｂ", u"ｃ", u"ｄ", u"ｅ", u"ｆ", u"ｇ", u"ｈ", u"ｉ",
           u"ｊ", u"ｋ", u"ｌ", u"ｍ", u"ｎ", u"ｏ", u"ｐ", u"ｑ", u"ｒ",
           u"ｓ", u"ｔ", u"ｕ", u"ｖ", u"ｗ", u"ｘ", u"ｙ", u"ｚ",
           u"Ａ", u"Ｂ", u"Ｃ", u"Ｄ", u"Ｅ", u"Ｆ", u"Ｇ", u"Ｈ", u"Ｉ",
           u"Ｊ", u"Ｋ", u"Ｌ", u"Ｍ", u"Ｎ", u"Ｏ", u"Ｐ", u"Ｑ", u"Ｒ",
           u"Ｓ", u"Ｔ", u"Ｕ", u"Ｖ", u"Ｗ", u"Ｘ", u"Ｙ", u"Ｚ",
           u"！", u"”", u"＃", u"＄", u"％", u"＆", u"’", u"（", u"）",
           u"＊", u"＋", u"，", u"−", u"．", u"／", u"：", u"；", u"＜",
           u"＝", u"＞", u"？", u"＠", u"［", u"￥", u"］", u"＾", u"＿",
           u"‘", u"｛", u"｜", u"｝", u"〜", u"　"]

z_digit = [u"０", u"１", u"２", u"３", u"４",
	   u"５", u"６", u"７", u"８", u"９"]

z_kana = [u"ア", u"イ", u"ウ", u"エ", u"オ",
          u"カ", u"キ", u"ク", u"ケ", u"コ",
          u"サ", u"シ", u"ス", u"セ", u"ソ",
          u"タ", u"チ", u"ツ", u"テ", u"ト",
          u"ナ", u"ニ", u"ヌ", u"ネ", u"ノ",
          u"ハ", u"ヒ", u"フ", u"ヘ", u"ホ",
          u"マ", u"ミ", u"ム", u"メ", u"モ",
          u"ヤ", u"ユ", u"ヨ",
          u"ラ", u"リ", u"ル", u"レ", u"ロ",
          u"ワ", u"ヲ", u"ン",
          u"ァ", u"ィ", u"ゥ", u"ェ", u"ォ",
          u"ッ", u"ャ", u"ュ", u"ョ", u"ヴ",
          u"ガ", u"ギ", u"グ", u"ゲ", u"ゴ",
          u"ザ", u"ジ", u"ズ", u"ゼ", u"ゾ",
          u"ダ", u"ヂ", u"ヅ", u"デ", u"ド",
          u"バ", u"ビ", u"ブ", u"ベ", u"ボ",
          u"パ", u"ピ", u"プ", u"ペ", u"ポ",
	  u"。", u"、", u"・", u"゛", u"゜", u"「", u"」", u"ー"]

# list of HANKAKU characters
h_ascii = [u"a", u"b", u"c", u"d", u"e", u"f", u"g", u"h", u"i",
           u"j", u"k", u"l", u"m", u"n", u"o", u"p", u"q", u"r",
           u"s", u"t", u"u", u"v", u"w", u"x", u"y", u"z",
           u"A", u"B", u"C", u"D", u"E", u"F", u"G", u"H", u"I",
           u"J", u"K", u"L", u"M", u"N", u"O", u"P", u"Q", u"R",
           u"S", u"T", u"U", u"V", u"W", u"X", u"Y", u"Z",
           u"!", u'"', u"#", u"$", u"%", u"&", u"'", u"(", u")",
           u"*", u"+", u",", u"-", u".", u"/", u":", u";", u"<",
           u"=", u">", u"?", u"@", u"[", u"\\", u"]", u"^", u"_",
           u"`", u"{", u"|", u"}", u"~", u" "]

h_digit = [u"0", u"1", u"2", u"3", u"4", u"5", u"6", u"7", u"8", u"9"]

h_kana = [u"ｱ", u"ｲ", u"ｳ", u"ｴ", u"ｵ",
          u"ｶ", u"ｷ", u"ｸ", u"ｹ", u"ｺ",
          u"ｻ", u"ｼ", u"ｽ", u"ｾ", u"ｿ",
          u"ﾀ", u"ﾁ", u"ﾂ", u"ﾃ", u"ﾄ",
          u"ﾅ", u"ﾆ", u"ﾇ", u"ﾈ", u"ﾉ",
          u"ﾊ", u"ﾋ", u"ﾌ", u"ﾍ", u"ﾎ",
          u"ﾏ", u"ﾐ", u"ﾑ", u"ﾒ", u"ﾓ",
          u"ﾔ", u"ﾕ", u"ﾖ",
          u"ﾗ", u"ﾘ", u"ﾙ", u"ﾚ", u"ﾛ",
          u"ﾜ", u"ｦ", u"ﾝ",
          u"ｧ", u"ｨ", u"ｩ", u"ｪ", u"ｫ",
          u"ｯ", u"ｬ", u"ｭ", u"ｮ", u"ｳﾞ",
          u"ｶﾞ", u"ｷﾞ", u"ｸﾞ", u"ｹﾞ", u"ｺﾞ",
          u"ｻﾞ", u"ｼﾞ", u"ｽﾞ", u"ｾﾞ", u"ｿﾞ",
          u"ﾀﾞ", u"ﾁﾞ", u"ﾂﾞ", u"ﾃﾞ", u"ﾄﾞ",
          u"ﾊﾞ", u"ﾋﾞ", u"ﾌﾞ", u"ﾍﾞ", u"ﾎﾞ",
          u"ﾊﾟ", u"ﾋﾟ", u"ﾌﾟ", u"ﾍﾟ", u"ﾎﾟ",
	  u"｡", u"､", u"･", u"ﾞ", u"ﾟ", u"｢", u"｣", u"ｰ"]

# maps of ascii
zh_ascii = {}
hz_ascii = {}

for i in range(len(z_ascii)):
    zh_ascii[z_ascii[i]] = h_ascii[i]
    hz_ascii[h_ascii[i]] = z_ascii[i]

del z_ascii, h_ascii

# maps of digit
zh_digit = {}
hz_digit = {}

for i in range(len(z_digit)):
    zh_digit[z_digit[i]] = h_digit[i]
    hz_digit[h_digit[i]] = z_digit[i]

del z_digit, h_digit

# maps of KANA
zh_kana = {}
hz_kana = {}

for i in range(len(z_kana)):
    zh_kana[z_kana[i]] = h_kana[i]
    hz_kana[h_kana[i]] = z_kana[i]

del z_kana, h_kana

# function check text
# argument and return: unicode string
def check_text(t):
    if isinstance(t, basestring) and t:
	return t
    else:
	sys.exit("Sorry... You must set UNICODE String.")

# function check convertion mode
# argument: integer
# return: integer between 0 to 7
def check_mode(m):
    if isinstance(m, int) and m >= 0 and m <= 7:
	return m
    else:
	sys.exit("Sorry... You set invalid mode.")

# function convert from ZENKAKU to HANKAKU
# argument and return: unicode string
def z2h(text="", mode=ALL, ignore=()):
    converted = ""

    text = check_text(text)
    mode = check_mode(mode)

    for c in text:
	if c in ignore:
	    converted += c
	elif mode in (1, 3, 5, 7) and zh_ascii.has_key(c):
	    converted += zh_ascii[c]
	elif mode in (2, 3, 6, 7) and zh_digit.has_key(c):
	    converted += zh_digit[c]
	elif mode in range(4, 8) and zh_kana.has_key(c):
	    converted += zh_kana[c]
	else:
	    converted += c

    return converted

# function convert from HANKAKU to ZENKAKU
# argument and return: unicode string
def h2z(text, mode=ALL, ignore=()):
    converted = ""

    text = check_text(text)
    mode = check_mode(mode)

    i = 0
    size = len(text)

    while i < size:
	if i < size - 1:
	    if text[i:i+2] in ignore:
		converted += text[i:i+2]
		i += 2
		continue
	    elif mode in range(4, 8) and hz_kana.has_key(text[i:i+2]): # DAKUON
		converted += hz_kana[text[i:i+2]]
		i += 2
		continue

	if text[i:i+1] in ignore:
	    converted += text[i:i+1]
	elif mode in (1, 3, 5, 7) and hz_ascii.has_key(text[i:i+1]):
	    converted += hz_ascii[text[i:i+1]]
	elif mode in (2, 3, 6, 7) and hz_digit.has_key(text[i:i+1]):
	    converted += hz_digit[text[i:i+1]]
	elif mode in range(4, 8) and hz_kana.has_key(text[i:i+1]):
	    converted += hz_kana[text[i:i+1]]
	else:
	    converted += text[i:i+1]

	i += 1

    return converted

if __name__ == "__main__":
    teststr = unicode("abcＤＥＦ123４５６ｱｶﾞｻダナバビﾌﾟﾍﾟ", "euc-jp")

    print "original:", teststr.encode("euc-jp")
    print "h2z ascii only:", h2z(teststr, ASCII).encode("euc-jp")
    print "h2z ascii and kana:", h2z(teststr, ASCII|KANA).encode("euc-jp")
    print "z2h digit only:", z2h(teststr, DIGIT).encode("euc-jp")
    print "z2h digit and kana:", z2h(teststr, DIGIT|KANA).encode("euc-jp")
    print "z2h digit and kana, but '５':", z2h(teststr, DIGIT|KANA, (u"５")).encode("euc-jp")