ai-content-maker/.venv/Lib/site-packages/pypinyin/style/bopomofo.py

75 lines
2.3 KiB
Python
Raw Normal View History

2024-05-03 04:18:51 +03:00
# -*- coding: utf-8 -*-
"""BOPOMOFO 相关的几个拼音风格实现:
Style.BOPOMOFO
Style.BOPOMOFO_FIRST
"""
from __future__ import unicode_literals
import re
from pypinyin.constants import Style
from pypinyin.style import register
from pypinyin.style._constants import RE_TONE3
from pypinyin.style._utils import replace_symbol_to_number
# 注音转换表
BOPOMOFO_REPLACE = (
(re.compile(r'^m(\d)$'), 'mu\\1'), # 呣
(re.compile(r'^n(\d)$'), 'N\\1'), # 嗯
(re.compile(r'^r5$'), 'er5'), # 〜兒
(re.compile(r'iu'), 'iou'),
(re.compile(r'ui'), 'uei'),
(re.compile(r'ong'), 'ung'),
(re.compile(r'^yi?'), 'i'),
(re.compile(r'^wu?'), 'u'),
(re.compile(r'iu'), 'v'),
(re.compile(r'^([jqx])u'), '\\1v'),
(re.compile(r'([iuv])n'), '\\1en'),
(re.compile(r'^zhi?'), 'Z'),
(re.compile(r'^chi?'), 'C'),
(re.compile(r'^shi?'), 'S'),
(re.compile(r'^([zcsr])i'), '\\1'),
(re.compile(r'ai'), 'A'),
(re.compile(r'ei'), 'I'),
(re.compile(r'ao'), 'O'),
(re.compile(r'ou'), 'U'),
(re.compile(r'ang'), 'K'),
(re.compile(r'eng'), 'G'),
(re.compile(r'an'), 'M'),
(re.compile(r'en'), 'N'),
(re.compile(r'er'), 'R'),
(re.compile(r'eh'), 'E'),
(re.compile(r'([iv])e'), '\\1E'),
(re.compile(r'([^0-4])$'), '\\g<1>0'),
(re.compile(r'1$'), ''),
)
BOPOMOFO_TABLE = dict(zip(
'bpmfdtnlgkhjqxZCSrzcsiuvaoeEAIOUMNKGR2340ê',
'ㄅㄆㄇㄈㄉㄊㄋㄌㄍㄎㄏㄐㄑㄒㄓㄔㄕㄖㄗㄘㄙㄧㄨㄩㄚㄛㄜㄝㄞㄟㄠㄡㄢㄣㄤㄥㄦˊˇˋ˙ㄝ'
))
class BopomofoConverter(object):
def to_bopomofo(self, pinyin, **kwargs):
pinyin = self._pre_convert(pinyin)
# 查表替换成注音
for find_re, replace in BOPOMOFO_REPLACE:
pinyin = find_re.sub(replace, pinyin)
pinyin = ''.join(BOPOMOFO_TABLE.get(x, x) for x in pinyin)
return pinyin
def to_bopomofo_first(self, pinyin, **kwargs):
pinyin = self.to_bopomofo(pinyin, **kwargs)
return pinyin[0]
def _pre_convert(self, pinyin):
# 用数字表示声调
pinyin = replace_symbol_to_number(pinyin)
# 将声调数字移动到最后
return RE_TONE3.sub(r'\1\3\2', pinyin)
converter = BopomofoConverter()
register(Style.BOPOMOFO, func=converter.to_bopomofo)
register(Style.BOPOMOFO_FIRST, func=converter.to_bopomofo_first)