mirror of
https://github.com/d0zingcat/MoviePilot-Plugins.git
synced 2026-05-18 15:09:29 +00:00
285 lines
11 KiB
Python
285 lines
11 KiB
Python
import concurrent
|
||
import re
|
||
from dataclasses import dataclass
|
||
from pathlib import Path
|
||
from typing import List
|
||
from app.chain.media import MediaChain
|
||
from app.chain.tmdb import TmdbChain
|
||
from app.core.metainfo import MetaInfoPath
|
||
from app.log import logger
|
||
from app.schemas import MediaType
|
||
|
||
season_patterns = [
|
||
{"pattern": re.compile(r"S(\d+)$", re.IGNORECASE), "group": 1},
|
||
{"pattern": re.compile(r"(\d+)$", re.IGNORECASE), "group": 1},
|
||
{"pattern": re.compile(r"(\d+)(st|nd|rd|th)?\s*season", re.IGNORECASE), "group": 1},
|
||
{"pattern": re.compile(r"(.*) ?\s*season (\d+)", re.IGNORECASE), "group": 2},
|
||
{"pattern": re.compile(r"\s(II|III|IV|V|VI|VII|VIII|IX|X)$", re.IGNORECASE), "group": "1"}
|
||
]
|
||
episode_patterns = [
|
||
{"pattern": re.compile(r"(\d+)\((\d+)\)", re.IGNORECASE), "group": 2},
|
||
{"pattern": re.compile(r"(\d+)", re.IGNORECASE), "group": 1},
|
||
{"pattern": re.compile(r'(\d+)v\d+', re.IGNORECASE), "group": 1},
|
||
]
|
||
|
||
ova_patterns = [
|
||
re.compile(r".*?(OVA|OAD).*?", re.IGNORECASE),
|
||
re.compile(r"\d+\.5"),
|
||
re.compile(r"00")
|
||
]
|
||
|
||
final_season_patterns = [
|
||
re.compile('final season', re.IGNORECASE),
|
||
re.compile('The Final', re.IGNORECASE),
|
||
re.compile(r'\sFinal')
|
||
]
|
||
|
||
movie_patterns = [
|
||
re.compile("Movie", re.IGNORECASE),
|
||
re.compile("the Movie", re.IGNORECASE),
|
||
]
|
||
|
||
|
||
@dataclass
|
||
class VCBMetaBase:
|
||
# 转化为小写后的原始文件名称 (不含后缀)
|
||
original_title: str = ""
|
||
# 解析后不包含季度和集数的标题
|
||
title: str = ""
|
||
# 类型:TV / Movie (默认TV)
|
||
type: str = "TV"
|
||
# 可能含有季度的标题，一级解析后的标题
|
||
season_title: str = ""
|
||
# 可能含有集数的字符串列表
|
||
ep_title: List[str] = None
|
||
# 识别出来的季度
|
||
season: int = None
|
||
# 识别出来的集数
|
||
ep: int = None
|
||
# 是否是OVA/OAD
|
||
is_ova: bool = False
|
||
# TMDB ID
|
||
tmdb_id: int = None
|
||
|
||
|
||
blocked_words = ["vcb-studio", "360p", "480p", "720p", "1080p", "2160p", "hdr", "x265", "x264", "aac", "flac"]
|
||
|
||
|
||
class ReMeta:
|
||
|
||
def __init__(self, ova_switch: bool = False, custom_season_patterns: list[dict] = None):
|
||
self.meta = None
|
||
# TODO:自定义季度匹配规则
|
||
self.custom_season_patterns = custom_season_patterns
|
||
self.season_patterns = season_patterns
|
||
self.ova_switch = ova_switch
|
||
self.vcb_meta = VCBMetaBase()
|
||
self.is_ova = False
|
||
|
||
def is_tv(self, title: str) -> bool:
|
||
"""
|
||
判断是否是TV
|
||
"""
|
||
if title.count("[") != 4 and title.count("]") != 4:
|
||
self.vcb_meta.type = "Movie"
|
||
self.vcb_meta.title = re.sub(r'\[.*?\]', '', title).strip()
|
||
return False
|
||
return True
|
||
|
||
def handel_file(self, file_path: Path):
|
||
file_name = file_path.stem.strip().lower()
|
||
self.vcb_meta.original_title = file_name
|
||
if not self.is_tv(file_name):
|
||
logger.warn(
|
||
"不符合VCB-Studio的剧集命名规范，归类为电影,跳过剧集模块处理。注意：年份较为久远的作品可能在此会判断错误")
|
||
self.parse_movie()
|
||
else:
|
||
self.tv_mode()
|
||
self.is_ova = self.vcb_meta.is_ova
|
||
meta = MetaInfoPath(file_path)
|
||
meta.title = self.vcb_meta.title
|
||
meta.en_name = self.vcb_meta.title
|
||
if self.vcb_meta.type == "Movie":
|
||
meta.type = MediaType.MOVIE
|
||
else:
|
||
meta.type = MediaType.TV
|
||
if self.vcb_meta.ep is not None:
|
||
meta.begin_episode = self.vcb_meta.ep
|
||
if self.vcb_meta.season is not None:
|
||
meta.begin_season = self.vcb_meta.season
|
||
if self.vcb_meta.tmdb_id is not None:
|
||
meta.tmdbid = self.vcb_meta.tmdb_id
|
||
return meta
|
||
|
||
def split_season_ep(self):
|
||
# 把所有的[] 里面的内容获取出来,不需要[]本身
|
||
self.vcb_meta.ep_title = re.findall(r'\[(.*?)\]', self.vcb_meta.original_title)
|
||
# 去除所有[]后只剩下剧名
|
||
self.vcb_meta.season_title = re.sub(r"\[.*?\]", "", self.vcb_meta.original_title).strip()
|
||
if self.vcb_meta.ep_title:
|
||
self.culling_blocked_words()
|
||
logger.info(
|
||
f"分离出包含可能季度的内容部分：{self.vcb_meta.season_title} | 可能包含集数的内容部分： {self.vcb_meta.ep_title}")
|
||
self.vcb_meta.title = self.vcb_meta.season_title
|
||
if not self.vcb_meta.ep_title:
|
||
self.vcb_meta.title = self.vcb_meta.season_title
|
||
logger.warn("未识别出可能存在集数位置的信息，跳过剩余识别步骤！")
|
||
|
||
def tv_mode(self):
|
||
logger.info("开始分离季度和集数部分")
|
||
self.split_season_ep()
|
||
if not self.vcb_meta.ep_title:
|
||
return
|
||
self.parse_season()
|
||
self.parse_episode()
|
||
|
||
def parse_season(self):
|
||
"""
|
||
从标题中解析季度
|
||
"""
|
||
flag = False
|
||
for pattern in season_patterns:
|
||
match = pattern["pattern"].search(self.vcb_meta.season_title)
|
||
if match:
|
||
if isinstance(pattern["group"], int):
|
||
self.vcb_meta.season = int(match.group(pattern["group"]))
|
||
else:
|
||
self.vcb_meta.season = self.roman_to_int(match.group(pattern["group"]))
|
||
# 匹配成功后，标题中去除季度信息
|
||
self.vcb_meta.title = pattern["pattern"].sub("", self.vcb_meta.season_title).strip
|
||
logger.info(f"识别出季度为{self.vcb_meta.season}")
|
||
return
|
||
logger.info(f"正常匹配季度失败，开始匹配ova/oad/最终季度")
|
||
if not flag:
|
||
# 匹配是否为最终季
|
||
for pattern in final_season_patterns:
|
||
if pattern.search(self.vcb_meta.season_title):
|
||
logger.info("命中到最终季匹配规则")
|
||
self.vcb_meta.title = pattern.sub("", self.vcb_meta.season_title).strip()
|
||
self.handle_final_season()
|
||
return
|
||
logger.info("未识别出最终季度，开始匹配OVA/OAD")
|
||
# 匹配是否为OVA/OAD
|
||
if "ova" in self.vcb_meta.season_title or "oad" in self.vcb_meta.season_title:
|
||
logger.info("季度部分命中到OVA/OAD匹配规则")
|
||
if self.ova_switch:
|
||
logger.info("开启OVA/OAD处理逻辑")
|
||
self.vcb_meta.is_ova = True
|
||
for pattern in ova_patterns:
|
||
if pattern.search(self.vcb_meta.season_title):
|
||
self.vcb_meta.title = pattern.sub("", self.vcb_meta.season_title).strip()
|
||
self.vcb_meta.title = re.sub("ova|oad", "", self.vcb_meta.season_title).strip()
|
||
self.vcb_meta.season = 0
|
||
return
|
||
logger.warn("未识别出季度,默认处理逻辑返回第一季")
|
||
self.vcb_meta.title = self.vcb_meta.season_title
|
||
self.vcb_meta.season = 1
|
||
|
||
def parse_episode(self):
|
||
"""
|
||
从标题中解析集数
|
||
"""
|
||
# 从ep_title中剔除不相关的内容之后只剩下存在集数的字符串
|
||
ep = self.vcb_meta.ep_title[0]
|
||
for pattern in episode_patterns:
|
||
match = pattern["pattern"].search(ep)
|
||
if match:
|
||
self.vcb_meta.ep = int(match.group(pattern["group"]))
|
||
logger.info(f"识别出集数为{self.vcb_meta.ep}")
|
||
return
|
||
# 直接进入判断是否为OVA/OAD
|
||
for pattern in ova_patterns:
|
||
if pattern.search(ep):
|
||
self.vcb_meta.is_ova = True
|
||
# 直接获取数字
|
||
self.vcb_meta.ep = int(re.search(r"\d+", ep).group()) or 1
|
||
logger.info(f"OVA模式下识别出集数为{self.vcb_meta.ep}")
|
||
self.vcb_meta.season = 0
|
||
return
|
||
|
||
def culling_blocked_words(self):
|
||
"""
|
||
从ep_title中剔除不相关的内容
|
||
"""
|
||
blocked_set = set(blocked_words) # 将阻止词列表转换为集合
|
||
result = [ep for ep in self.vcb_meta.ep_title if not any(word in ep for word in blocked_set)]
|
||
self.vcb_meta.ep_title = result
|
||
|
||
def handle_final_season(self):
|
||
|
||
_, medias = MediaChain().search(title=self.vcb_meta.title)
|
||
if not medias:
|
||
logger.warning("匹配到最终季时无法找到对应的媒体信息！季度返回默认值：1")
|
||
self.vcb_meta.season = 1
|
||
return
|
||
|
||
filter_medias = [media for media in medias if media.type == MediaType.TV]
|
||
if not filter_medias:
|
||
logger.warning("匹配到最终季时无法找到对应的媒体信息！季度返回默认值：1")
|
||
self.vcb_meta.season = 1
|
||
return
|
||
medias = [media for media in filter_medias if media.popularity or media.vote_average]
|
||
if not medias:
|
||
logger.warning("匹配到最终季时无法找到对应的媒体信息！季度返回默认值：1")
|
||
self.vcb_meta.season = 1
|
||
return
|
||
# 获取欢迎度最高或者评分最高的媒体
|
||
medias_sorted = sorted(medias, key=lambda x: x.popularity or x.vote_average, reverse=True)[0]
|
||
self.vcb_meta.tmdb_id = medias_sorted.tmdb_id
|
||
if medias_sorted.tmdb_id:
|
||
seasons_info = TmdbChain().tmdb_seasons(tmdbid=medias_sorted.tmdb_id)
|
||
if seasons_info:
|
||
self.vcb_meta.season = len(seasons_info)
|
||
logger.info(f"获取到最终季度，季度为{self.vcb_meta.season}")
|
||
return
|
||
logger.warning("无法获取到最终季度信息，季度返回默认值：1")
|
||
self.vcb_meta.season = 1
|
||
|
||
|
||
|
||
def parse_movie(self):
|
||
logger.info("开始尝试剧场版模式解析")
|
||
for pattern in movie_patterns:
|
||
if pattern.search(self.vcb_meta.title):
|
||
logger.info("命中剧场版匹配规则,加上剧场版标识辅助识别")
|
||
self.vcb_meta.type = "Movie"
|
||
self.vcb_meta.title = pattern.sub("", self.vcb_meta.title).strip()
|
||
self.vcb_meta.title = self.vcb_meta.title
|
||
return
|
||
|
||
def find_ova_episode(self):
|
||
"""
|
||
搜索OVA的集数
|
||
TODO:模糊匹配OVA的集数
|
||
"""
|
||
pass
|
||
|
||
|
||
@staticmethod
|
||
def roman_to_int(s) -> int:
|
||
"""
|
||
:param s: 罗马数字字符串
|
||
罗马数字转整数
|
||
"""
|
||
roman_dict = {'I': 1, 'V': 5, 'X': 10, 'L': 50, 'C': 100, 'D': 500, 'M': 1000}
|
||
total = 0
|
||
prev_value = 0
|
||
|
||
for char in reversed(s): # 反向遍历罗马数字字符串
|
||
current_value = roman_dict[char]
|
||
if current_value >= prev_value:
|
||
total += current_value # 如果当前值大于等于前一个值，加上当前值
|
||
else:
|
||
total -= current_value # 如果当前值小于前一个值，减去当前值
|
||
prev_value = current_value
|
||
|
||
return total
|
||
|
||
|
||
|
||
# if __name__ == '__main__':
|
||
# ReMeta(
|
||
# ova_switch=True,
|
||
# ).handel_file(Path(
|
||
# r"[Airota&Nekomoe kissaten&VCB-Studio] Yuru Camp [Heya Camp EP00][Ma10p_1080p][x265_flac].mkv"))
|