mirror of
https://github.com/EstrellaXD/Auto_Bangumi.git
synced 2026-04-24 02:20:38 +08:00
cba4988e52
Backend: shared HTTP connection pool, concurrent RSS/torrent/notification operations, TMDB/Mikan result caching, database indexes, pre-compiled regex, __slots__ on dataclasses, O(1) set-based dedup, frozenset lookups, batch RSS enable/disable, asyncio.to_thread for blocking calls. Frontend: shallowRef for large arrays, computed table columns, watch instead of watchEffect, scoped style fix, typed emits, noImplicitAny, useIntervalFn lifecycle management, shared useClipboard instance, shallow clone for shared template objects. Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
79 lines
2.3 KiB
Python
79 lines
2.3 KiB
Python
import logging
|
|
import re
|
|
from os import PathLike
|
|
|
|
from module.conf import PLATFORM, settings
|
|
from module.models import Bangumi, BangumiUpdate
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
if PLATFORM == "Windows":
|
|
from pathlib import PureWindowsPath as Path
|
|
else:
|
|
from pathlib import Path
|
|
|
|
|
|
_MEDIA_SUFFIXES = frozenset({".mp4", ".mkv"})
|
|
_SUBTITLE_SUFFIXES = frozenset({".ass", ".srt"})
|
|
|
|
|
|
class TorrentPath:
|
|
def __init__(self):
|
|
pass
|
|
|
|
@staticmethod
|
|
def check_files(files: list[dict]):
|
|
media_list = []
|
|
subtitle_list = []
|
|
for f in files:
|
|
file_name = f["name"]
|
|
suffix = Path(file_name).suffix.lower()
|
|
if suffix in _MEDIA_SUFFIXES:
|
|
media_list.append(file_name)
|
|
elif suffix in _SUBTITLE_SUFFIXES:
|
|
subtitle_list.append(file_name)
|
|
return media_list, subtitle_list
|
|
|
|
@staticmethod
|
|
def _path_to_bangumi(save_path: PathLike[str] | str):
|
|
# Split save path and download path
|
|
save_parts = Path(save_path).parts
|
|
download_parts = Path(settings.downloader.path).parts
|
|
# Get bangumi name and season
|
|
bangumi_name = ""
|
|
season = 1
|
|
for part in save_parts:
|
|
if re.match(r"S\d+|[Ss]eason \d+", part):
|
|
season = int(re.findall(r"\d+", part)[0])
|
|
elif part not in download_parts:
|
|
bangumi_name = part
|
|
return bangumi_name, season
|
|
|
|
@staticmethod
|
|
def _file_depth(file_path: PathLike[str] | str):
|
|
return len(Path(file_path).parts)
|
|
|
|
def is_ep(self, file_path: PathLike[str] | str):
|
|
return self._file_depth(file_path) <= 2
|
|
|
|
@staticmethod
|
|
def _gen_save_path(data: Bangumi | BangumiUpdate):
|
|
folder = (
|
|
f"{data.official_title} ({data.year})" if data.year else data.official_title
|
|
)
|
|
save_path = Path(settings.downloader.path) / folder / f"Season {data.season}"
|
|
return str(save_path)
|
|
|
|
@staticmethod
|
|
def _rule_name(data: Bangumi):
|
|
rule_name = (
|
|
f"[{data.group_name}] {data.official_title} S{data.season}"
|
|
if settings.bangumi_manage.group_tag
|
|
else f"{data.official_title} S{data.season}"
|
|
)
|
|
return rule_name
|
|
|
|
@staticmethod
|
|
def _join_path(*args):
|
|
return str(Path(*args))
|