sbcrumb
97a9f3431a
Local Docker Build (Dev) / build-dev (push) Successful in 35s
moving to an all DB approach since radarr overwrites .nfo files
86 lines
2.2 KiB
Python
86 lines
2.2 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
IMDb ID Extraction Utilities
|
|
Parses IMDb IDs from directory/file names (no file I/O)
|
|
Phase 3: Replaces NFOManager for IMDb ID extraction only
|
|
"""
|
|
import re
|
|
from pathlib import Path
|
|
from typing import Optional
|
|
|
|
|
|
def parse_imdb_from_path(path: Path) -> Optional[str]:
|
|
"""
|
|
Extract IMDb ID from directory path or filename using regex patterns.
|
|
Does NOT read any files - only parses the path string.
|
|
|
|
Supported patterns:
|
|
- [imdb-tt1234567]
|
|
- [tt1234567]
|
|
- {imdb-tt1234567}
|
|
- (imdb-tt1234567)
|
|
- -tt1234567 (at end)
|
|
- _tt1234567 (at end)
|
|
|
|
Args:
|
|
path: Path object to parse
|
|
|
|
Returns:
|
|
IMDb ID (e.g., "tt1234567") or None if not found
|
|
"""
|
|
path_str = str(path).lower()
|
|
|
|
# Try [imdb-ttXXXXXXX] format first (most explicit)
|
|
match = re.search(r'\[imdb-?(tt\d+)\]', path_str)
|
|
if match:
|
|
return match.group(1)
|
|
|
|
# Try standalone [ttXXXXXXX] format in brackets
|
|
match = re.search(r'\[(tt\d+)\]', path_str)
|
|
if match:
|
|
return match.group(1)
|
|
|
|
# Try {imdb-ttXXXXXXX} format with curly braces
|
|
match = re.search(r'\{imdb-?(tt\d+)\}', path_str)
|
|
if match:
|
|
return match.group(1)
|
|
|
|
# Try (imdb-ttXXXXXXX) format with parentheses
|
|
match = re.search(r'\(imdb-?(tt\d+)\)', path_str)
|
|
if match:
|
|
return match.group(1)
|
|
|
|
# Try ttXXXXXXX at end of filename/dirname (common pattern)
|
|
match = re.search(r'[-_\s](tt\d+)$', path_str)
|
|
if match:
|
|
return match.group(1)
|
|
|
|
return None
|
|
|
|
|
|
def find_imdb_in_directory(directory: Path) -> Optional[str]:
|
|
"""
|
|
Find IMDb ID from directory name or filenames within the directory.
|
|
Does NOT read file contents - only checks filenames.
|
|
|
|
Args:
|
|
directory: Directory path to search
|
|
|
|
Returns:
|
|
IMDb ID or None if not found
|
|
"""
|
|
# First try directory name itself
|
|
imdb_id = parse_imdb_from_path(directory)
|
|
if imdb_id:
|
|
return imdb_id
|
|
|
|
# Try all filenames in the directory
|
|
if directory.is_dir():
|
|
for file_path in directory.iterdir():
|
|
if file_path.is_file():
|
|
imdb_id = parse_imdb_from_path(file_path)
|
|
if imdb_id:
|
|
return imdb_id
|
|
|
|
return None
|