mirror of
https://github.com/shadoll/moma.git
synced 2026-08-28 19:43:28 +00:00
sha
917d25b360
- Introduced `DurationDecorators` for full and short duration formatting. - Added `ExtensionDecorators` for formatting extension information. - Created `ResolutionDecorators` for formatting resolution dimensions. - Implemented `SizeDecorators` for full and short size formatting. - Enhanced `TextDecorators` with additional formatting options including blue and grey text, URL formatting, and escaping rich markup. - Developed `TrackDecorators` for formatting video, audio, and subtitle track data. - Refactored `MediaPanelView` to utilize a new `MediaPanelProperties` class for cleaner property management and formatting. - Updated `media_panel_properties.py` to include formatted properties for file info, TMDB data, metadata extraction, media info extraction, and filename extraction. - Bumped version to 0.6.5 in `uv.lock`.
118 lines
4.2 KiB
Python
118 lines
4.2 KiB
Python
"""Conditional formatting decorators.
|
|
|
|
Provides decorators for conditional formatting (wrap, replace_slashes, default):
|
|
|
|
@conditional_decorators.wrap("[", "]")
|
|
def get_order(self):
|
|
return self.extractor.get('order')
|
|
"""
|
|
|
|
from functools import wraps
|
|
from typing import Callable, Any
|
|
|
|
|
|
class ConditionalDecorators:
|
|
"""Conditional formatting decorators (wrap, replace_slashes, default)."""
|
|
|
|
@staticmethod
|
|
def wrap(left: str, right: str = "") -> Callable:
|
|
"""Decorator to wrap value with delimiters if it exists.
|
|
|
|
Can be used for prefix-only (right=""), suffix-only (left=""), or both.
|
|
Supports format string placeholders that will be filled from function arguments.
|
|
|
|
Usage:
|
|
@conditional_decorators.wrap("[", "]")
|
|
def get_order(self):
|
|
return self.extractor.get('order')
|
|
|
|
# Prefix only
|
|
@conditional_decorators.wrap(" ")
|
|
def get_source(self):
|
|
return self.extractor.get('source')
|
|
|
|
# Suffix only
|
|
@conditional_decorators.wrap("", ",")
|
|
def get_hdr(self):
|
|
return self.extractor.get('hdr')
|
|
|
|
# With placeholders
|
|
@conditional_decorators.wrap("Track {index}: ")
|
|
def get_track(self, data, index):
|
|
return data
|
|
"""
|
|
def decorator(func: Callable) -> Callable:
|
|
@wraps(func)
|
|
def wrapper(*args, **kwargs) -> str:
|
|
result = func(*args, **kwargs)
|
|
if not result:
|
|
return ""
|
|
|
|
# Extract format arguments from function signature
|
|
# Skip 'self' (args[0]) and the main data argument
|
|
format_kwargs = {}
|
|
if len(args) > 2: # self, data, index, ...
|
|
# Try to detect named parameters from function signature
|
|
import inspect
|
|
sig = inspect.signature(func)
|
|
param_names = list(sig.parameters.keys())
|
|
# Skip first two params (self, data/track/value)
|
|
for i, param_name in enumerate(param_names[2:], start=2):
|
|
if i < len(args):
|
|
format_kwargs[param_name] = args[i]
|
|
|
|
# Also add explicit kwargs
|
|
format_kwargs.update(kwargs)
|
|
|
|
# Format left and right with available arguments
|
|
formatted_left = left.format(**format_kwargs) if format_kwargs else left
|
|
formatted_right = right.format(**format_kwargs) if format_kwargs else right
|
|
|
|
return f"{formatted_left}{result}{formatted_right}"
|
|
return wrapper
|
|
return decorator
|
|
|
|
@staticmethod
|
|
def replace_slashes() -> Callable:
|
|
"""Decorator to replace forward and back slashes with dashes.
|
|
|
|
Usage:
|
|
@conditional_decorators.replace_slashes()
|
|
def get_title(self):
|
|
return self.extractor.get('title')
|
|
"""
|
|
def decorator(func: Callable) -> Callable:
|
|
@wraps(func)
|
|
def wrapper(*args, **kwargs) -> str:
|
|
result = func(*args, **kwargs)
|
|
if result:
|
|
return str(result).replace("/", "-").replace("\\", "-")
|
|
return result or ""
|
|
return wrapper
|
|
return decorator
|
|
|
|
@staticmethod
|
|
def default(default_value: Any) -> Callable:
|
|
"""Decorator to provide a default value if result is None or empty.
|
|
|
|
NOTE: It's better to handle defaults in the extractor itself rather than
|
|
using this decorator. This decorator should only be used when the extractor
|
|
cannot provide a sensible default.
|
|
|
|
Usage:
|
|
@conditional_decorators.default("Unknown")
|
|
def get_value(self):
|
|
return self.extractor.get('value')
|
|
"""
|
|
def decorator(func: Callable) -> Callable:
|
|
@wraps(func)
|
|
def wrapper(*args, **kwargs) -> Any:
|
|
result = func(*args, **kwargs)
|
|
return result if result else default_value
|
|
return wrapper
|
|
return decorator
|
|
|
|
|
|
# Singleton instance
|
|
conditional_decorators = ConditionalDecorators()
|