mirror of
https://github.com/TriM-Organization/LiteyukiBot-TriM.git
synced 2025-09-07 20:56:23 +00:00
142 lines
4.5 KiB
Python
142 lines
4.5 KiB
Python
import random
|
|
|
|
from pathlib import Path
|
|
|
|
|
|
from typing import Dict, List, Tuple
|
|
from PIL import ImageFont
|
|
|
|
from PIL.ImageFont import FreeTypeFont
|
|
|
|
# from watchdog.observers import Observer
|
|
# from watchdog.events import FileSystemEventHandler, FileModifiedEvent
|
|
|
|
from pypinyin import Style, pinyin
|
|
|
|
resource_dir = Path(__file__).parent / "resources"
|
|
fonts_dir = resource_dir / "fonts"
|
|
data_dir = resource_dir / "data"
|
|
idiom_path = resource_dir / "idioms_p.txt"
|
|
|
|
|
|
|
|
|
|
# fmt: off
|
|
# 声母
|
|
INITIALS = [
|
|
"zh", "z", "y", "x", "w", "t", "sh", "s", "r", "q", "p",
|
|
"n", "m", "l", "k", "j", "h", "g", "f", "d", "ch", "c", "b"
|
|
]
|
|
# 韵母
|
|
FINALS = [
|
|
"ün", "üe", "üan", "ü", "uo", "un", "ui", "ue", "uang",
|
|
"uan", "uai","ua", "ou", "iu", "iong", "ong", "io", "ing",
|
|
"in", "ie", "iao", "iang", "ian", "ia", "er", "eng", "en",
|
|
"ei", "ao", "ang", "an", "ai", "u", "o", "i", "e", "a"
|
|
]
|
|
# fmt: on
|
|
|
|
|
|
def get_pinyin_of_n(word: str, which: int) -> List[Tuple[str, str, str]]:
|
|
pys = pinyin(word, style=Style.TONE3, v_to_u=True,heteronym=True)[which]
|
|
# py = p[0]
|
|
results = []
|
|
for py in pys:
|
|
if py[-1].isdigit():
|
|
tone = py[-1]
|
|
py = py[:-1]
|
|
else:
|
|
tone = ""
|
|
initial = ""
|
|
for i in INITIALS:
|
|
if py.startswith(i):
|
|
initial = i
|
|
break
|
|
final = ""
|
|
for f in FINALS:
|
|
if py.endswith(f):
|
|
final = f
|
|
break
|
|
results.append((initial, final, tone)) # 声母，韵母，声调
|
|
return results
|
|
|
|
|
|
|
|
|
|
|
|
|
|
LEGAL_PHRASES = [
|
|
idiom.strip() for idiom in idiom_path.open("r", encoding="utf-8").readlines()
|
|
]
|
|
sorted_phrases = dict([i for j in [[(py[0]+py[1],{"":[],"1":[],"2":[],"3":[],"4":[]}) for py in get_pinyin_of_n(idiom[0],0)] for idiom in LEGAL_PHRASES] for i in j])
|
|
|
|
for idiom in LEGAL_PHRASES:
|
|
for py in get_pinyin_of_n(idiom[0],0):
|
|
sorted_phrases[py[0]+py[1]][py[2]].append(idiom)
|
|
|
|
|
|
# class LegalPhrasesModifiedHandler(FileSystemEventHandler):
|
|
# """
|
|
# Handler for resource file changes
|
|
# """
|
|
|
|
# def on_modified(self, event):
|
|
# print(f"{event.src_path} modified, reloading resource...")
|
|
# if "idioms.txt" in event.src_path:
|
|
# global LEGAL_PHRASES
|
|
# LEGAL_PHRASES = [
|
|
# idiom.strip()
|
|
# for idiom in idiom_path.open("r", encoding="utf-8").readlines()
|
|
# ]
|
|
# sorted_phrases = dict([i for j in [[(py[0]+py[1],{"":[],"1":[],"2":[],"3":[],"4":[]}) for py in get_pinyin_of_n(idiom[0],0)] for idiom in LEGAL_PHRASES] for i in j])
|
|
|
|
# for idiom in LEGAL_PHRASES:
|
|
# for py in get_pinyin_of_n(idiom[0],0):
|
|
# sorted_phrases[py[0]+py[1]][py[2]].append(idiom)
|
|
|
|
|
|
# Observer().schedule(
|
|
# LegalPhrasesModifiedHandler(),
|
|
# data_dir,
|
|
# recursive=False,
|
|
# event_filter=FileModifiedEvent,
|
|
# )
|
|
|
|
|
|
def legal_idiom(word: str) -> bool:
|
|
return word in LEGAL_PHRASES
|
|
|
|
|
|
def random_idiom() -> str:
|
|
return random.choice(LEGAL_PHRASES)
|
|
|
|
def legal_patted_idiom(former:str, laster: str, diff_word: bool,homophonic: bool) -> bool:
|
|
"""
|
|
判断成语是否符合接龙条件
|
|
|
|
Parameters
|
|
==========
|
|
former: str
|
|
前一个成语
|
|
laster: str
|
|
后一个成语
|
|
diff_word: bool
|
|
异字模式：接龙之字无须一致
|
|
homophonic: bool
|
|
谐音模式：接龙之字可不同音调
|
|
|
|
"""
|
|
return legal_idiom(laster) and legal_idiom(former) and ((((len({i[:2] for i in get_pinyin_of_n(laster[0],0)}.intersection({i[:2] for i in get_pinyin_of_n(former[-1],0)})))>0) if homophonic else (get_pinyin_of_n(laster,0)[0] == get_pinyin_of_n(former,-1)[0])) if diff_word else (former[-1] == laster[0] if homophonic else ((former[-1] == laster[0])and(get_pinyin_of_n(laster,0)[0] == get_pinyin_of_n(former,-1)[0]))))
|
|
|
|
|
|
|
|
|
|
|
|
def get_idiom(idiom: str,diff_word: bool,homophonic: bool) -> str:
|
|
return random.choice(([k for o in [[i for j in sorted_phrases[py[0]+py[1]].values() for i in j] for py in get_pinyin_of_n(idiom[-1],0)] for k in o] if homophonic else sorted_phrases[(py:=get_pinyin_of_n(idiom,-1)[0])[0]+py[1]][py[2]])if diff_word else ([k for o in [[i for j in sorted_phrases[py[0]+py[1]].values() for i in j if i[0] == idiom[-1]] for py in get_pinyin_of_n(idiom[-1],0)] for k in o] if homophonic else (lambda py:[i for i in sorted_phrases[py[0]+py[1]][py[2]] if i[0] == idiom[-1]])(get_pinyin_of_n(idiom,-1)[0])))
|
|
|
|
|
|
|
|
def load_font(name: str, fontsize: int) -> FreeTypeFont:
|
|
return ImageFont.truetype(str(fonts_dir / name), fontsize, encoding="utf-8")
|