mirror of
https://github.com/shadoll/moma.git
synced 2026-08-28 11:33:25 +00:00
feat: restructure renamer package and implement media extraction features
- 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.
This commit is contained in:
@@ -0,0 +1,59 @@
|
||||
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
|
||||
Reference in New Issue
Block a user