fix schemas

This commit is contained in:
jxxghp 2023-07-07 17:48:47 +08:00
parent 9ccb2b482e
commit de93b18673
3 changed files with 19 additions and 7 deletions

View File

@ -502,4 +502,8 @@ class Context:
"""
转换为字典
"""
return asdict(self)
return {
"meta_info": self.meta_info.to_dict() if self.meta_info else None,
"torrent_info": self.torrent_info.to_dict() if self.torrent_info else None,
"media_info": self.media_info.to_dict() if self.media_info else None
}

View File

@ -1,4 +1,4 @@
from dataclasses import dataclass
from dataclasses import dataclass, asdict
from typing import Union, Optional, List
import cn2an
@ -439,3 +439,11 @@ class MetaBase(object):
elif str(ep).isdigit():
self.begin_episode = int(ep)
self.end_episode = None
def to_dict(self):
"""
转为字典
"""
dicts = asdict(self)
dicts["type"] = self.type.value if self.type else None
return dicts

View File

@ -11,7 +11,7 @@ class MetaInfo(BaseModel):
识别元数据
"""
# 是否处理的文件
isfile: bool = False
isfile: Optional[bool] = False
# 原字符串
org_string: Optional[str] = None
# 副标题
@ -25,13 +25,13 @@ class MetaInfo(BaseModel):
# 年份
year: Optional[str] = None
# 总季数
total_seasons: int = 0
total_seasons: Optional[int] = 0
# 识别的开始季 数字
begin_season: Optional[int] = None
# 识别的结束季 数字
end_season: Optional[int] = None
# 总集数
total_episodes: int = 0
total_episodes: Optional[int] = 0
# 识别的开始集
begin_episode: Optional[int] = None
# 识别的结束集
@ -89,9 +89,9 @@ class MediaInfo(BaseModel):
# 二级分类
category: Optional[str] = ""
# 季季集清单
seasons: Dict[int, list] = {}
seasons: Optional[Dict[int, list]] = {}
# 季详情
season_info: List[dict] = []
season_info: Optional[List[dict]] = []
# 别名和译名
names: Optional[list] = []
# 演员