francwa
249c5de76a
- Refactor memory system (episodic/STM/LTM with components) - Implement complete subtitle domain (scanner, matcher, placer) - Add YAML workflow infrastructure - Externalize knowledge base (patterns, release groups) - Add comprehensive testing suite - Create manual testing CLIs
94 lines
2.7 KiB
Python
94 lines
2.7 KiB
Python
"""SubtitlePlacer — hard-links matched subtitle tracks next to the destination video."""
|
|
|
|
import logging
|
|
import os
|
|
from dataclasses import dataclass
|
|
from pathlib import Path
|
|
|
|
from ..entities import SubtitleTrack
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@dataclass
|
|
class PlacedTrack:
|
|
source: Path
|
|
destination: Path
|
|
filename: str
|
|
|
|
|
|
@dataclass
|
|
class PlaceResult:
|
|
placed: list[PlacedTrack]
|
|
skipped: list[tuple[SubtitleTrack, str]] # (track, reason)
|
|
|
|
@property
|
|
def placed_count(self) -> int:
|
|
return len(self.placed)
|
|
|
|
@property
|
|
def skipped_count(self) -> int:
|
|
return len(self.skipped)
|
|
|
|
|
|
class SubtitlePlacer:
|
|
"""
|
|
Hard-links matched SubtitleTrack files next to a destination video.
|
|
|
|
Uses the same hard-link strategy as FileManager.copy_file:
|
|
instant, no data duplication, qBittorrent keeps seeding.
|
|
|
|
Embedded tracks are skipped — nothing to place on disk.
|
|
"""
|
|
|
|
def place(
|
|
self,
|
|
tracks: list[SubtitleTrack],
|
|
destination_video: Path,
|
|
) -> PlaceResult:
|
|
placed: list[PlacedTrack] = []
|
|
skipped: list[tuple[SubtitleTrack, str]] = []
|
|
|
|
dest_dir = destination_video.parent
|
|
|
|
for track in tracks:
|
|
if track.is_embedded:
|
|
logger.debug(f"SubtitlePlacer: skip embedded track ({track.language})")
|
|
skipped.append((track, "embedded — no file to place"))
|
|
continue
|
|
|
|
if not track.file_path or not track.file_path.exists():
|
|
skipped.append((track, "source file not found"))
|
|
continue
|
|
|
|
try:
|
|
dest_name = track.destination_name
|
|
except ValueError as e:
|
|
skipped.append((track, str(e)))
|
|
continue
|
|
|
|
dest_path = dest_dir / dest_name
|
|
|
|
if dest_path.exists():
|
|
logger.debug(f"SubtitlePlacer: skip {dest_name} — already exists")
|
|
skipped.append((track, "destination already exists"))
|
|
continue
|
|
|
|
try:
|
|
os.link(track.file_path, dest_path)
|
|
placed.append(PlacedTrack(
|
|
source=track.file_path,
|
|
destination=dest_path,
|
|
filename=dest_name,
|
|
))
|
|
logger.info(f"SubtitlePlacer: placed {dest_name}")
|
|
except OSError as e:
|
|
logger.warning(f"SubtitlePlacer: failed to place {dest_name}: {e}")
|
|
skipped.append((track, str(e)))
|
|
|
|
logger.info(
|
|
f"SubtitlePlacer: {len(placed)} placed, {len(skipped)} skipped "
|
|
f"for {destination_video.name}"
|
|
)
|
|
return PlaceResult(placed=placed, skipped=skipped)
|