MoviePilot/app/core/meta_info.py
2023-06-06 07:15:17 +08:00

44 lines
1.3 KiB
Python

from pathlib import Path
import regex as re
from app.core.config import settings
from app.core.meta import MetaAnime, MetaVideo
def MetaInfo(title: str, subtitle: str = None):
"""
媒体整理入口,根据名称和副标题,判断是哪种类型的识别,返回对应对象
:param title: 标题、种子名、文件名
:param subtitle: 副标题、描述
:return: MetaAnime、MetaVideo
"""
# 判断是否处理文件
if title and Path(title).suffix.lower() in settings.RMT_MEDIAEXT:
isfile = True
else:
isfile = False
return MetaAnime(title, subtitle, isfile) if is_anime(title) else MetaVideo(title, subtitle, isfile)
def is_anime(name: str):
"""
判断是否为动漫
:param name: 名称
:return: 是否动漫
"""
if not name:
return False
if re.search(r'【[+0-9XVPI-]+】\s*【', name, re.IGNORECASE):
return True
if re.search(r'\s+-\s+[\dv]{1,4}\s+', name, re.IGNORECASE):
return True
if re.search(r"S\d{2}\s*-\s*S\d{2}|S\d{2}|\s+S\d{1,2}|EP?\d{2,4}\s*-\s*EP?\d{2,4}|EP?\d{2,4}|\s+EP?\d{1,4}", name,
re.IGNORECASE):
return False
if re.search(r'\[[+0-9XVPI-]+]\s*\[', name, re.IGNORECASE):
return True
return False