mirror of
https://github.com/EstrellaXD/Auto_Bangumi.git
synced 2026-02-14 07:46:05 +08:00
103 lines
3.6 KiB
Python
103 lines
3.6 KiB
Python
import re
|
||
import logging
|
||
from dataclasses import dataclass
|
||
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
@dataclass
|
||
class DownloadInfo:
|
||
name: str
|
||
season: int
|
||
suffix: str
|
||
file_name: str
|
||
folder_name: str
|
||
|
||
|
||
class DownloadParser:
|
||
def __init__(self):
|
||
self.rules = [
|
||
r"(.*) - (\d{1,4}|\d{1,4}\.\d{1,2})(?:v\d{1,2})?(?: )?(?:END)?(.*)",
|
||
r"(.*)[\[ E](\d{1,3}|\d{1,3}\.\d{1,2})(?:v\d{1,2})?(?: )?(?:END)?[\] ](.*)",
|
||
r"(.*)\[第(\d*\.*\d*)话(?:END)?\](.*)",
|
||
r"(.*)\[第(\d*\.*\d*)話(?:END)?\](.*)",
|
||
r"(.*)第(\d*\.*\d*)话(?:END)?(.*)",
|
||
r"(.*)第(\d*\.*\d*)話(?:END)?(.*)",
|
||
]
|
||
|
||
@staticmethod
|
||
def rename_init(name, folder_name, season, suffix) -> DownloadInfo:
|
||
n = re.split(r"[\[\]()【】（）]", name)
|
||
suffix = suffix if suffix is not None else n[-1]
|
||
file_name = name.replace(f"[{n[1]}]", "")
|
||
if season < 10:
|
||
season = f"0{season}"
|
||
return DownloadInfo(name, season, suffix, file_name, folder_name)
|
||
|
||
def rename_normal(self, info: DownloadInfo):
|
||
for rule in self.rules:
|
||
match_obj = re.match(rule, info.name, re.I)
|
||
if match_obj is not None:
|
||
title = re.sub(r"([Ss]|Season )\d{1,3}", "", match_obj.group(1)).strip()
|
||
new_name = f"{title} S{info.season}E{match_obj.group(2)}{match_obj.group(3)}"
|
||
return new_name
|
||
|
||
def rename_pn(self, info: DownloadInfo):
|
||
for rule in self.rules:
|
||
match_obj = re.match(rule, info.file_name, re.I)
|
||
if match_obj is not None:
|
||
title = re.sub(r"([Ss]|Season )\d{1,3}", "", match_obj.group(1)).strip()
|
||
title = title if title != "" else info.folder_name
|
||
new_name = re.sub(
|
||
r"[\[\]]",
|
||
"",
|
||
f"{title} S{info.season}E{match_obj.group(2)}{info.suffix}",
|
||
)
|
||
return new_name
|
||
|
||
def rename_advance(self, info: DownloadInfo):
|
||
for rule in self.rules:
|
||
match_obj = re.match(rule, info.file_name, re.I)
|
||
if match_obj is not None:
|
||
new_name = re.sub(
|
||
r"[\[\]]",
|
||
"",
|
||
f"{info.folder_name} S{info.season}E{match_obj.group(2)}{info.suffix}",
|
||
)
|
||
return new_name
|
||
|
||
def rename_no_season_pn(self, info: DownloadInfo):
|
||
for rule in self.rules:
|
||
match_obj = re.match(rule, info.file_name, re.I)
|
||
if match_obj is not None:
|
||
title = match_obj.group(1).strip()
|
||
new_name = re.sub(
|
||
r"[\[\]]",
|
||
"",
|
||
f"{title} E{match_obj.group(2)}{info.suffix}",
|
||
)
|
||
return new_name
|
||
|
||
@staticmethod
|
||
def rename_none(info: DownloadInfo):
|
||
return info.name
|
||
|
||
def download_rename(self, name, folder_name, season, suffix, method):
|
||
rename_info = self.rename_init(name, folder_name, season, suffix)
|
||
method_dict = {
|
||
"normal": self.rename_normal,
|
||
"pn": self.rename_pn,
|
||
"advance": self.rename_advance,
|
||
"no_season_pn": self.rename_no_season_pn,
|
||
"none": self.rename_none
|
||
}
|
||
return method_dict[method.lower()](rename_info)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
name = "[Lilith-Raws] Tate no Yuusha no Nariagari S02 - 02 [Baha][WEB-DL][1080p][AVC AAC][CHT][MP4]"
|
||
rename = DownloadParser()
|
||
new_name = rename.download_rename(name, "异世界舅舅（2022）", 1, ".mp4", "normal")
|
||
print(new_name)
|