constants.py 4.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125
  1. #!/usr/bin/env python
  2. # -*- coding: utf-8 -*-
  3. from __future__ import unicode_literals
  4. import os
  5. import re
  6. from enum import IntEnum, unique
  7. from pypinyin import pinyin_dict
  8. from pypinyin.compat import SUPPORT_UCS4
  9. # 词语拼音库
  10. if os.environ.get('PYPINYIN_NO_PHRASES'):
  11. PHRASES_DICT = {}
  12. else:
  13. from pypinyin import phrases_dict
  14. PHRASES_DICT = phrases_dict.phrases_dict
  15. # 单字拼音库
  16. PINYIN_DICT = pinyin_dict.pinyin_dict
  17. # 利用环境变量控制不做copy操作(无自定义拼音库的情况), 以减少内存使用
  18. if not os.environ.get('PYPINYIN_NO_DICT_COPY'):
  19. PINYIN_DICT = PINYIN_DICT.copy()
  20. PHRASES_DICT = PHRASES_DICT.copy()
  21. # 匹配使用数字标识声调的字符的正则表达式
  22. RE_TONE2 = re.compile(r'([aeoiuvnm])([1-4])$')
  23. # 有拼音的汉字
  24. if SUPPORT_UCS4:
  25. RE_HANS = re.compile(
  26. r'^(?:['
  27. r'\u3007' # 〇
  28. r'\ue815-\ue864'
  29. r'\ufa18'
  30. r'\u3400-\u4dbf' # CJK扩展A:[3400-4DBF]
  31. r'\u4e00-\u9fff' # CJK基本:[4E00-9FFF]
  32. r'\uf900-\ufaff' # CJK兼容:[F900-FAFF]
  33. r'\U00020000-\U0002A6DF' # CJK扩展B:[20000-2A6DF]
  34. r'\U0002A703-\U0002B73F' # CJK扩展C:[2A700-2B73F]
  35. r'\U0002B740-\U0002B81D' # CJK扩展D:[2B740-2B81D]
  36. r'\U0002B825-\U0002BF6E' # CJK扩展E:
  37. r'\U0002C029-\U0002CE93' # CJK扩展F:
  38. r'\U0002D016'
  39. r'\U0002D11B-\U0002EBD9'
  40. r'\U0002F80A-\U0002FA1F' # CJK兼容扩展:[2F800-2FA1F]
  41. r'\U00030000-\U0003134A' # CJK扩展G:
  42. r'\U000300F7-\U00031288'
  43. r'\U00030EDD'
  44. r'\U00030EDE'
  45. r'\U00031350-\U00032389'
  46. r'])+$'
  47. )
  48. else:
  49. RE_HANS = re.compile(
  50. r'^(?:['
  51. r'\u3007' # 〇
  52. r'\ue815-\ue864'
  53. r'\ufa18'
  54. r'\u3400-\u4dbf' # CJK扩展A:[3400-4DBF]
  55. r'\u4e00-\u9fff' # CJK基本:[4E00-9FFF]
  56. r'\uf900-\ufaff' # CJK兼容:[F900-FAFF]
  57. r'])+$'
  58. )
  59. @unique
  60. class Style(IntEnum):
  61. """拼音风格"""
  62. #: 普通风格,不带声调。如: 中国 -> ``zhong guo``
  63. NORMAL = 0
  64. #: 标准声调风格,拼音声调在韵母第一个字母上(默认风格)。如: 中国 -> ``zhōng guó``
  65. TONE = 1
  66. #: 声调风格2,即拼音声调在各个韵母之后,用数字 [1-4] 进行表示。如: 中国 -> ``zho1ng guo2``
  67. TONE2 = 2
  68. #: 声调风格3,即拼音声调在各个拼音之后,用数字 [1-4] 进行表示。如: 中国 -> ``zhong1 guo2``
  69. TONE3 = 8
  70. #: 声母风格,只返回各个拼音的声母部分(注:有的拼音没有声母,详见 `#27`_)。如: 中国 -> ``zh g``
  71. INITIALS = 3
  72. #: 首字母风格,只返回拼音的首字母部分。如: 中国 -> ``z g``
  73. FIRST_LETTER = 4
  74. #: 韵母风格,只返回各个拼音的韵母部分,不带声调。如: 中国 -> ``ong uo``
  75. FINALS = 5
  76. #: 标准韵母风格,带声调,声调在韵母第一个字母上。如:中国 -> ``ōng uó``
  77. FINALS_TONE = 6
  78. #: 韵母风格2,带声调,声调在各个韵母之后,用数字 [1-4] 进行表示。如: 中国 -> ``o1ng uo2``
  79. FINALS_TONE2 = 7
  80. #: 韵母风格3,带声调,声调在各个拼音之后,用数字 [1-4] 进行表示。如: 中国 -> ``ong1 uo2``
  81. FINALS_TONE3 = 9
  82. #: 注音风格,带声调,阴平(第一声)不标。如: 中国 -> ``ㄓㄨㄥ ㄍㄨㄛˊ``
  83. BOPOMOFO = 10
  84. #: 注音风格,仅首字母。如: 中国 -> ``ㄓ ㄍ``
  85. BOPOMOFO_FIRST = 11
  86. #: 汉语拼音与俄语字母对照风格,声调在各个拼音之后,用数字 [1-4] 进行表示。如: 中国 -> ``чжун1 го2``
  87. CYRILLIC = 12
  88. #: 汉语拼音与俄语字母对照风格,仅首字母。如: 中国 -> ``ч г``
  89. CYRILLIC_FIRST = 13
  90. #: 威妥玛拼音/韦氏拼音/威式拼音风格,无声调
  91. WADEGILES = 14
  92. #: 国语罗马字风格。如:中国 -> ``jong gwo``
  93. GWOYEU = 15
  94. #: 中国内地盲文风格,无声调。如:中国 -> ``⠌⠲ ⠛⠕``
  95. BRAILLE_MAINLAND = 16
  96. #: 中国内地盲文风格,带声调。如:中国 -> ``⠌⠲⠁ ⠛⠕⠂``
  97. BRAILLE_MAINLAND_TONE = 17
  98. NORMAL = STYLE_NORMAL = Style.NORMAL
  99. TONE = STYLE_TONE = Style.TONE
  100. TONE2 = STYLE_TONE2 = Style.TONE2
  101. TONE3 = STYLE_TONE3 = Style.TONE3
  102. INITIALS = STYLE_INITIALS = Style.INITIALS
  103. FIRST_LETTER = STYLE_FIRST_LETTER = Style.FIRST_LETTER
  104. FINALS = STYLE_FINALS = Style.FINALS
  105. FINALS_TONE = STYLE_FINALS_TONE = Style.FINALS_TONE
  106. FINALS_TONE2 = STYLE_FINALS_TONE2 = Style.FINALS_TONE2
  107. FINALS_TONE3 = STYLE_FINALS_TONE3 = Style.FINALS_TONE3
  108. BOPOMOFO = STYLE_BOPOMOFO = Style.BOPOMOFO
  109. BOPOMOFO_FIRST = STYLE_BOPOMOFO_FIRST = Style.BOPOMOFO_FIRST
  110. CYRILLIC = STYLE_CYRILLIC = Style.CYRILLIC
  111. CYRILLIC_FIRST = STYLE_CYRILLIC_FIRST = Style.CYRILLIC_FIRST