"""其他几个拼音风格实现:
Style.NORMAL
Style.FIRST_LETTER
"""
from typing import Text
from typing import Any

from pypinyin.constants import Style
from pypinyin.style import register
from pypinyin.style._utils import replace_symbol_to_no_symbol


class OthersConverter():
    def to_normal(self, pinyin: Text, **kwargs: Any) -> Text:
        pinyin = replace_symbol_to_no_symbol(pinyin)
        return pinyin

    def to_first_letter(self, pinyin: Text, **kwargs: Any) -> Text:
        # 用数字表示声调
        pinyin = self.to_normal(pinyin)
        return pinyin[0]


converter = OthersConverter()
register(Style.NORMAL, func=converter.to_normal)
register(Style.FIRST_LETTER, func=converter.to_first_letter)