228 lines
7.6 KiB
Python
228 lines
7.6 KiB
Python
import re
|
||
from datetime import datetime, date
|
||
from bs4 import BeautifulSoup
|
||
|
||
from bot.entries.entry import Entry
|
||
import bot.entries.expressions as Expressions
|
||
|
||
|
||
class _JitenonEntry(Entry):
|
||
ID_TO_ENTRY = {}
|
||
|
||
def __init__(self, entry_id):
|
||
super().__init__(entry_id)
|
||
if entry_id not in self.ID_TO_ENTRY:
|
||
self.ID_TO_ENTRY[entry_id] = self
|
||
else:
|
||
raise Exception(f"Duplicate entry ID: {entry_id}")
|
||
self.modified_date = date(1970, 1, 1)
|
||
self.attribution = ""
|
||
for column in self._COLUMNS.values():
|
||
setattr(self, column[0], column[1])
|
||
|
||
def set_page(self, page):
|
||
soup = BeautifulSoup(page, features="html5lib")
|
||
self.__set_modified_date(page)
|
||
self.__set_attribution(soup)
|
||
table = soup.find(class_="kanjirighttb")
|
||
if table is None:
|
||
raise ValueError("Error: table data not found in page.")
|
||
rows = table.find("tbody").find_all("tr")
|
||
colname = ""
|
||
for row in rows:
|
||
colname = row.th.text if row.th is not None else colname
|
||
colval = self.__clean_text(row.td.text)
|
||
self.__set_column(colname, colval)
|
||
self._page = table.decode()
|
||
|
||
def get_page_soup(self):
|
||
soup = BeautifulSoup(self._page, "html5lib")
|
||
return soup
|
||
|
||
def get_headwords(self):
|
||
if self._headwords is not None:
|
||
return self._headwords
|
||
self._set_headwords()
|
||
self._set_variant_headwords()
|
||
return self._headwords
|
||
|
||
def get_part_of_speech_tags(self):
|
||
# Jitenon doesn't have any
|
||
return []
|
||
|
||
def _set_headwords(self):
|
||
headwords = {}
|
||
for yomikata in self._yomikatas():
|
||
headwords[yomikata] = [self.expression]
|
||
ikei_headwords = self._ikei_headwords()
|
||
for reading, expressions in ikei_headwords.items():
|
||
if reading not in headwords:
|
||
headwords[reading] = []
|
||
for expression in expressions:
|
||
if expression not in headwords[reading]:
|
||
headwords[reading].append(expression)
|
||
self._headwords = headwords
|
||
|
||
def __set_modified_date(self, page):
|
||
m = re.search(r"\"dateModified\": \"(\d{4}-\d{2}-\d{2})", page)
|
||
if m is None:
|
||
return
|
||
date = datetime.strptime(m.group(1), '%Y-%m-%d').date()
|
||
self.modified_date = date
|
||
|
||
def __set_attribution(self, soup):
|
||
attribution = soup.find(class_="copyright")
|
||
if attribution is not None:
|
||
self.attribution = soup.find(class_="copyright").text
|
||
else:
|
||
self.attribution = ""
|
||
|
||
def __set_column(self, colname, colval):
|
||
attr_name = self._COLUMNS[colname][0]
|
||
attr_value = getattr(self, attr_name)
|
||
if isinstance(attr_value, str):
|
||
setattr(self, attr_name, colval)
|
||
elif isinstance(attr_value, list):
|
||
if len(attr_value) == 0:
|
||
setattr(self, attr_name, [colval])
|
||
else:
|
||
attr_value.append(colval)
|
||
|
||
def _yomikatas(self):
|
||
yomikata = self.yomikata
|
||
m = re.search(r"^[ぁ-ヿ、]+$", yomikata)
|
||
if m:
|
||
return [yomikata]
|
||
m = re.search(r"^([ぁ-ヿ、]+)※", yomikata)
|
||
if m:
|
||
return [m.group(1)]
|
||
m = re.search(r"^[ぁ-ヿ、]+（[ぁ-ヿ、]）[ぁ-ヿ、]+$", yomikata)
|
||
if m:
|
||
return Expressions.expand_abbreviation(yomikata)
|
||
m = re.search(r"^([ぁ-ヿ、]+)（([ぁ-ヿ/\s、]+)）$", yomikata)
|
||
if m:
|
||
yomikatas = [m.group(1)]
|
||
alts = m.group(2).split("/")
|
||
for alt in alts:
|
||
yomikatas.append(alt.strip())
|
||
return yomikatas
|
||
print(f"Invalid 読み方 format: {self.yomikata}\n{self}\n")
|
||
return [""]
|
||
|
||
def _ikei_headwords(self):
|
||
ikei_headwords = {}
|
||
for val in self.ikei:
|
||
m = re.search(r"^([^（]+)（([ぁ-ヿ、]+)）$", val)
|
||
if not m:
|
||
print(f"Invalid 異形 format: {val}\n{self}\n")
|
||
continue
|
||
expression = m.group(1)
|
||
reading = m.group(2)
|
||
if reading not in ikei_headwords:
|
||
ikei_headwords[reading] = []
|
||
if expression not in ikei_headwords[reading]:
|
||
ikei_headwords[reading].append(expression)
|
||
return ikei_headwords
|
||
|
||
@staticmethod
|
||
def __clean_text(text):
|
||
text = text.replace("\n", "")
|
||
text = text.replace(",", "、")
|
||
text = text.replace(" ", "")
|
||
text = text.strip()
|
||
return text
|
||
|
||
def __str__(self):
|
||
colvals = [str(self.entry_id)]
|
||
for attr in self._COLUMNS.values():
|
||
attr_val = getattr(self, attr[0])
|
||
if isinstance(attr_val, str):
|
||
colvals.append(attr_val)
|
||
elif isinstance(attr_val, list):
|
||
colvals.append("；".join(attr_val))
|
||
return ",".join(colvals)
|
||
|
||
|
||
class JitenonYojiEntry(_JitenonEntry):
|
||
_COLUMNS = {
|
||
"四字熟語": ["expression", ""],
|
||
"読み方": ["yomikata", ""],
|
||
"意味": ["imi", ""],
|
||
"出典": ["shutten", ""],
|
||
"漢検級": ["kankenkyuu", ""],
|
||
"場面用途": ["bamenyouto", ""],
|
||
"異形": ["ikei", []],
|
||
"類義語": ["ruigigo", []],
|
||
}
|
||
|
||
def __init__(self, entry_id):
|
||
super().__init__(entry_id)
|
||
|
||
def _set_variant_headwords(self):
|
||
for expressions in self._headwords.values():
|
||
Expressions.add_variant_kanji(expressions)
|
||
|
||
|
||
class JitenonKotowazaEntry(_JitenonEntry):
|
||
_COLUMNS = {
|
||
"言葉": ["expression", ""],
|
||
"読み方": ["yomikata", ""],
|
||
"意味": ["imi", ""],
|
||
"出典": ["shutten", ""],
|
||
"例文": ["reibun", ""],
|
||
"異形": ["ikei", []],
|
||
"類句": ["ruiku", []],
|
||
}
|
||
|
||
def __init__(self, entry_id):
|
||
super().__init__(entry_id)
|
||
|
||
def _set_headwords(self):
|
||
if self.expression == "金棒引き・鉄棒引き":
|
||
self._headwords = {
|
||
"かなぼうひき": ["金棒引き", "鉄棒引き"]
|
||
}
|
||
else:
|
||
super()._set_headwords()
|
||
|
||
def _set_variant_headwords(self):
|
||
for expressions in self._headwords.values():
|
||
Expressions.add_variant_kanji(expressions)
|
||
Expressions.add_fullwidth(expressions)
|
||
|
||
|
||
class JitenonKokugoEntry(_JitenonEntry):
|
||
_COLUMNS = {
|
||
"言葉": ["expression", ""],
|
||
"読み方": ["yomikata", ""],
|
||
"意味": ["imi", ""],
|
||
"例文": ["reibun", ""],
|
||
"別表記": ["betsuhyouki", ""],
|
||
"対義語": ["taigigo", ""],
|
||
"活用": ["katsuyou", ""],
|
||
"用例": ["yourei", ""],
|
||
"類語": ["ruigo", ""],
|
||
}
|
||
|
||
def __init__(self, entry_id):
|
||
super().__init__(entry_id)
|
||
|
||
def _set_headwords(self):
|
||
headwords = {}
|
||
for reading in self.yomikata.split("・"):
|
||
if reading not in headwords:
|
||
headwords[reading] = []
|
||
for expression in self.expression.split("・"):
|
||
headwords[reading].append(expression)
|
||
if self.betsuhyouki.strip() != "":
|
||
for expression in self.betsuhyouki.split("・"):
|
||
headwords[reading].append(expression)
|
||
self._headwords = headwords
|
||
|
||
def _set_variant_headwords(self):
|
||
for expressions in self._headwords.values():
|
||
Expressions.add_variant_kanji(expressions)
|
||
Expressions.add_fullwidth(expressions)
|
||
Expressions.remove_iteration_mark(expressions)
|
||
Expressions.add_iteration_mark(expressions)
|