add tmdb apis

app/api/endpoints/douban.py

@@ -30,7 +30,7 @@ async def sync_douban(
     return schemas.Response(success=True, message="任务已启动")
 
 
-@router.get("/id", summary="豆瓣ID识别", response_model=schemas.Context)
+@router.get("/recognize/{doubanid}", summary="豆瓣ID识别", response_model=schemas.Context)
 async def recognize_doubanid(doubanid: str,
                              _: schemas.TokenPayload = Depends(verify_token)) -> Any:
     """
@@ -41,7 +41,7 @@ async def recognize_doubanid(doubanid: str,
     return context.to_dict()
 
 
-@router.get("/info", summary="查询豆瓣详情", response_model=schemas.MediaInfo)
+@router.get("/{doubanid}", summary="查询豆瓣详情", response_model=schemas.MediaInfo)
 async def douban_info(doubanid: str, _: schemas.TokenPayload = Depends(verify_token)) -> Any:
     """
     根据豆瓣ID查询豆瓣媒体信息

app/api/endpoints/tmdb.py

@@ -11,8 +11,34 @@ from app.schemas.types import MediaType
 router = APIRouter()
 
 
-@router.get("/info", summary="TMDB详情", response_model=schemas.MediaInfo)
+@router.get("/{tmdbid}/seasons", summary="TMDB所有季", response_model=List[schemas.TmdbSeason])
-async def tmdb_info(tmdbid: int, type_name: str) -> Any:
+async def tmdb_seasons(tmdbid: int, _: schemas.TokenPayload = Depends(verify_token)) -> Any:
+    """
+    根据TMDBID查询themoviedb所有季信息，type_name: 电影/电视剧
+    """
+    seasons_info = TmdbChain().tmdb_seasons(tmdbid=tmdbid)
+    if not seasons_info:
+        return []
+    else:
+        return seasons_info
+
+
+@router.get("/{tmdbid}/{season}", summary="TMDB季所有集", response_model=List[schemas.TmdbEpisode])
+async def tmdb_season_episodes(tmdbid: int, season: int,
+                               _: schemas.TokenPayload = Depends(verify_token)) -> Any:
+    """
+    根据TMDBID查询某季的所有信信息
+    """
+    episodes_info = TmdbChain().tmdb_episodes(tmdbid=tmdbid, season=season)
+    if not episodes_info:
+        return []
+    else:
+        return episodes_info
+
+
+@router.get("/{tmdbid}", summary="TMDB详情", response_model=schemas.MediaInfo)
+async def tmdb_info(tmdbid: int, type_name: str,
+                    _: schemas.TokenPayload = Depends(verify_token)) -> Any:
     """
     根据TMDBID查询themoviedb媒体信息，type_name: 电影/电视剧
     """

app/chain/tmdb.py

@@ -1,5 +1,6 @@
 from typing import Optional, List
 
+from app import schemas
 from app.chain import ChainBase
 from app.schemas import MediaType
 
@@ -31,3 +32,18 @@ class TmdbChain(ChainBase):
         :return: TMDB信息列表
         """
         return self.run_module("tmdb_trending", page=page)
+
+    def tmdb_seasons(self, tmdbid: int) -> List[schemas.TmdbSeason]:
+        """
+        根据TMDBID查询themoviedb所有季信息
+        :param tmdbid:  TMDBID
+        """
+        return self.run_module("tmdb_seasons", tmdbid=tmdbid)
+
+    def tmdb_episodes(self, tmdbid: int, season: int) -> List[schemas.TmdbEpisode]:
+        """
+        根据TMDBID查询某季的所有信信息
+        :param tmdbid:  TMDBID
+        :param season:  季
+        """
+        return self.run_module("tmdb_episodes", tmdbid=tmdbid, season=season)

app/modules/douban/__init__.py

@@ -1,27 +1,26 @@
-import time
 from pathlib import Path
 from typing import List, Optional, Tuple, Union
-from xml.dom import minidom
 
-from app.core.context import MediaInfo
 from app.core.config import settings
-from app.core.metainfo import MetaInfo
+from app.core.context import MediaInfo
 from app.core.meta import MetaBase
+from app.core.metainfo import MetaInfo
 from app.log import logger
 from app.modules import _ModuleBase
 from app.modules.douban.apiv2 import DoubanApi
-from app.utils.dom import DomUtils
+from app.modules.douban.scraper import DoubanScraper
-from app.utils.http import RequestUtils
-from app.utils.system import SystemUtils
 from app.schemas.types import MediaType
+from app.utils.system import SystemUtils
 
 
 class DoubanModule(_ModuleBase):
 
     doubanapi: DoubanApi = None
+    scraper: DoubanScraper = None
 
     def init_module(self) -> None:
         self.doubanapi = DoubanApi()
+        self.scraper = DoubanScraper()
 
     def stop(self):
         pass
@@ -130,7 +129,7 @@ class DoubanModule(_ModuleBase):
 
         return ret_medias
 
-    def match(self, name: str, year: str, season: int = None) -> dict:
+    def __match(self, name: str, year: str, season: int = None) -> dict:
         """
         搜索和匹配豆瓣信息
         """
@@ -177,183 +176,12 @@ class DoubanModule(_ModuleBase):
                 if not meta.name:
                     continue
                 # 根据名称查询豆瓣数据
-                doubaninfo = self.match(name=mediainfo.title, year=mediainfo.year, season=meta.begin_season)
+                doubaninfo = self.__match(name=mediainfo.title, year=mediainfo.year, season=meta.begin_season)
                 if not doubaninfo:
                     logger.warn(f"未找到 {mediainfo.title} 的豆瓣信息")
                     break
                 # 刮削
-                self.gen_scraper_files(meta, MediaInfo(douban_info=doubaninfo), file)
+                self.scraper.gen_scraper_files(meta, MediaInfo(douban_info=doubaninfo), file)
             except Exception as e:
                 logger.error(f"刮削文件 {file} 失败，原因：{e}")
             logger.info(f"{file} 刮削完成")
-
-    def gen_scraper_files(self, meta: MetaBase, mediainfo: MediaInfo, file_path: Path):
-        """
-        生成刮削文件
-        :param meta: 元数据
-        :param mediainfo: 媒体信息
-        :param file_path: 文件路径
-        """
-
-        try:
-            # 电影
-            if mediainfo.type == MediaType.MOVIE:
-                # 强制或者不已存在时才处理
-                if not file_path.with_name("movie.nfo").exists() \
-                        and not file_path.with_suffix(".nfo").exists():
-                    #  生成电影描述文件
-                    self.__gen_movie_nfo_file(mediainfo=mediainfo,
-                                              file_path=file_path)
-                # 生成电影图片
-                self.__save_image(url=mediainfo.poster_path,
-                                  file_path=file_path.with_name(f"poster{Path(mediainfo.poster_path).suffix}"))
-            # 电视剧
-            else:
-                # 不存在时才处理
-                if not file_path.parent.with_name("tvshow.nfo").exists():
-                    # 根目录描述文件
-                    self.__gen_tv_nfo_file(mediainfo=mediainfo,
-                                           dir_path=file_path.parents[1])
-                # 生成根目录图片
-                self.__save_image(url=mediainfo.poster_path,
-                                  file_path=file_path.with_name(f"poster{Path(mediainfo.poster_path).suffix}"))
-                # 季目录NFO
-                if not file_path.with_name("season.nfo").exists():
-                    self.__gen_tv_season_nfo_file(mediainfo=mediainfo,
-                                                  season=meta.begin_season,
-                                                  season_path=file_path.parent)
-        except Exception as e:
-            logger.error(f"{file_path} 刮削失败：{e}")
-
-    @staticmethod
-    def __gen_common_nfo(mediainfo: MediaInfo, doc, root):
-        # 添加时间
-        DomUtils.add_node(doc, root, "dateadded",
-                          time.strftime('%Y-%m-%d %H:%M:%S',
-                                        time.localtime(time.time())))
-        # 简介
-        xplot = DomUtils.add_node(doc, root, "plot")
-        xplot.appendChild(doc.createCDATASection(mediainfo.overview or ""))
-        xoutline = DomUtils.add_node(doc, root, "outline")
-        xoutline.appendChild(doc.createCDATASection(mediainfo.overview or ""))
-        # 导演
-        for director in mediainfo.directors:
-            DomUtils.add_node(doc, root, "director", director.get("name") or "")
-        # 演员
-        for actor in mediainfo.actors:
-            xactor = DomUtils.add_node(doc, root, "actor")
-            DomUtils.add_node(doc, xactor, "name", actor.get("name") or "")
-            DomUtils.add_node(doc, xactor, "type", "Actor")
-            DomUtils.add_node(doc, xactor, "role", actor.get("character") or actor.get("role") or "")
-            DomUtils.add_node(doc, xactor, "thumb", actor.get('avatar', {}).get('normal'))
-            DomUtils.add_node(doc, xactor, "profile", actor.get('url'))
-        # 评分
-        DomUtils.add_node(doc, root, "rating", mediainfo.vote_average or "0")
-
-        return doc
-
-    def __gen_movie_nfo_file(self,
-                             mediainfo: MediaInfo,
-                             file_path: Path):
-        """
-        生成电影的NFO描述文件
-        :param mediainfo: 豆瓣信息
-        :param file_path: 电影文件路径
-        """
-        # 开始生成XML
-        logger.info(f"正在生成电影NFO文件：{file_path.name}")
-        doc = minidom.Document()
-        root = DomUtils.add_node(doc, doc, "movie")
-        # 公共部分
-        doc = self.__gen_common_nfo(mediainfo=mediainfo,
-                                    doc=doc,
-                                    root=root)
-        # 标题
-        DomUtils.add_node(doc, root, "title", mediainfo.title or "")
-        # 年份
-        DomUtils.add_node(doc, root, "year", mediainfo.year or "")
-        # 保存
-        self.__save_nfo(doc, file_path.with_suffix(".nfo"))
-
-    def __gen_tv_nfo_file(self,
-                          mediainfo: MediaInfo,
-                          dir_path: Path):
-        """
-        生成电视剧的NFO描述文件
-        :param mediainfo: 媒体信息
-        :param dir_path: 电视剧根目录
-        """
-        # 开始生成XML
-        logger.info(f"正在生成电视剧NFO文件：{dir_path.name}")
-        doc = minidom.Document()
-        root = DomUtils.add_node(doc, doc, "tvshow")
-        # 公共部分
-        doc = self.__gen_common_nfo(mediainfo=mediainfo,
-                                    doc=doc,
-                                    root=root)
-        # 标题
-        DomUtils.add_node(doc, root, "title", mediainfo.title or "")
-        # 年份
-        DomUtils.add_node(doc, root, "year", mediainfo.year or "")
-        DomUtils.add_node(doc, root, "season", "-1")
-        DomUtils.add_node(doc, root, "episode", "-1")
-        # 保存
-        self.__save_nfo(doc, dir_path.joinpath("tvshow.nfo"))
-
-    def __gen_tv_season_nfo_file(self, mediainfo: MediaInfo, season: int, season_path: Path):
-        """
-        生成电视剧季的NFO描述文件
-        :param mediainfo: 媒体信息
-        :param season: 季号
-        :param season_path: 电视剧季的目录
-        """
-        logger.info(f"正在生成季NFO文件：{season_path.name}")
-        doc = minidom.Document()
-        root = DomUtils.add_node(doc, doc, "season")
-        # 添加时间
-        DomUtils.add_node(doc, root, "dateadded", time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time())))
-        # 简介
-        xplot = DomUtils.add_node(doc, root, "plot")
-        xplot.appendChild(doc.createCDATASection(mediainfo.overview or ""))
-        xoutline = DomUtils.add_node(doc, root, "outline")
-        xoutline.appendChild(doc.createCDATASection(mediainfo.overview or ""))
-        # 标题
-        DomUtils.add_node(doc, root, "title", "季 %s" % season)
-        # 发行日期
-        DomUtils.add_node(doc, root, "premiered", mediainfo.release_date or "")
-        DomUtils.add_node(doc, root, "releasedate", mediainfo.release_date or "")
-        # 发行年份
-        DomUtils.add_node(doc, root, "year", mediainfo.release_date[:4] if mediainfo.release_date else "")
-        # seasonnumber
-        DomUtils.add_node(doc, root, "seasonnumber", str(season))
-        # 保存
-        self.__save_nfo(doc, season_path.joinpath("season.nfo"))
-
-    @staticmethod
-    def __save_image(url: str, file_path: Path):
-        """
-        下载图片并保存
-        """
-        if file_path.exists():
-            return
-        try:
-            logger.info(f"正在下载{file_path.stem}图片：{url} ...")
-            r = RequestUtils().get_res(url=url)
-            if r:
-                file_path.write_bytes(r.content)
-                logger.info(f"图片已保存：{file_path}")
-            else:
-                logger.info(f"{file_path.stem}图片下载失败，请检查网络连通性")
-        except Exception as err:
-            logger.error(f"{file_path.stem}图片下载失败：{err}")
-
-    @staticmethod
-    def __save_nfo(doc, file_path: Path):
-        """
-        保存NFO
-        """
-        if file_path.exists():
-            return
-        xml_str = doc.toprettyxml(indent="  ", encoding="utf-8")
-        file_path.write_bytes(xml_str)
-        logger.info(f"NFO文件已保存：{file_path}")

app/modules/douban/scraper.py

@@ -0,0 +1,184 @@
+import time
+from pathlib import Path
+from xml.dom import minidom
+
+from app.core.context import MediaInfo
+from app.core.meta import MetaBase
+from app.log import logger
+from app.schemas.types import MediaType
+from app.utils.dom import DomUtils
+from app.utils.http import RequestUtils
+
+
+class DoubanScraper:
+
+    def gen_scraper_files(self, meta: MetaBase, mediainfo: MediaInfo, file_path: Path):
+        """
+        生成刮削文件
+        :param meta: 元数据
+        :param mediainfo: 媒体信息
+        :param file_path: 文件路径
+        """
+
+        try:
+            # 电影
+            if mediainfo.type == MediaType.MOVIE:
+                # 强制或者不已存在时才处理
+                if not file_path.with_name("movie.nfo").exists() \
+                        and not file_path.with_suffix(".nfo").exists():
+                    #  生成电影描述文件
+                    self.__gen_movie_nfo_file(mediainfo=mediainfo,
+                                              file_path=file_path)
+                # 生成电影图片
+                self.__save_image(url=mediainfo.poster_path,
+                                  file_path=file_path.with_name(f"poster{Path(mediainfo.poster_path).suffix}"))
+            # 电视剧
+            else:
+                # 不存在时才处理
+                if not file_path.parent.with_name("tvshow.nfo").exists():
+                    # 根目录描述文件
+                    self.__gen_tv_nfo_file(mediainfo=mediainfo,
+                                           dir_path=file_path.parents[1])
+                # 生成根目录图片
+                self.__save_image(url=mediainfo.poster_path,
+                                  file_path=file_path.with_name(f"poster{Path(mediainfo.poster_path).suffix}"))
+                # 季目录NFO
+                if not file_path.with_name("season.nfo").exists():
+                    self.__gen_tv_season_nfo_file(mediainfo=mediainfo,
+                                                  season=meta.begin_season,
+                                                  season_path=file_path.parent)
+        except Exception as e:
+            logger.error(f"{file_path} 刮削失败：{e}")
+
+    @staticmethod
+    def __gen_common_nfo(mediainfo: MediaInfo, doc, root):
+        # 添加时间
+        DomUtils.add_node(doc, root, "dateadded",
+                          time.strftime('%Y-%m-%d %H:%M:%S',
+                                        time.localtime(time.time())))
+        # 简介
+        xplot = DomUtils.add_node(doc, root, "plot")
+        xplot.appendChild(doc.createCDATASection(mediainfo.overview or ""))
+        xoutline = DomUtils.add_node(doc, root, "outline")
+        xoutline.appendChild(doc.createCDATASection(mediainfo.overview or ""))
+        # 导演
+        for director in mediainfo.directors:
+            DomUtils.add_node(doc, root, "director", director.get("name") or "")
+        # 演员
+        for actor in mediainfo.actors:
+            xactor = DomUtils.add_node(doc, root, "actor")
+            DomUtils.add_node(doc, xactor, "name", actor.get("name") or "")
+            DomUtils.add_node(doc, xactor, "type", "Actor")
+            DomUtils.add_node(doc, xactor, "role", actor.get("character") or actor.get("role") or "")
+            DomUtils.add_node(doc, xactor, "thumb", actor.get('avatar', {}).get('normal'))
+            DomUtils.add_node(doc, xactor, "profile", actor.get('url'))
+        # 评分
+        DomUtils.add_node(doc, root, "rating", mediainfo.vote_average or "0")
+
+        return doc
+
+    def __gen_movie_nfo_file(self,
+                             mediainfo: MediaInfo,
+                             file_path: Path):
+        """
+        生成电影的NFO描述文件
+        :param mediainfo: 豆瓣信息
+        :param file_path: 电影文件路径
+        """
+        # 开始生成XML
+        logger.info(f"正在生成电影NFO文件：{file_path.name}")
+        doc = minidom.Document()
+        root = DomUtils.add_node(doc, doc, "movie")
+        # 公共部分
+        doc = self.__gen_common_nfo(mediainfo=mediainfo,
+                                    doc=doc,
+                                    root=root)
+        # 标题
+        DomUtils.add_node(doc, root, "title", mediainfo.title or "")
+        # 年份
+        DomUtils.add_node(doc, root, "year", mediainfo.year or "")
+        # 保存
+        self.__save_nfo(doc, file_path.with_suffix(".nfo"))
+
+    def __gen_tv_nfo_file(self,
+                          mediainfo: MediaInfo,
+                          dir_path: Path):
+        """
+        生成电视剧的NFO描述文件
+        :param mediainfo: 媒体信息
+        :param dir_path: 电视剧根目录
+        """
+        # 开始生成XML
+        logger.info(f"正在生成电视剧NFO文件：{dir_path.name}")
+        doc = minidom.Document()
+        root = DomUtils.add_node(doc, doc, "tvshow")
+        # 公共部分
+        doc = self.__gen_common_nfo(mediainfo=mediainfo,
+                                    doc=doc,
+                                    root=root)
+        # 标题
+        DomUtils.add_node(doc, root, "title", mediainfo.title or "")
+        # 年份
+        DomUtils.add_node(doc, root, "year", mediainfo.year or "")
+        DomUtils.add_node(doc, root, "season", "-1")
+        DomUtils.add_node(doc, root, "episode", "-1")
+        # 保存
+        self.__save_nfo(doc, dir_path.joinpath("tvshow.nfo"))
+
+    def __gen_tv_season_nfo_file(self, mediainfo: MediaInfo, season: int, season_path: Path):
+        """
+        生成电视剧季的NFO描述文件
+        :param mediainfo: 媒体信息
+        :param season: 季号
+        :param season_path: 电视剧季的目录
+        """
+        logger.info(f"正在生成季NFO文件：{season_path.name}")
+        doc = minidom.Document()
+        root = DomUtils.add_node(doc, doc, "season")
+        # 添加时间
+        DomUtils.add_node(doc, root, "dateadded", time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time())))
+        # 简介
+        xplot = DomUtils.add_node(doc, root, "plot")
+        xplot.appendChild(doc.createCDATASection(mediainfo.overview or ""))
+        xoutline = DomUtils.add_node(doc, root, "outline")
+        xoutline.appendChild(doc.createCDATASection(mediainfo.overview or ""))
+        # 标题
+        DomUtils.add_node(doc, root, "title", "季 %s" % season)
+        # 发行日期
+        DomUtils.add_node(doc, root, "premiered", mediainfo.release_date or "")
+        DomUtils.add_node(doc, root, "releasedate", mediainfo.release_date or "")
+        # 发行年份
+        DomUtils.add_node(doc, root, "year", mediainfo.release_date[:4] if mediainfo.release_date else "")
+        # seasonnumber
+        DomUtils.add_node(doc, root, "seasonnumber", str(season))
+        # 保存
+        self.__save_nfo(doc, season_path.joinpath("season.nfo"))
+
+    @staticmethod
+    def __save_image(url: str, file_path: Path):
+        """
+        下载图片并保存
+        """
+        if file_path.exists():
+            return
+        try:
+            logger.info(f"正在下载{file_path.stem}图片：{url} ...")
+            r = RequestUtils().get_res(url=url)
+            if r:
+                file_path.write_bytes(r.content)
+                logger.info(f"图片已保存：{file_path}")
+            else:
+                logger.info(f"{file_path.stem}图片下载失败，请检查网络连通性")
+        except Exception as err:
+            logger.error(f"{file_path.stem}图片下载失败：{err}")
+
+    @staticmethod
+    def __save_nfo(doc, file_path: Path):
+        """
+        保存NFO
+        """
+        if file_path.exists():
+            return
+        xml_str = doc.toprettyxml(indent="  ", encoding="utf-8")
+        file_path.write_bytes(xml_str)
+        logger.info(f"NFO文件已保存：{file_path}")

app/modules/themoviedb/__init__.py

@@ -1,21 +1,18 @@
-import time
 from pathlib import Path
 from typing import Optional, List, Tuple, Union
-from xml.dom import minidom
 
+from app import schemas
 from app.core.config import settings
 from app.core.context import MediaInfo
-from app.core.metainfo import MetaInfo
 from app.core.meta import MetaBase
 from app.log import logger
 from app.modules import _ModuleBase
 from app.modules.themoviedb.category import CategoryHelper
+from app.modules.themoviedb.scraper import TmdbScraper
 from app.modules.themoviedb.tmdb import TmdbHelper
 from app.modules.themoviedb.tmdb_cache import TmdbCache
-from app.utils.dom import DomUtils
-from app.utils.http import RequestUtils
-from app.utils.system import SystemUtils
 from app.schemas.types import MediaType, MediaImageType
+from app.utils.system import SystemUtils
 
 
 class TheMovieDbModule(_ModuleBase):
@@ -29,11 +26,14 @@ class TheMovieDbModule(_ModuleBase):
     tmdb: TmdbHelper = None
     # 二级分类
     category: CategoryHelper = None
+    # 刮削器
+    scraper: TmdbScraper = None
 
     def init_module(self) -> None:
         self.cache = TmdbCache()
         self.tmdb = TmdbHelper()
         self.category = CategoryHelper()
+        self.scraper = TmdbScraper(self.tmdb)
 
     def stop(self):
         self.cache.save()
@@ -194,8 +194,8 @@ class TheMovieDbModule(_ModuleBase):
             if not file:
                 continue
             logger.info(f"开始刮削媒体库文件：{file} ...")
-            self.gen_scraper_files(mediainfo=mediainfo,
+            self.scraper.gen_scraper_files(mediainfo=mediainfo,
-                                   file_path=file)
+                                           file_path=file)
             logger.info(f"{file} 刮削完成")
 
     def tmdb_discover(self, mtype: MediaType, sort_by: str, with_genres: str, with_original_language: str,
@@ -229,334 +229,26 @@ class TheMovieDbModule(_ModuleBase):
         """
         return self.tmdb.trending.all_week(page=page)
 
-    def gen_scraper_files(self, mediainfo: MediaInfo, file_path: Path):
+    def tmdb_seasons(self, tmdbid: int) -> List[schemas.TmdbSeason]:
         """
-        生成刮削文件
+        根据TMDBID查询themoviedb所有季信息
-        :param mediainfo: 媒体信息
+        :param tmdbid:  TMDBID
-        :param file_path: 文件路径
         """
+        tmdb_info = self.tmdb.get_info(tmdbid=tmdbid, mtype=MediaType.TV)
+        if not tmdb_info:
+            return []
+        return [schemas.TmdbSeason(**season) for season in tmdb_info.get("seasons", [])]
 
-        def __get_episode_detail(_seasoninfo: dict, _episode: int):
+    def tmdb_episodes(self, tmdbid: int, season: int) -> List[schemas.TmdbEpisode]:
-            """
-            根据季信息获取集的信息
-            """
-            for _episode_info in _seasoninfo.get("episodes") or []:
-                if _episode_info.get("episode_number") == _episode:
-                    return _episode_info
-            return {}
-
-        try:
-            # 电影
-            if mediainfo.type == MediaType.MOVIE:
-                # 强制或者不已存在时才处理
-                if not file_path.with_name("movie.nfo").exists() \
-                        and not file_path.with_suffix(".nfo").exists():
-                    #  生成电影描述文件
-                    self.__gen_movie_nfo_file(mediainfo=mediainfo,
-                                              file_path=file_path)
-                # 生成电影图片
-                for attr_name, attr_value in vars(mediainfo).items():
-                    if attr_value \
-                            and attr_name.endswith("_path") \
-                            and attr_value \
-                            and isinstance(attr_value, str) \
-                            and attr_value.startswith("http"):
-                        image_name = attr_name.replace("_path", "") + Path(attr_value).suffix
-                        self.__save_image(url=attr_value,
-                                          file_path=file_path.with_name(image_name))
-            # 电视剧
-            else:
-                # 识别
-                meta = MetaInfo(file_path.stem)
-                # 不存在时才处理
-                if not file_path.parent.with_name("tvshow.nfo").exists():
-                    # 根目录描述文件
-                    self.__gen_tv_nfo_file(mediainfo=mediainfo,
-                                           dir_path=file_path.parents[1])
-                # 生成根目录图片
-                for attr_name, attr_value in vars(mediainfo).items():
-                    if attr_value \
-                            and attr_name.endswith("_path") \
-                            and not attr_name.startswith("season") \
-                            and attr_value \
-                            and isinstance(attr_value, str) \
-                            and attr_value.startswith("http"):
-                        image_name = attr_name.replace("_path", "") + Path(attr_value).suffix
-                        self.__save_image(url=attr_value,
-                                          file_path=file_path.parent.with_name(image_name))
-                # 查询季信息
-                seasoninfo = self.tmdb.get_tv_season_detail(mediainfo.tmdb_id, meta.begin_season)
-                if seasoninfo:
-                    # 季目录NFO
-                    if not file_path.with_name("season.nfo").exists():
-                        self.__gen_tv_season_nfo_file(seasoninfo=seasoninfo,
-                                                      season=meta.begin_season,
-                                                      season_path=file_path.parent)
-                    # 季的图片
-                    for attr_name, attr_value in vars(mediainfo).items():
-                        if attr_value \
-                                and attr_name.startswith("season") \
-                                and attr_value \
-                                and isinstance(attr_value, str) \
-                                and attr_value.startswith("http"):
-                            image_name = attr_name.replace("_path",
-                                                           "").replace("season",
-                                                                       f"{str(meta.begin_season).rjust(2, '0')}-") \
-                                         + Path(attr_value).suffix
-                            self.__save_image(url=attr_value,
-                                              file_path=file_path.parent.with_name(f"season{image_name}"))
-                # 查询集详情
-                episodeinfo = __get_episode_detail(seasoninfo, meta.begin_episode)
-                if episodeinfo:
-                    # 集NFO
-                    if not file_path.with_suffix(".nfo").exists():
-                        self.__gen_tv_episode_nfo_file(episodeinfo=episodeinfo,
-                                                       season=meta.begin_season,
-                                                       episode=meta.begin_episode,
-                                                       file_path=file_path)
-                    # 集的图片
-                    if episodeinfo.get('still_path'):
-                        self.__save_image(
-                            f"https://{settings.TMDB_IMAGE_DOMAIN}/t/p/original{episodeinfo.get('still_path')}",
-                            file_path.with_suffix(Path(episodeinfo.get('still_path')).suffix))
-        except Exception as e:
-            logger.error(f"{file_path} 刮削失败：{e}")
-
-    @staticmethod
-    def __gen_common_nfo(mediainfo: MediaInfo, doc, root):
         """
-        生成公共NFO
+        根据TMDBID查询某季的所有信信息
+        :param tmdbid:  TMDBID
+        :param season:  季
         """
-        # TMDBINFO
+        season_info = self.tmdb.get_tv_season_detail(tmdbid=tmdbid, season=season)
-        tmdbinfo = mediainfo.tmdb_info
+        if not season_info:
-        # 添加时间
+            return []
-        DomUtils.add_node(doc, root, "dateadded",
+        return [schemas.TmdbEpisode(**episode) for episode in season_info.get("episodes", [])]
-                          time.strftime('%Y-%m-%d %H:%M:%S',
-                                        time.localtime(time.time())))
-        # TMDB
-        DomUtils.add_node(doc, root, "tmdbid", mediainfo.tmdb_id or "")
-        uniqueid_tmdb = DomUtils.add_node(doc, root, "uniqueid", mediainfo.tmdb_id or "")
-        uniqueid_tmdb.setAttribute("type", "tmdb")
-        uniqueid_tmdb.setAttribute("default", "true")
-        # TVDB
-        if mediainfo.tvdb_id:
-            DomUtils.add_node(doc, root, "tvdbid", str(mediainfo.tvdb_id))
-            uniqueid_tvdb = DomUtils.add_node(doc, root, "uniqueid", str(mediainfo.tvdb_id))
-            uniqueid_tvdb.setAttribute("type", "tvdb")
-        # IMDB
-        if mediainfo.imdb_id:
-            DomUtils.add_node(doc, root, "imdbid", mediainfo.imdb_id)
-            uniqueid_imdb = DomUtils.add_node(doc, root, "uniqueid", mediainfo.imdb_id)
-            uniqueid_imdb.setAttribute("type", "imdb")
-            uniqueid_imdb.setAttribute("default", "true")
-            uniqueid_tmdb.setAttribute("default", "false")
-
-        # 简介
-        xplot = DomUtils.add_node(doc, root, "plot")
-        xplot.appendChild(doc.createCDATASection(mediainfo.overview or ""))
-        xoutline = DomUtils.add_node(doc, root, "outline")
-        xoutline.appendChild(doc.createCDATASection(mediainfo.overview or ""))
-        # 导演
-        for director in mediainfo.directors:
-            xdirector = DomUtils.add_node(doc, root, "director", director.get("name") or "")
-            xdirector.setAttribute("tmdbid", str(director.get("id") or ""))
-        # 演员
-        for actor in mediainfo.actors:
-            xactor = DomUtils.add_node(doc, root, "actor")
-            DomUtils.add_node(doc, xactor, "name", actor.get("name") or "")
-            DomUtils.add_node(doc, xactor, "type", "Actor")
-            DomUtils.add_node(doc, xactor, "role", actor.get("character") or actor.get("role") or "")
-            DomUtils.add_node(doc, xactor, "order", actor.get("order") if actor.get("order") is not None else "")
-            DomUtils.add_node(doc, xactor, "tmdbid", actor.get("id") or "")
-            DomUtils.add_node(doc, xactor, "thumb", actor.get('image'))
-            DomUtils.add_node(doc, xactor, "profile", actor.get('profile'))
-        # 风格
-        genres = tmdbinfo.get("genres") or []
-        for genre in genres:
-            DomUtils.add_node(doc, root, "genre", genre.get("name") or "")
-        # 评分
-        DomUtils.add_node(doc, root, "rating", mediainfo.vote_average or "0")
-        # 评级
-        if tmdbinfo.get("releases") and tmdbinfo.get("releases").get("countries"):
-            releases = [i for i in tmdbinfo.get("releases").get("countries") if
-                        i.get("certification") and i.get("certification").strip()]
-            # 国内没有分级，所以沿用美国的分级
-            us_release = next((c for c in releases if c.get("iso_3166_1") == "US"), None)
-            if us_release:
-                DomUtils.add_node(doc, root, "mpaa", us_release.get("certification") or "")
-
-        return doc
-
-    def __gen_movie_nfo_file(self,
-                             mediainfo: MediaInfo,
-                             file_path: Path):
-        """
-        生成电影的NFO描述文件
-        :param mediainfo: 识别后的媒体信息
-        :param file_path: 电影文件路径
-        """
-        # 开始生成XML
-        logger.info(f"正在生成电影NFO文件：{file_path.name}")
-        doc = minidom.Document()
-        root = DomUtils.add_node(doc, doc, "movie")
-        # 公共部分
-        doc = self.__gen_common_nfo(mediainfo=mediainfo,
-                                    doc=doc,
-                                    root=root)
-        # 标题
-        DomUtils.add_node(doc, root, "title", mediainfo.title or "")
-        DomUtils.add_node(doc, root, "originaltitle", mediainfo.original_title or "")
-        # 发布日期
-        DomUtils.add_node(doc, root, "premiered", mediainfo.release_date or "")
-        # 年份
-        DomUtils.add_node(doc, root, "year", mediainfo.year or "")
-        # 保存
-        self.__save_nfo(doc, file_path.with_suffix(".nfo"))
-
-    def __gen_tv_nfo_file(self,
-                          mediainfo: MediaInfo,
-                          dir_path: Path):
-        """
-        生成电视剧的NFO描述文件
-        :param mediainfo: 媒体信息
-        :param dir_path: 电视剧根目录
-        """
-        # 开始生成XML
-        logger.info(f"正在生成电视剧NFO文件：{dir_path.name}")
-        doc = minidom.Document()
-        root = DomUtils.add_node(doc, doc, "tvshow")
-        # 公共部分
-        doc = self.__gen_common_nfo(mediainfo=mediainfo,
-                                    doc=doc,
-                                    root=root)
-        # 标题
-        DomUtils.add_node(doc, root, "title", mediainfo.title or "")
-        DomUtils.add_node(doc, root, "originaltitle", mediainfo.original_title or "")
-        # 发布日期
-        DomUtils.add_node(doc, root, "premiered", mediainfo.release_date or "")
-        # 年份
-        DomUtils.add_node(doc, root, "year", mediainfo.year or "")
-        DomUtils.add_node(doc, root, "season", "-1")
-        DomUtils.add_node(doc, root, "episode", "-1")
-        # 保存
-        self.__save_nfo(doc, dir_path.joinpath("tvshow.nfo"))
-
-    def __gen_tv_season_nfo_file(self, seasoninfo: dict, season: int, season_path: Path):
-        """
-        生成电视剧季的NFO描述文件
-        :param seasoninfo: TMDB季媒体信息
-        :param season: 季号
-        :param season_path: 电视剧季的目录
-        """
-        logger.info(f"正在生成季NFO文件：{season_path.name}")
-        doc = minidom.Document()
-        root = DomUtils.add_node(doc, doc, "season")
-        # 添加时间
-        DomUtils.add_node(doc, root, "dateadded", time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time())))
-        # 简介
-        xplot = DomUtils.add_node(doc, root, "plot")
-        xplot.appendChild(doc.createCDATASection(seasoninfo.get("overview") or ""))
-        xoutline = DomUtils.add_node(doc, root, "outline")
-        xoutline.appendChild(doc.createCDATASection(seasoninfo.get("overview") or ""))
-        # 标题
-        DomUtils.add_node(doc, root, "title", "季 %s" % season)
-        # 发行日期
-        DomUtils.add_node(doc, root, "premiered", seasoninfo.get("air_date") or "")
-        DomUtils.add_node(doc, root, "releasedate", seasoninfo.get("air_date") or "")
-        # 发行年份
-        DomUtils.add_node(doc, root, "year", seasoninfo.get("air_date")[:4] if seasoninfo.get("air_date") else "")
-        # seasonnumber
-        DomUtils.add_node(doc, root, "seasonnumber", str(season))
-        # 保存
-        self.__save_nfo(doc, season_path.joinpath("season.nfo"))
-
-    def __gen_tv_episode_nfo_file(self,
-                                  episodeinfo: dict,
-                                  season: int,
-                                  episode: int,
-                                  file_path: Path):
-        """
-        生成电视剧集的NFO描述文件
-        :param episodeinfo: 集TMDB元数据
-        :param season: 季号
-        :param episode: 集号
-        :param file_path: 集文件的路径
-        """
-        # 开始生成集的信息
-        logger.info(f"正在生成剧集NFO文件：{file_path.name}")
-        doc = minidom.Document()
-        root = DomUtils.add_node(doc, doc, "episodedetails")
-        # 添加时间
-        DomUtils.add_node(doc, root, "dateadded", time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time())))
-        # TMDBID
-        uniqueid = DomUtils.add_node(doc, root, "uniqueid", episodeinfo.get("id") or "")
-        uniqueid.setAttribute("type", "tmdb")
-        uniqueid.setAttribute("default", "true")
-        # tmdbid
-        DomUtils.add_node(doc, root, "tmdbid", episodeinfo.get("id") or "")
-        # 标题
-        DomUtils.add_node(doc, root, "title", episodeinfo.get("name") or "第 %s 集" % episode)
-        # 简介
-        xplot = DomUtils.add_node(doc, root, "plot")
-        xplot.appendChild(doc.createCDATASection(episodeinfo.get("overview") or ""))
-        xoutline = DomUtils.add_node(doc, root, "outline")
-        xoutline.appendChild(doc.createCDATASection(episodeinfo.get("overview") or ""))
-        # 发布日期
-        DomUtils.add_node(doc, root, "aired", episodeinfo.get("air_date") or "")
-        # 年份
-        DomUtils.add_node(doc, root, "year",
-                          episodeinfo.get("air_date")[:4] if episodeinfo.get("air_date") else "")
-        # 季
-        DomUtils.add_node(doc, root, "season", str(season))
-        # 集
-        DomUtils.add_node(doc, root, "episode", str(episode))
-        # 评分
-        DomUtils.add_node(doc, root, "rating", episodeinfo.get("vote_average") or "0")
-        # 导演
-        directors = episodeinfo.get("crew") or []
-        for director in directors:
-            if director.get("known_for_department") == "Directing":
-                xdirector = DomUtils.add_node(doc, root, "director", director.get("name") or "")
-                xdirector.setAttribute("tmdbid", str(director.get("id") or ""))
-        # 演员
-        actors = episodeinfo.get("guest_stars") or []
-        for actor in actors:
-            if actor.get("known_for_department") == "Acting":
-                xactor = DomUtils.add_node(doc, root, "actor")
-                DomUtils.add_node(doc, xactor, "name", actor.get("name") or "")
-                DomUtils.add_node(doc, xactor, "type", "Actor")
-                DomUtils.add_node(doc, xactor, "tmdbid", actor.get("id") or "")
-        # 保存文件
-        self.__save_nfo(doc, file_path.with_suffix(".nfo"))
-
-    @staticmethod
-    def __save_image(url: str, file_path: Path):
-        """
-        下载图片并保存
-        """
-        if file_path.exists():
-            return
-        try:
-            logger.info(f"正在下载{file_path.stem}图片：{url} ...")
-            r = RequestUtils().get_res(url=url)
-            if r:
-                file_path.write_bytes(r.content)
-                logger.info(f"图片已保存：{file_path}")
-            else:
-                logger.info(f"{file_path.stem}图片下载失败，请检查网络连通性")
-        except Exception as err:
-            logger.error(f"{file_path.stem}图片下载失败：{err}")
-
-    @staticmethod
-    def __save_nfo(doc, file_path: Path):
-        """
-        保存NFO
-        """
-        if file_path.exists():
-            return
-        xml_str = doc.toprettyxml(indent="  ", encoding="utf-8")
-        file_path.write_bytes(xml_str)
-        logger.info(f"NFO文件已保存：{file_path}")
 
     def scheduler_job(self) -> None:
         """

app/modules/themoviedb/scraper.py

@@ -0,0 +1,348 @@
+import time
+from pathlib import Path
+from xml.dom import minidom
+
+from app.core.config import settings
+from app.core.context import MediaInfo
+from app.core.metainfo import MetaInfo
+from app.log import logger
+from app.schemas.types import MediaType
+from app.utils.dom import DomUtils
+from app.utils.http import RequestUtils
+
+
+class TmdbScraper:
+
+    tmdb = None
+
+    def __init__(self, tmdb):
+        self.tmdb = tmdb
+
+    def gen_scraper_files(self, mediainfo: MediaInfo, file_path: Path):
+        """
+        生成刮削文件
+        :param mediainfo: 媒体信息
+        :param file_path: 文件路径
+        """
+
+        def __get_episode_detail(_seasoninfo: dict, _episode: int):
+            """
+            根据季信息获取集的信息
+            """
+            for _episode_info in _seasoninfo.get("episodes") or []:
+                if _episode_info.get("episode_number") == _episode:
+                    return _episode_info
+            return {}
+
+        try:
+            # 电影
+            if mediainfo.type == MediaType.MOVIE:
+                # 强制或者不已存在时才处理
+                if not file_path.with_name("movie.nfo").exists() \
+                        and not file_path.with_suffix(".nfo").exists():
+                    #  生成电影描述文件
+                    self.__gen_movie_nfo_file(mediainfo=mediainfo,
+                                              file_path=file_path)
+                # 生成电影图片
+                for attr_name, attr_value in vars(mediainfo).items():
+                    if attr_value \
+                            and attr_name.endswith("_path") \
+                            and attr_value \
+                            and isinstance(attr_value, str) \
+                            and attr_value.startswith("http"):
+                        image_name = attr_name.replace("_path", "") + Path(attr_value).suffix
+                        self.__save_image(url=attr_value,
+                                          file_path=file_path.with_name(image_name))
+            # 电视剧
+            else:
+                # 识别
+                meta = MetaInfo(file_path.stem)
+                # 不存在时才处理
+                if not file_path.parent.with_name("tvshow.nfo").exists():
+                    # 根目录描述文件
+                    self.__gen_tv_nfo_file(mediainfo=mediainfo,
+                                           dir_path=file_path.parents[1])
+                # 生成根目录图片
+                for attr_name, attr_value in vars(mediainfo).items():
+                    if attr_value \
+                            and attr_name.endswith("_path") \
+                            and not attr_name.startswith("season") \
+                            and attr_value \
+                            and isinstance(attr_value, str) \
+                            and attr_value.startswith("http"):
+                        image_name = attr_name.replace("_path", "") + Path(attr_value).suffix
+                        self.__save_image(url=attr_value,
+                                          file_path=file_path.parent.with_name(image_name))
+                # 查询季信息
+                seasoninfo = self.tmdb.get_tv_season_detail(mediainfo.tmdb_id, meta.begin_season)
+                if seasoninfo:
+                    # 季目录NFO
+                    if not file_path.with_name("season.nfo").exists():
+                        self.__gen_tv_season_nfo_file(seasoninfo=seasoninfo,
+                                                      season=meta.begin_season,
+                                                      season_path=file_path.parent)
+                    # 季的图片
+                    for attr_name, attr_value in vars(mediainfo).items():
+                        if attr_value \
+                                and attr_name.startswith("season") \
+                                and attr_value \
+                                and isinstance(attr_value, str) \
+                                and attr_value.startswith("http"):
+                            image_name = attr_name.replace("_path",
+                                                           "").replace("season",
+                                                                       f"{str(meta.begin_season).rjust(2, '0')}-") \
+                                         + Path(attr_value).suffix
+                            self.__save_image(url=attr_value,
+                                              file_path=file_path.parent.with_name(f"season{image_name}"))
+                # 查询集详情
+                episodeinfo = __get_episode_detail(seasoninfo, meta.begin_episode)
+                if episodeinfo:
+                    # 集NFO
+                    if not file_path.with_suffix(".nfo").exists():
+                        self.__gen_tv_episode_nfo_file(episodeinfo=episodeinfo,
+                                                       season=meta.begin_season,
+                                                       episode=meta.begin_episode,
+                                                       file_path=file_path)
+                    # 集的图片
+                    if episodeinfo.get('still_path'):
+                        self.__save_image(
+                            f"https://{settings.TMDB_IMAGE_DOMAIN}/t/p/original{episodeinfo.get('still_path')}",
+                            file_path.with_suffix(Path(episodeinfo.get('still_path')).suffix))
+        except Exception as e:
+            logger.error(f"{file_path} 刮削失败：{e}")
+
+    @staticmethod
+    def __gen_common_nfo(mediainfo: MediaInfo, doc, root):
+        """
+        生成公共NFO
+        """
+        # TMDBINFO
+        tmdbinfo = mediainfo.tmdb_info
+        # 添加时间
+        DomUtils.add_node(doc, root, "dateadded",
+                          time.strftime('%Y-%m-%d %H:%M:%S',
+                                        time.localtime(time.time())))
+        # TMDB
+        DomUtils.add_node(doc, root, "tmdbid", mediainfo.tmdb_id or "")
+        uniqueid_tmdb = DomUtils.add_node(doc, root, "uniqueid", mediainfo.tmdb_id or "")
+        uniqueid_tmdb.setAttribute("type", "tmdb")
+        uniqueid_tmdb.setAttribute("default", "true")
+        # TVDB
+        if mediainfo.tvdb_id:
+            DomUtils.add_node(doc, root, "tvdbid", str(mediainfo.tvdb_id))
+            uniqueid_tvdb = DomUtils.add_node(doc, root, "uniqueid", str(mediainfo.tvdb_id))
+            uniqueid_tvdb.setAttribute("type", "tvdb")
+        # IMDB
+        if mediainfo.imdb_id:
+            DomUtils.add_node(doc, root, "imdbid", mediainfo.imdb_id)
+            uniqueid_imdb = DomUtils.add_node(doc, root, "uniqueid", mediainfo.imdb_id)
+            uniqueid_imdb.setAttribute("type", "imdb")
+            uniqueid_imdb.setAttribute("default", "true")
+            uniqueid_tmdb.setAttribute("default", "false")
+
+        # 简介
+        xplot = DomUtils.add_node(doc, root, "plot")
+        xplot.appendChild(doc.createCDATASection(mediainfo.overview or ""))
+        xoutline = DomUtils.add_node(doc, root, "outline")
+        xoutline.appendChild(doc.createCDATASection(mediainfo.overview or ""))
+        # 导演
+        for director in mediainfo.directors:
+            xdirector = DomUtils.add_node(doc, root, "director", director.get("name") or "")
+            xdirector.setAttribute("tmdbid", str(director.get("id") or ""))
+        # 演员
+        for actor in mediainfo.actors:
+            xactor = DomUtils.add_node(doc, root, "actor")
+            DomUtils.add_node(doc, xactor, "name", actor.get("name") or "")
+            DomUtils.add_node(doc, xactor, "type", "Actor")
+            DomUtils.add_node(doc, xactor, "role", actor.get("character") or actor.get("role") or "")
+            DomUtils.add_node(doc, xactor, "order", actor.get("order") if actor.get("order") is not None else "")
+            DomUtils.add_node(doc, xactor, "tmdbid", actor.get("id") or "")
+            DomUtils.add_node(doc, xactor, "thumb", actor.get('image'))
+            DomUtils.add_node(doc, xactor, "profile", actor.get('profile'))
+        # 风格
+        genres = tmdbinfo.get("genres") or []
+        for genre in genres:
+            DomUtils.add_node(doc, root, "genre", genre.get("name") or "")
+        # 评分
+        DomUtils.add_node(doc, root, "rating", mediainfo.vote_average or "0")
+        # 评级
+        if tmdbinfo.get("releases") and tmdbinfo.get("releases").get("countries"):
+            releases = [i for i in tmdbinfo.get("releases").get("countries") if
+                        i.get("certification") and i.get("certification").strip()]
+            # 国内没有分级，所以沿用美国的分级
+            us_release = next((c for c in releases if c.get("iso_3166_1") == "US"), None)
+            if us_release:
+                DomUtils.add_node(doc, root, "mpaa", us_release.get("certification") or "")
+
+        return doc
+
+    def __gen_movie_nfo_file(self,
+                             mediainfo: MediaInfo,
+                             file_path: Path):
+        """
+        生成电影的NFO描述文件
+        :param mediainfo: 识别后的媒体信息
+        :param file_path: 电影文件路径
+        """
+        # 开始生成XML
+        logger.info(f"正在生成电影NFO文件：{file_path.name}")
+        doc = minidom.Document()
+        root = DomUtils.add_node(doc, doc, "movie")
+        # 公共部分
+        doc = self.__gen_common_nfo(mediainfo=mediainfo,
+                                    doc=doc,
+                                    root=root)
+        # 标题
+        DomUtils.add_node(doc, root, "title", mediainfo.title or "")
+        DomUtils.add_node(doc, root, "originaltitle", mediainfo.original_title or "")
+        # 发布日期
+        DomUtils.add_node(doc, root, "premiered", mediainfo.release_date or "")
+        # 年份
+        DomUtils.add_node(doc, root, "year", mediainfo.year or "")
+        # 保存
+        self.__save_nfo(doc, file_path.with_suffix(".nfo"))
+
+    def __gen_tv_nfo_file(self,
+                          mediainfo: MediaInfo,
+                          dir_path: Path):
+        """
+        生成电视剧的NFO描述文件
+        :param mediainfo: 媒体信息
+        :param dir_path: 电视剧根目录
+        """
+        # 开始生成XML
+        logger.info(f"正在生成电视剧NFO文件：{dir_path.name}")
+        doc = minidom.Document()
+        root = DomUtils.add_node(doc, doc, "tvshow")
+        # 公共部分
+        doc = self.__gen_common_nfo(mediainfo=mediainfo,
+                                    doc=doc,
+                                    root=root)
+        # 标题
+        DomUtils.add_node(doc, root, "title", mediainfo.title or "")
+        DomUtils.add_node(doc, root, "originaltitle", mediainfo.original_title or "")
+        # 发布日期
+        DomUtils.add_node(doc, root, "premiered", mediainfo.release_date or "")
+        # 年份
+        DomUtils.add_node(doc, root, "year", mediainfo.year or "")
+        DomUtils.add_node(doc, root, "season", "-1")
+        DomUtils.add_node(doc, root, "episode", "-1")
+        # 保存
+        self.__save_nfo(doc, dir_path.joinpath("tvshow.nfo"))
+
+    def __gen_tv_season_nfo_file(self, seasoninfo: dict, season: int, season_path: Path):
+        """
+        生成电视剧季的NFO描述文件
+        :param seasoninfo: TMDB季媒体信息
+        :param season: 季号
+        :param season_path: 电视剧季的目录
+        """
+        logger.info(f"正在生成季NFO文件：{season_path.name}")
+        doc = minidom.Document()
+        root = DomUtils.add_node(doc, doc, "season")
+        # 添加时间
+        DomUtils.add_node(doc, root, "dateadded", time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time())))
+        # 简介
+        xplot = DomUtils.add_node(doc, root, "plot")
+        xplot.appendChild(doc.createCDATASection(seasoninfo.get("overview") or ""))
+        xoutline = DomUtils.add_node(doc, root, "outline")
+        xoutline.appendChild(doc.createCDATASection(seasoninfo.get("overview") or ""))
+        # 标题
+        DomUtils.add_node(doc, root, "title", "季 %s" % season)
+        # 发行日期
+        DomUtils.add_node(doc, root, "premiered", seasoninfo.get("air_date") or "")
+        DomUtils.add_node(doc, root, "releasedate", seasoninfo.get("air_date") or "")
+        # 发行年份
+        DomUtils.add_node(doc, root, "year", seasoninfo.get("air_date")[:4] if seasoninfo.get("air_date") else "")
+        # seasonnumber
+        DomUtils.add_node(doc, root, "seasonnumber", str(season))
+        # 保存
+        self.__save_nfo(doc, season_path.joinpath("season.nfo"))
+
+    def __gen_tv_episode_nfo_file(self,
+                                  episodeinfo: dict,
+                                  season: int,
+                                  episode: int,
+                                  file_path: Path):
+        """
+        生成电视剧集的NFO描述文件
+        :param episodeinfo: 集TMDB元数据
+        :param season: 季号
+        :param episode: 集号
+        :param file_path: 集文件的路径
+        """
+        # 开始生成集的信息
+        logger.info(f"正在生成剧集NFO文件：{file_path.name}")
+        doc = minidom.Document()
+        root = DomUtils.add_node(doc, doc, "episodedetails")
+        # 添加时间
+        DomUtils.add_node(doc, root, "dateadded", time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time())))
+        # TMDBID
+        uniqueid = DomUtils.add_node(doc, root, "uniqueid", episodeinfo.get("id") or "")
+        uniqueid.setAttribute("type", "tmdb")
+        uniqueid.setAttribute("default", "true")
+        # tmdbid
+        DomUtils.add_node(doc, root, "tmdbid", episodeinfo.get("id") or "")
+        # 标题
+        DomUtils.add_node(doc, root, "title", episodeinfo.get("name") or "第 %s 集" % episode)
+        # 简介
+        xplot = DomUtils.add_node(doc, root, "plot")
+        xplot.appendChild(doc.createCDATASection(episodeinfo.get("overview") or ""))
+        xoutline = DomUtils.add_node(doc, root, "outline")
+        xoutline.appendChild(doc.createCDATASection(episodeinfo.get("overview") or ""))
+        # 发布日期
+        DomUtils.add_node(doc, root, "aired", episodeinfo.get("air_date") or "")
+        # 年份
+        DomUtils.add_node(doc, root, "year",
+                          episodeinfo.get("air_date")[:4] if episodeinfo.get("air_date") else "")
+        # 季
+        DomUtils.add_node(doc, root, "season", str(season))
+        # 集
+        DomUtils.add_node(doc, root, "episode", str(episode))
+        # 评分
+        DomUtils.add_node(doc, root, "rating", episodeinfo.get("vote_average") or "0")
+        # 导演
+        directors = episodeinfo.get("crew") or []
+        for director in directors:
+            if director.get("known_for_department") == "Directing":
+                xdirector = DomUtils.add_node(doc, root, "director", director.get("name") or "")
+                xdirector.setAttribute("tmdbid", str(director.get("id") or ""))
+        # 演员
+        actors = episodeinfo.get("guest_stars") or []
+        for actor in actors:
+            if actor.get("known_for_department") == "Acting":
+                xactor = DomUtils.add_node(doc, root, "actor")
+                DomUtils.add_node(doc, xactor, "name", actor.get("name") or "")
+                DomUtils.add_node(doc, xactor, "type", "Actor")
+                DomUtils.add_node(doc, xactor, "tmdbid", actor.get("id") or "")
+        # 保存文件
+        self.__save_nfo(doc, file_path.with_suffix(".nfo"))
+
+    @staticmethod
+    def __save_image(url: str, file_path: Path):
+        """
+        下载图片并保存
+        """
+        if file_path.exists():
+            return
+        try:
+            logger.info(f"正在下载{file_path.stem}图片：{url} ...")
+            r = RequestUtils().get_res(url=url)
+            if r:
+                file_path.write_bytes(r.content)
+                logger.info(f"图片已保存：{file_path}")
+            else:
+                logger.info(f"{file_path.stem}图片下载失败，请检查网络连通性")
+        except Exception as err:
+            logger.error(f"{file_path.stem}图片下载失败：{err}")
+
+    @staticmethod
+    def __save_nfo(doc, file_path: Path):
+        """
+        保存NFO
+        """
+        if file_path.exists():
+            return
+        xml_str = doc.toprettyxml(indent="  ", encoding="utf-8")
+        file_path.write_bytes(xml_str)
+        logger.info(f"NFO文件已保存：{file_path}")

app/schemas/context.py

@@ -7,6 +7,9 @@ from app.schemas.types import MediaType
 
 
 class MetaInfo(BaseModel):
+    """
+    识别元数据
+    """
     # 是否处理的文件
     isfile: bool = False
     # 原字符串
@@ -50,6 +53,9 @@ class MetaInfo(BaseModel):
 
 
 class MediaInfo(BaseModel):
+    """
+    识别媒体信息
+    """
     # 类型 电影、电视剧
     type: Optional[str] = None
     # 媒体标题
@@ -118,6 +124,9 @@ class MediaInfo(BaseModel):
 
 
 class TorrentInfo(BaseModel):
+    """
+    搜索种子信息
+    """
     # 站点ID
     site: Optional[int] = None
     # 站点名称
@@ -165,6 +174,9 @@ class TorrentInfo(BaseModel):
 
 
 class Context(BaseModel):
+    """
+    上下文
+    """
     # 元数据
     meta_info: Optional[MetaInfo]
     # 媒体信息
@@ -174,6 +186,9 @@ class Context(BaseModel):
 
 
 class TransferTorrent(BaseModel):
+    """
+    待转移任务信息
+    """
     title: Optional[str] = None
     path: Optional[Path] = None
     hash: Optional[str] = None
@@ -181,6 +196,9 @@ class TransferTorrent(BaseModel):
 
 
 class DownloadingTorrent(BaseModel):
+    """
+    下载中任务信息
+    """
     hash: Optional[str] = None
     title: Optional[str] = None
     name: Optional[str] = None
@@ -195,6 +213,9 @@ class DownloadingTorrent(BaseModel):
 
 
 class TransferInfo(BaseModel):
+    """
+    文件转移结果信息
+    """
     # 转移⼁路径
     path: Optional[Path] = None
     # 转移后路径
@@ -210,6 +231,9 @@ class TransferInfo(BaseModel):
 
 
 class ExistMediaInfo(BaseModel):
+    """
+    媒体服务器存在媒体信息
+    """
     # 类型 电影、电视剧
     type: MediaType
     # 季
@@ -217,6 +241,9 @@ class ExistMediaInfo(BaseModel):
 
 
 class NotExistMediaInfo(BaseModel):
+    """
+    媒体服务器不存在媒体信息
+    """
     # 季
     season: int
     # 剧集列表
@@ -228,6 +255,9 @@ class NotExistMediaInfo(BaseModel):
 
 
 class RefreshMediaItem(BaseModel):
+    """
+    媒体库刷新信息
+    """
     # 标题
     title: str
     # 年份
@@ -238,3 +268,32 @@ class RefreshMediaItem(BaseModel):
     category: Optional[str] = None
     # 目录
     target_path: Optional[Path] = None
+
+
+class TmdbSeason(BaseModel):
+    """
+    TMDB季信息
+    """
+    air_date: Optional[str] = None
+    episode_count: Optional[int] = None
+    name: Optional[str] = None
+    overview: Optional[str] = None
+    poster_path: Optional[str] = None
+    season_number: Optional[int] = None
+    vote_average: Optional[float] = None
+
+
+class TmdbEpisode(BaseModel):
+    """
+    TMDB集信息
+    """
+    air_date: Optional[str] = None
+    episode_number: Optional[int] = None
+    name: Optional[str] = None
+    overview: Optional[str] = None
+    runtime: Optional[int] = None
+    season_number: Optional[int] = None
+    still_path: Optional[str] = None
+    vote_average: Optional[float] = None
+    crew: Optional[list] = []
+    guest_stars: Optional[list] = []