From e85537450865007a46fb221e9cdd34bf624405a1 Mon Sep 17 00:00:00 2001 From: Shurelol Date: Sat, 8 Jul 2023 23:10:13 +0800 Subject: [PATCH] fix dataclass init --- .gitignore | 4 +++- app/core/context.py | 19 +++++-------------- 2 files changed, 8 insertions(+), 15 deletions(-) diff --git a/.gitignore b/.gitignore index e9a16e1b..7c6cc139 100644 --- a/.gitignore +++ b/.gitignore @@ -3,4 +3,6 @@ build/ test.py app/helper/sites.py -config/user.db \ No newline at end of file +config/user.db +*.pyc +*.log diff --git a/app/core/context.py b/app/core/context.py index a68c3b2d..b5f46706 100644 --- a/app/core/context.py +++ b/app/core/context.py @@ -147,21 +147,12 @@ class MediaInfo: # 演员 actors: List[dict] = field(default_factory=dict) - def __init__(self, tmdb_info: dict = None, douban_info: dict = None): - # 初始化 - self.seasons = {} - self.season_info = [] - self.season_years = {} - self.names = [] - self.directors = [] - self.actors = [] - self.tmdb_info = {} - self.douban_info = {} + def __post_init__(self): # 设置媒体信息 - if tmdb_info: - self.set_tmdb_info(tmdb_info) - if douban_info: - self.set_douban_info(douban_info) + if self.tmdb_info: + self.set_tmdb_info(self.tmdb_info) + if self.douban_info: + self.set_douban_info(self.douban_info) def from_dict(self, data: dict): """