mirror of
https://github.com/shadoll/moma.git
synced 2026-08-28 11:33:25 +00:00
sha
305dd5f43e
- Updated `pyproject.toml` to reflect new package structure. - Created `renamer/__init__.py` to initialize the package. - Implemented `RenamerApp` in `renamer/app.py` for the main application interface. - Added constants for video extensions in `renamer/constants.py`. - Developed `MediaExtractor` class in `renamer/extractor.py` for extracting metadata from media files. - Created various extractor classes in `renamer/extractors/` for handling filename, metadata, and media info extraction. - Added formatting classes in `renamer/formatters/` for displaying media information and proposed filenames. - Implemented utility functions in `renamer/utils.py` for detecting file types and extracting media track information. - Introduced `OpenScreen` in `renamer/screens.py` for user input of directory paths. - Enhanced error handling and user feedback throughout the application.
59 lines
2.1 KiB
Python
59 lines
2.1 KiB
Python
import re
|
|
from pathlib import Path
|
|
from ..constants import SOURCE_DICT
|
|
|
|
|
|
class FilenameExtractor:
|
|
"""Class to extract information from filename"""
|
|
|
|
@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_resolution(file_path: Path) -> str | None:
|
|
"""Extract resolution from filename (e.g., 2160p, 1080p, 720p)"""
|
|
file_name = file_path.name
|
|
match = re.search(r'(\d{3,4})[pi]', file_name, re.IGNORECASE)
|
|
if match:
|
|
height = int(match.group(1))
|
|
if height >= 2160:
|
|
return '2160p'
|
|
elif height >= 1080:
|
|
return '1080p'
|
|
elif height >= 720:
|
|
return '720p'
|
|
elif height >= 480:
|
|
return '480p'
|
|
else:
|
|
return f'{height}p'
|
|
return None |