chore: move Auto_Bangumi/src -> Auto_Bangumi/backend/src, prepare for merge WebUI repo

2026-04-28 12:30:20 +08:00 · 2023-06-06 22:33:00 +08:00
parent f8411f9c8a
commit 069a86faa0
99 changed files with 179 additions and 0 deletions
--- a/backend/src/module/parser/title_parser.py
+++ b/backend/src/module/parser/title_parser.py
@@ -0,0 +1,73 @@
+import logging
+
+from module.conf import settings
+from module.models import BangumiData
+
+from .analyser import raw_parser, tmdb_parser, torrent_parser
+
+logger = logging.getLogger(__name__)
+
+
+class TitleParser:
+    def __init__(self):
+        pass
+
+    @staticmethod
+    def torrent_parser(
+        torrent_path: str,
+        torrent_name: str | None = None,
+        season: int | None = None,
+        file_type: str = "media",
+    ):
+        try:
+            return torrent_parser(torrent_path, torrent_name, season, file_type)
+        except Exception as e:
+            logger.warning(f"Cannot parse {torrent_path} with error {e}")
+
+    @staticmethod
+    def tmdb_parser(title: str, season: int, language: str):
+        official_title, tmdb_season, year = title, season, None
+        tmdb_info = tmdb_parser(title, language)
+        if tmdb_info:
+            logger.debug(f"TMDB Matched, official title is {tmdb_info.title}")
+            tmdb_season = tmdb_info.last_season if tmdb_info.last_season else season
+            official_title = tmdb_info.title
+            year = tmdb_info.year
+        else:
+            logger.warning(f"Cannot match {title} in TMDB. Use raw title instead.")
+            logger.warning("Please change bangumi info manually.")
+        return official_title, tmdb_season, year
+
+    @staticmethod
+    def raw_parser(raw: str, rss_link: str) -> BangumiData | None:
+        language = settings.rss_parser.language
+        try:
+            episode = raw_parser(raw)
+            titles = {
+                "zh": episode.title_zh,
+                "en": episode.title_en,
+                "jp": episode.title_jp,
+            }
+            title_raw = episode.title_en if episode.title_en else episode.title_zh
+            official_title = titles[language] if titles[language] else titles["zh"]
+            _season = episode.season
+            data = BangumiData(
+                official_title=official_title,
+                title_raw=title_raw,
+                season=_season,
+                season_raw=episode.season_raw,
+                group_name=episode.group,
+                dpi=episode.resolution,
+                source=episode.source,
+                subtitle=episode.sub,
+                eps_collect=False if episode.episode > 1 else True,
+                offset=0,
+                filter=settings.rss_parser.filter,
+                rss_link=[rss_link],
+            )
+            logger.debug(f"RAW:{raw} >> {title_raw}")
+            return data
+        except Exception as e:
+            logger.debug(e)
+            logger.warning(f"Cannot parse {raw}.")
+            return None