146 lines
8.7 KiB
Python
Executable File
146 lines
8.7 KiB
Python
Executable File
import regex
|
||
from etpgrf.config import DEFAULT_MODE, DEFAULT_LANGS, SHY_ENTITIES, MODE_UNICODE
|
||
from etpgrf.comutil import parce_and_validate_mode, parse_and_validate_langs
|
||
|
||
_RU_VOWELS_UPPER = frozenset(['А', 'О', 'И', 'Е', 'Ё', 'Э', 'Ы', 'У', 'Ю', 'Я'])
|
||
_RU_CONSONANTS_UPPER = frozenset(['Б', 'В', 'Г', 'Д', 'Ж', 'З', 'К', 'Л', 'М', 'Н', 'П', 'Р', 'С', 'Т', 'Ф', 'Х', 'Ц', 'Ч', 'Ш', 'Щ'])
|
||
_RU_J_SOUND_UPPER = frozenset(['Й'])
|
||
_RU_SIGNS_UPPER = frozenset(['Ь', 'Ъ'])
|
||
|
||
_EN_VOWELS_UPPER = frozenset(['A', 'E', 'I', 'O', 'U'])
|
||
_EN_CONSONANTS_UPPER = frozenset(['B', 'C', 'D', 'F', 'G', 'H', 'J', 'K', 'L', 'M', 'N', 'P', 'Q', 'R', 'S', 'T', 'V', 'W', 'X', 'Y', 'Z'])
|
||
|
||
|
||
class Hyphenator:
|
||
"""Правила расстановки переносов для разных языков.
|
||
"""
|
||
def __init__(self,
|
||
langs: frozenset[str] = DEFAULT_LANGS, # Языки, которые обрабатываем в переносе слов
|
||
mode: str = DEFAULT_MODE, # Режим обработки текста
|
||
max_unhyphenated_len: int = 14, # Максимальная длина непереносимой группы
|
||
min_chars_per_part: int = 3): # Минимальная длина после переноса (хвост, который разрешено переносить)
|
||
self.langs: frozenset[str] = parse_and_validate_langs(langs)
|
||
self.mode: str = parce_and_validate_mode(mode)
|
||
self.max_unhyphenated_len = max_unhyphenated_len
|
||
self.min_chars_per_part = min_chars_per_part
|
||
|
||
# Внутренние языковые ресурсы, если нужны специфично для переносов
|
||
self._vowels: frozenset = frozenset()
|
||
self._consonants: frozenset = frozenset()
|
||
self._j_sound_upper: frozenset = frozenset()
|
||
self._signs_upper: frozenset = frozenset()
|
||
# Загружает наборы символов на основе self.langs
|
||
self._load_language_resources_for_hyphenation()
|
||
# Определяем символ переноса в зависимости от режима
|
||
self._split_code: str = SHY_ENTITIES['SHY'][0] if self.mode == MODE_UNICODE else SHY_ENTITIES['SHY'][1]
|
||
|
||
|
||
def _load_language_resources_for_hyphenation(self):
|
||
# Определяем наборы гласных, согласных и т.д. в зависимости языков.
|
||
if "ru" in self.langs:
|
||
self._vowels |= _RU_VOWELS_UPPER
|
||
self._consonants |= _RU_CONSONANTS_UPPER
|
||
self._j_sound_upper |= _RU_J_SOUND_UPPER
|
||
self._signs_upper |= _RU_SIGNS_UPPER
|
||
if "en" in self.langs:
|
||
self._vowels |= _EN_VOWELS_UPPER
|
||
self._consonants |= _EN_CONSONANTS_UPPER
|
||
# ... и для других языков, если они поддерживаются переносами
|
||
|
||
# --- Сюда переносятся все методы, связанные с переносами ---
|
||
# (адаптированные версии _is_vow, _is_cons, _is_j_sound, _is_sign,
|
||
# _hyphenate_one_word, _recursive_split_word, _find_hyphen_point_in_sub_word, _is_valid_split_point)
|
||
# Они будут использовать self._vowels, self.hyphen_char и т.д.
|
||
|
||
# Проверка гласных букв
|
||
def _is_vow(self, char: str) -> bool:
|
||
return char.upper() in self._vowels
|
||
|
||
|
||
# Проверка согласных букв
|
||
def _is_cons(self, char: str) -> bool:
|
||
return char.upper() in self._consonants
|
||
|
||
|
||
# Проверка полугласной буквы "й"
|
||
def _is_j_sound(self, char: str) -> bool:
|
||
return char.upper() in self._j_sound_upper
|
||
|
||
|
||
# Проверка мягкого/твердого знака
|
||
def _is_sign(self, char: str) -> bool:
|
||
return char.upper() in self._signs_upper
|
||
|
||
|
||
def hyp_in_word(self, word: str) -> str:
|
||
""" Расстановка переносов в русском слове с учетом максимальной длины непереносимой группы.
|
||
Переносы ставятся половинным делением слова, рекурсивно.
|
||
|
||
:param word: Слово, в котором надо расставить переносы
|
||
:return: Слово с расставленными переносами
|
||
"""
|
||
|
||
# Поиск допустимой позиции для переноса около заданного индекса
|
||
def find_hyphen_point(word_segment: str, start_idx: int) -> int:
|
||
vow_indices = [i for i, char_w in enumerate(word_segment) if self._is_vow(char_w)]
|
||
if not vow_indices:
|
||
# Если в слове нет гласных, то перенос невозможен
|
||
return -1
|
||
# Ищем ближайшую гласную до или после start_idx
|
||
for i in vow_indices:
|
||
if i >= start_idx - self.min_chars_per_part and i + self.min_chars_per_part < len(word_segment):
|
||
# Проверяем, что после гласной есть минимум символов "хвоста"
|
||
ind = i + 1
|
||
if (self._is_cons(word_segment[ind]) or self._is_j_sound(word_segment[ind])) and not self._is_vow(word_segment[ind + 1]):
|
||
# Й -- полугласная. Перенос после неё только в случае, если дальше идет согласная
|
||
# (например, "бой-кий"), но запретить, если идет гласная (например, "ма-йка" не переносится).
|
||
ind += 1
|
||
if ind <= self.min_chars_per_part or ind >= len(word_segment) - self.min_chars_per_part:
|
||
# Не отделяем 3 символ с начала или конца (это некрасиво)
|
||
continue
|
||
if self._is_sign(word_segment[ind]) or self._is_sign(word_segment[-1]):
|
||
# Пропускаем мягкий/твердый знак. Согласно правилам русской типографики (например, ГОСТ 7.62-2008
|
||
# или рекомендации по набору текста), перенос не должен разрывать слово так, чтобы мягкий или
|
||
# твердый знак оказывался в начале или конце строки
|
||
continue
|
||
return ind
|
||
return -1 # Не нашли подходящую позицию
|
||
|
||
# Рекурсивное деление слова
|
||
def split_word(word_to_split: str) -> str:
|
||
if len(word_to_split) <= self.max_unhyphenated_len: # Если длина укладывается в лимит, перенос не нужен
|
||
return word_to_split
|
||
|
||
hyphen_idx = find_hyphen_point(word_to_split, len(word_to_split) // 2) # Ищем точку переноса около середины
|
||
|
||
if hyphen_idx == -1: # Если не нашли точку переноса
|
||
return word_to_split
|
||
|
||
left_part = word_to_split[:hyphen_idx]
|
||
right_part = word_to_split[hyphen_idx:]
|
||
|
||
# Рекурсивно делим левую и правую части и соединяем их через символ переноса
|
||
return split_word(left_part) + self._split_code + split_word(right_part)
|
||
|
||
# Основная логика
|
||
if len(word) <= self.max_unhyphenated_len or not any(self._is_vow(c) for c in word):
|
||
# Короткое слово или без гласных "делению не подлежит", выходим из рекурсии
|
||
return word
|
||
return split_word(word) # Рекурсивно делим слово на части с переносами
|
||
|
||
|
||
def hyp_in_text(self, text: str) -> str:
|
||
""" Расстановка переносов в тексте
|
||
|
||
:param text: Строка, которую надо обработать (главный аргумент).
|
||
:return: str:
|
||
"""
|
||
rus_worlds = regex.findall(r'\b[а-яА-Я]+\b', text) # ищем все русскоязычные слова в тексте
|
||
for word in rus_worlds:
|
||
if len(word) > self.max_unhyphenated_len:
|
||
hyphenated_word = self.hyp_in_word(word)
|
||
print(f'{word} -> {hyphenated_word}')
|
||
text = text.replace(word, hyphenated_word)
|
||
return text
|
||
|