pinyin.py

dict={
'-':0,
'a': 1 ,
'o': 2 ,
'e': 3 ,
'i': 4 ,
'u': 5 ,
'v': 6 ,
'ai': 7 ,
'ei': 8 ,
'ao': 9 ,
'ou': 10 ,
'er': 11 ,
'an': 12 ,
'en': 13 ,
'ang': 14 ,
'ba': 15 ,
'bo': 16 ,
'bi': 17 ,
'bu': 18 ,
'bai': 19 ,
'bei': 20 ,
'bao': 21 ,
'bie': 22 ,
'ban': 23 ,
'ben': 24 ,
'bin': 25 ,
'bang': 26 ,
'beng': 27 ,
'bing': 28 ,
'pa': 29 ,
'po': 30 ,
'pi': 31 ,
'pu': 32 ,
'pai': 33 ,
'pei': 34 ,
'pao': 35 ,
'pou': 36 ,
'pie': 37 ,
'pan': 38 ,
'pen': 39 ,
'pin': 40 ,
'pang': 41 ,
'peng': 42 ,
'ping': 43 ,
'ma': 44 ,
'mo': 45 ,
'me': 46 ,
'mi': 47 ,
'mu': 48 ,
'mai': 49 ,
'mei': 50 ,
'mao': 51 ,
'mou': 52 ,
'miu': 53 ,
'mie': 54 ,
'man': 55 ,
'men': 56 ,
'min': 57 ,
'mang': 58 ,
'meng': 59 ,
'ming': 60 ,
'fa': 61 ,
'fo': 62 ,
'fu': 63 ,
'fei': 64 ,
'fou': 65 ,
'fan': 66 ,
'fen': 67 ,
'fang': 68 ,
'feng': 69 ,
'da': 70 ,
'de': 71 ,
'di': 72 ,
'du': 73 ,
'dai': 74 ,
'dei': 75 ,
'dui': 76 ,
'dao': 77 ,
'dou': 78 ,
'diu': 79 ,
'die': 80 ,
'dan': 81 ,
'dun': 82 ,
'dang': 83 ,
'deng': 84 ,
'ding': 85 ,
'dong': 86 ,
'ta': 87 ,
'te': 88 ,
'ti': 89 ,
'tu': 90 ,
'tai': 91 ,
'tui': 92 ,
'tao': 93 ,
'tou': 94 ,
'tie': 95 ,
'tan': 96 ,
'tun': 97 ,
'tang': 98 ,
'teng': 99 ,
'ting': 100 ,
'tong': 101 ,
'na': 102 ,
'ne': 103 ,
'ni': 104 ,
'nu': 105 ,
'nv': 106 ,
'nai': 107 ,
'nei': 108 ,
'nao': 109 ,
'nou': 110 ,
'niu': 111 ,
'nie': 112 ,
'nve': 113 ,
'nan': 114 ,
'nen': 115 ,
'nin': 116 ,
'nang': 117 ,
'neng': 118 ,
'ning': 119 ,
'nong': 120 ,
'la': 121 ,
'le': 122 ,
'li': 123 ,
'lu': 124 ,
'lv': 125 ,
'lai': 126 ,
'lei': 127 ,
'lao': 128 ,
'lou': 129 ,
'liu': 130 ,
'lie': 131 ,
'lve': 132 ,
'lan': 133 ,
'lin': 134 ,
'lun': 135 ,
'lang': 136 ,
'leng': 137 ,
'ling': 138 ,
'long': 139 ,
'ga': 140 ,
'ge': 141 ,
'gu': 142 ,
'gai': 143 ,
'gei': 144 ,
'gui': 145 ,
'gao': 146 ,
'gou': 147 ,
'gan': 148 ,
'gen': 149 ,
'gun': 150 ,
'gang': 151 ,
'geng': 152 ,
'gong': 153 ,
'ka': 154 ,
'ke': 155 ,
'ku': 156 ,
'kai': 157 ,
'kui': 158 ,
'kao': 159 ,
'kou': 160 ,
'kan': 161 ,
'ken': 162 ,
'kun': 163 ,
'kang': 164 ,
'keng': 165 ,
'kong': 166 ,
'ha': 167 ,
'he': 168 ,
'hu': 169 ,
'hai': 170 ,
'hei': 171 ,
'hui': 172 ,
'hao': 173 ,
'hou': 174 ,
'han': 175 ,
'hen': 176 ,
'hun': 177 ,
'hang': 178 ,
'heng': 179 ,
'hong': 180 ,
'ji': 181 ,
'ju': 182 ,
'jiu': 183 ,
'jie': 184 ,
'jue': 185 ,
'jin': 186 ,
'jun': 187 ,
'jing': 188 ,
'qi': 189 ,
'qu': 190 ,
'qiu': 191 ,
'qie': 192 ,
'que': 193 ,
'qin': 194 ,
'qun': 195 ,
'qing': 196 ,
'xi': 197 ,
'xu': 198 ,
'xiu': 199 ,
'xie': 200 ,
'xue': 201 ,
'xin': 202 ,
'xun': 203 ,
'xing': 204 ,
'za': 205 ,
'ze': 206 ,
'zi': 207 ,
'zu': 208 ,
'zai': 209 ,
'zei': 210 ,
'zui': 211 ,
'zao': 212 ,
'zou': 213 ,
'zan': 214 ,
'zen': 215 ,
'zun': 216 ,
'zang': 217 ,
'zeng': 218 ,
'zong': 219 ,
'ca': 220 ,
'ce': 221 ,
'ci': 222 ,
'cu': 223 ,
'cai': 224 ,
'cui': 225 ,
'cao': 226 ,
'cou': 227 ,
'can': 228 ,
'cen': 229 ,
'cun': 230 ,
'cang': 231 ,
'ceng': 232 ,
'cong': 233 ,
'sa': 234 ,
'se': 235 ,
'si': 236 ,
'su': 237 ,
'sai': 238 ,
'sui': 239 ,
'sao': 240 ,
'sou': 241 ,
'san': 242 ,
'sen': 243 ,
'sun': 244 ,
'sang': 245 ,
'seng': 246 ,
'song': 247 ,
'zha': 248 ,
'zhe': 249 ,
'zhi': 250 ,
'zhu': 251 ,
'zhai': 252 ,
'zhei': 253 ,
'zhui': 254 ,
'zhao': 255 ,
'zhou': 256 ,
'zhan': 257 ,
'zhen': 258 ,
'zhun': 259 ,
'zhang': 260 ,
'zheng': 261 ,
'zhong': 262 ,
'cha': 263 ,
'che': 264 ,
'chi': 265 ,
'chu': 266 ,
'chai': 267 ,
'chui': 268 ,
'chao': 269 ,
'chou': 270 ,
'chan': 271 ,
'chen': 272 ,
'chun': 273 ,
'chang': 274 ,
'cheng': 275 ,
'chong': 276 ,
'sha': 277 ,
'she': 278 ,
'shi': 279 ,
'shu': 280 ,
'shai': 281 ,
'shei': 282 ,
'shui': 283 ,
'shao': 284 ,
'shou': 285 ,
'shan': 286 ,
'shen': 287 ,
'shun': 288 ,
'shang': 289 ,
'sheng': 290 ,
're': 291 ,
'ri': 292 ,
'ru': 293 ,
'rui': 294 ,
'rao': 295 ,
'rou': 296 ,
'ran': 297 ,
'ren': 298 ,
'run': 299 ,
'rang': 300 ,
'reng': 301 ,
'rong': 302 ,
'ya': 303 ,
'ye': 304 ,
'yi': 305 ,
'yu': 306 ,
'yao': 307 ,
'you': 308 ,
'yue': 309 ,
'yan': 310 ,
'yin': 311 ,
'yun': 312 ,
'yang': 313 ,
'ying': 314 ,
'yong': 315 ,
'wa': 316 ,
'wo': 317 ,
'wu': 318 ,
'wai': 319 ,
'wei': 320 ,
'wan': 321 ,
'wen': 322 ,
'wang': 323 ,
'weng': 324 ,
'biao': 325 ,
'bian': 326 ,
'piao': 327 ,
'pian': 328 ,
'miao': 329 ,
'mian': 330 ,
'diao': 331 ,
'dian': 332 ,
'duo': 333 ,
'duan': 334 ,
'tiao': 335 ,
'tian': 336 ,
'tuo': 337 ,
'tuan': 338 ,
'niao': 339 ,
'nian': 340 ,
'niang': 341 ,
'nuo': 342 ,
'nuan': 343 ,
'lia': 344 ,
'liao': 345 ,
'lian': 346 ,
'liang': 347 ,
'luo': 348 ,
'luan': 349 ,
'gua': 350 ,
'guo': 351 ,
'guai': 352 ,
'guan': 353 ,
'guang': 354 ,
'kua': 355 ,
'kuo': 356 ,
'kuai': 357 ,
'kuan': 358 ,
'kuang': 359 ,
'hua': 360 ,
'huo': 361 ,
'huai': 362 ,
'huan': 363 ,
'huang': 364 ,
'jia': 365 ,
'jiao': 366 ,
'jian': 367 ,
'jiang': 368 ,
'jiong': 369 ,
'juan': 370 ,
'qia': 371 ,
'qiao': 372 ,
'qian': 373 ,
'qiang': 374 ,
'qiong': 375 ,
'quan': 376 ,
'xia': 377 ,
'xiao': 378 ,
'xian': 379 ,
'xiang': 380 ,
'xiong': 381 ,
'xuan': 382 ,
'zuo': 383 ,
'zuan': 384 ,
'cuo': 385 ,
'cuan': 386 ,
'suo': 387 ,
'suan': 388 ,
'zhua': 389 ,
'zhuo': 390 ,
'zhuai': 391 ,
'zhuan': 392 ,
'zhuang': 393 ,
'chua': 394 ,
'chuo': 395 ,
'chuai': 396 ,
'chuan': 397 ,
'chuang': 398 ,
'shua': 399 ,
'shuo': 400 ,
'shuai': 401 ,
'shuan': 402 ,
'shuang': 403 ,
'ruo': 404 ,
'ruan': 405 ,
'yuan': 406
}
dictC={
    'b':[1,0],
    'p':[2,0],
    'm':[0,1],
    'f':[5,2],
    'z':[3,3],
    'c':[4,3],
    's':[5,3],
    'd':[1,4],
    't':[2,4],
    'n':[0,5],
    'l':[7,5],
    'zh':[3,6],
    'ch':[4,6],
    'sh':[5,6],
    'r':[5,7],
    'j':[3,8],
    'q':[4,8],
    'x':[5,8],
    'g':[1,9],
    'k':[2,9],
    'h':[5,9],
    'y':[8,10],
    'w':[233,233]
}
dictV={
    'i':[0,1],
    'u':[0,2],
    'v':[0,3],
    'a':[1,0],
    'ia':[1,1],
    'ua':[1,2],
    'o':[2,0],
    'uo':[2,2],
    'e':[3,0],
    'ie':[4,1],
    'ue':[4,3],
    'er':[5,0],
    'ai':[6,0],
    'uai':[6,2],
    'ei':[7,0],
    'uei':[7,2],
    'ao':[8,0],
    'iao':[8,1],
    'ou':[9,0],
    'iou':[9,1],
    'an':[10,0],
    'ian':[10,1],
    'uan':[10,2],
    'van':[10,3],
    'en':[11,0],
    'uen':[11,2],
    'in':[12,1],
    'un':[12,2],
    'vn':[12,3],
    'ang':[13,0],
    'iang':[13,1],
    'uang':[13,2],
    'eng':[14,0],
    'ing':[14,1],
    'ueng':[14,2],
    'ong':[15,2],
    'iong':[15,3]
}

def py2ints(word):
    word=word.strip()
    if word[0] in dictC:
        if word[1]=='h':
            C=word[:2]
            V=word[2:]
        else:
            C=word[:1]
            V=word[1:]
    else:
        C=''
        V=word
    if C in ['j','q','x','y'] and V in ['uan','un','u']:
        V.replace('u','v')
    if C=='w' and V=='u':
        C=''
    if C=='w':
        C=''
        V='u'+V
    if C=='y' and V in ['a','e','ao','ou','an','ang','ong']:
        C=''
        V='i'+V
    if C=='y' and V in ['i','ue','in','v','vn','ing','van']:
        C=''
    V=V.replace('iu','iou')
    V=V.replace('ui','uei')
    if C=='':
        return ([],dictV[V])
    else:
        return (dictC[C],dictV[V])



def py2int(word):
    try:
        return dict[word]
    except KeyError:
        print('KeyError: ',word)