moma/renamer/extractors/filename_extractor.py

import re
from pathlib import Path
from ..constants import SOURCE_DICT, FRAME_CLASSES


class FilenameExtractor:
    """Class to extract information from filename"""

    @staticmethod
    def _get_frame_class_from_height(height: int) -> str:
        """Get frame class from video height using FRAME_CLASSES constant"""
        for frame_class, info in FRAME_CLASSES.items():
            if height == info['nominal_height']:
                return frame_class
        return 'Unclassified'

    @staticmethod
    def extract_title(file_path: Path) -> str | None:
        """Extract movie title from filename"""
        file_name = file_path.name
        temp_name = re.sub(r'\s*\(\d{4}\)\s*|\s*\d{4}\s*|\.\d{4}\.', '', file_name)

        # Find and remove source
        source = FilenameExtractor.extract_source(file_path)
        if source:
            for alias in SOURCE_DICT[source]:
                temp_name = re.sub(r'\b' + re.escape(alias) + r'\b', '', temp_name, flags=re.IGNORECASE)

        return temp_name.rsplit('.', 1)[0].strip()

    @staticmethod
    def extract_year(file_path: Path) -> str | None:
        """Extract year from filename"""
        file_name = file_path.name
        year_match = re.search(r'\((\d{4})\)|(\d{4})', file_name)
        return (year_match.group(1) or year_match.group(2)) if year_match else None

    @staticmethod
    def extract_source(file_path: Path) -> str | None:
        """Extract video source from filename"""
        file_name = file_path.name
        temp_name = re.sub(r'\s*\(\d{4}\)\s*|\s*\d{4}\s*|\.\d{4}\.', '', file_name)

        for src, aliases in SOURCE_DICT.items():
            for alias in aliases:
                if re.search(r'\b' + re.escape(alias) + r'\b', temp_name, re.IGNORECASE):
                    return src
        return None

    @staticmethod
    def extract_frame_class(file_path: Path) -> str | None:
        """Extract frame class from filename (480p, 720p, 1080p, 2160p, etc.)"""
        file_name = file_path.name
        match = re.search(r'(\d{3,4})[pi]', file_name, re.IGNORECASE)
        if match:
            height = int(match.group(1))
            return FilenameExtractor._get_frame_class_from_height(height)
        return 'Unclassified'