mirror of
https://github.com/shadoll/moma.git
synced 2026-08-28 03:27:34 +00:00
113 lines
3.5 KiB
Python
113 lines
3.5 KiB
Python
"""Text formatting decorators.
|
|
|
|
Provides decorator versions of TextFormatter methods:
|
|
|
|
@text_decorators.bold()
|
|
def get_title(self):
|
|
return self.title
|
|
"""
|
|
|
|
from functools import wraps
|
|
from typing import Callable
|
|
from .text_formatter import TextFormatter
|
|
|
|
|
|
class TextDecorators:
|
|
"""Text styling and color decorators."""
|
|
|
|
@staticmethod
|
|
def bold() -> Callable:
|
|
"""Decorator to make text bold."""
|
|
def decorator(func: Callable) -> Callable:
|
|
@wraps(func)
|
|
def wrapper(*args, **kwargs) -> str:
|
|
result = func(*args, **kwargs)
|
|
if result == "":
|
|
return ""
|
|
return TextFormatter.bold(str(result))
|
|
return wrapper
|
|
return decorator
|
|
|
|
@staticmethod
|
|
def italic() -> Callable:
|
|
"""Decorator to make text italic."""
|
|
def decorator(func: Callable) -> Callable:
|
|
@wraps(func)
|
|
def wrapper(*args, **kwargs) -> str:
|
|
result = func(*args, **kwargs)
|
|
if result == "":
|
|
return ""
|
|
return TextFormatter.italic(str(result))
|
|
return wrapper
|
|
return decorator
|
|
|
|
@staticmethod
|
|
def colour(name) -> Callable:
|
|
"""Decorator to colour text."""
|
|
def decorator(func: Callable) -> Callable:
|
|
@wraps(func)
|
|
def wrapper(*args, **kwargs) -> str:
|
|
result = func(*args, **kwargs)
|
|
if not result:
|
|
return ""
|
|
return TextFormatter.colour(name, str(result))
|
|
return wrapper
|
|
return decorator
|
|
|
|
@staticmethod
|
|
def uppercase() -> Callable:
|
|
"""Decorator to convert text to uppercase."""
|
|
def decorator(func: Callable) -> Callable:
|
|
@wraps(func)
|
|
def wrapper(*args, **kwargs) -> str:
|
|
result = func(*args, **kwargs)
|
|
if not result:
|
|
return ""
|
|
return TextFormatter.uppercase(str(result))
|
|
return wrapper
|
|
return decorator
|
|
|
|
@staticmethod
|
|
def lowercase() -> Callable:
|
|
"""Decorator to convert text to lowercase."""
|
|
def decorator(func: Callable) -> Callable:
|
|
@wraps(func)
|
|
def wrapper(*args, **kwargs) -> str:
|
|
result = func(*args, **kwargs)
|
|
if not result:
|
|
return ""
|
|
return TextFormatter.lowercase(str(result))
|
|
return wrapper
|
|
return decorator
|
|
|
|
@staticmethod
|
|
def url() -> Callable:
|
|
"""Decorator to format text as a clickable URL."""
|
|
def decorator(func: Callable) -> Callable:
|
|
@wraps(func)
|
|
def wrapper(*args, **kwargs) -> str:
|
|
result = func(*args, **kwargs)
|
|
if not result:
|
|
return ""
|
|
return TextFormatter.format_url(str(result))
|
|
return wrapper
|
|
return decorator
|
|
|
|
@staticmethod
|
|
def escape() -> Callable:
|
|
"""Decorator to escape rich markup in text."""
|
|
def decorator(func: Callable) -> Callable:
|
|
@wraps(func)
|
|
def wrapper(*args, **kwargs) -> str:
|
|
from rich.markup import escape
|
|
result = func(*args, **kwargs)
|
|
if not result:
|
|
return ""
|
|
return escape(str(result))
|
|
return wrapper
|
|
return decorator
|
|
|
|
|
|
# Singleton instance
|
|
text_decorators = TextDecorators()
|