pypinyin.style 源代码

# -*- coding: utf-8 -*-
from functools import wraps
import glob
import os

current_dir = os.path.dirname(os.path.realpath(__file__))
# 存储各拼音风格对应的实现
_registry = {}


def convert(pinyin, style, strict, default=None, **kwargs):
    """根据拼音风格把原始拼音转换为不同的格式

    :param pinyin: 原始有声调的单个拼音
    :type pinyin: unicode
    :param style: 拼音风格
    :param strict: 是否严格遵照《汉语拼音方案》来处理声母和韵母
    :type strict: bool
    :param default: 拼音风格对应的实现不存在时返回的默认值
    :return: 按照拼音风格进行处理过后的拼音字符串
    :rtype: unicode
    """
    if style in _registry:
        return _registry[style](pinyin, strict=strict, **kwargs)
    return default


def register(style, func=None):
    """注册一个拼音风格实现

    ::

        @register('echo')
        def echo(pinyin, **kwargs):
            return pinyin

        # or
        register('echo', echo)
    """
    if func is not None:
        _registry[style] = func
        return

    def decorator(func):
        _registry[style] = func

        @wraps(func)
        def wrapper(pinyin, **kwargs):
            return func(pinyin, **kwargs)

        return wrapper
    return decorator


def auto_discover():
    """自动发现内置的拼音风格实现"""
    for path in glob.glob(current_dir + os.path.sep + '*.py'):
        filename = os.path.basename(path)
        module_name = filename.split('.')[0]
        if (not module_name) or module_name.startswith('_'):
            continue

        full_module_name = 'pypinyin.style.{0}'.format(module_name)
        __import__(full_module_name)