107 lines
2.6 KiB
Python
107 lines
2.6 KiB
Python
from enum import Enum
|
|
|
|
from pydantic import BaseModel, validator, field_validator, Field
|
|
from typing import Optional, Literal, List
|
|
|
|
|
|
class PosEnumFr(str, Enum):
|
|
# noun
|
|
n = "n."
|
|
n_f = "n.f."
|
|
n_f_pl = "n.f.pl."
|
|
n_m = "n.m."
|
|
n_m_pl = "n.m.pl."
|
|
# verb
|
|
v = "v."
|
|
v_t = "v.t."
|
|
v_i = "v.i."
|
|
v_pr = "v.pr."
|
|
v_t_i = "v.t./v.i."
|
|
v_t_dir = "v.t.dir."
|
|
v_t_ind = "v.t.ind."
|
|
v_t_pr = "v.t.(v.pr.)"
|
|
v_i_ind = "v.t.ind./v.i."
|
|
|
|
adj = "adj." # adj
|
|
adv = "adv." # adv
|
|
prep = "prep." # prep
|
|
pron = "pron." # pron
|
|
conj = "conj."
|
|
interj = "interj."
|
|
chauff = "chauff"
|
|
art = "art."
|
|
|
|
|
|
|
|
class PosEnumJp(str, Enum):
|
|
noun = "名词"
|
|
adj = "形容词"
|
|
adj_v = "形容动词"
|
|
v1 = "一段动词"
|
|
v5 = "五段动词"
|
|
help = "助词"
|
|
|
|
|
|
class CreateWord(BaseModel):
|
|
word: str
|
|
language: Literal["fr", "jp"]
|
|
pos: str = Field(title="词性", description="必须符合对应语言词性枚举")
|
|
meaning: str
|
|
example: Optional[str]
|
|
eng_explanation: Optional[str]
|
|
|
|
class Config:
|
|
orm_mode = True
|
|
title = "接受新词条模型"
|
|
|
|
@classmethod
|
|
@field_validator("eng_explanation")
|
|
def validate_eng_explanation(cls, v):
|
|
if cls.language == "jp" and v:
|
|
raise ValueError("Japanese word has no English explanation")
|
|
if cls.language == "fr" and v is None or v == "":
|
|
raise ValueError("French word must have English explanation")
|
|
return v
|
|
|
|
@classmethod
|
|
@field_validator("pos")
|
|
def validate_pos(cls, v):
|
|
if cls.language == "fr" and v not in PosEnumFr:
|
|
raise ValueError("Pos is not a valid type")
|
|
if cls.language == "jp" and v not in PosEnumJp:
|
|
raise ValueError("Pos is not a valid type")
|
|
return v
|
|
|
|
|
|
class UpdateWord(BaseModel):
|
|
id: int
|
|
word: str
|
|
language: Literal["fr", "jp"]
|
|
eng_explanation: Optional[str]
|
|
example: Optional[str]
|
|
pos: Optional[str]
|
|
meaning: Optional[str]
|
|
|
|
class Config:
|
|
orm_mode = True # 允许从 ORM 实例中提取字段，而不仅限于 dict 类型
|
|
|
|
|
|
class UpdateWordSet(List[UpdateWord]):
|
|
pass
|
|
|
|
|
|
class SearchWordRequest(BaseModel):
|
|
word: str
|
|
language: Literal["fr", "jp"]
|
|
pos: Optional[str]
|
|
|
|
@classmethod
|
|
@field_validator("pos")
|
|
def validate_pos(cls, v):
|
|
if v is not None:
|
|
if cls.language is "fr" and v not in PosEnumFr:
|
|
raise ValueError("Pos is not a valid type")
|
|
if cls.language is "jp" and v not in PosEnumJp:
|
|
raise ValueError("Pos is not a valid type")
|
|
return v
|