William Kray
60 lines
1.9 KiB
Python
60 lines
1.9 KiB
Python
"""Decorators for common bot operations."""
|
|
|
|
import functools
|
|
from typing import Callable, Any, Optional
|
|
from mautrix.types import UserID, MessageEvent
|
|
|
|
|
|
def require_permission(min_level: int = 50, room_id: Optional[str] = None):
|
|
"""Decorator to require user permission for command execution.
|
|
|
|
Args:
|
|
min_level: Minimum required power level (default 50 for moderator)
|
|
room_id: Room ID to check permissions in (None for parent room)
|
|
"""
|
|
|
|
def decorator(func: Callable) -> Callable:
|
|
@functools.wraps(func)
|
|
async def wrapper(self, evt: MessageEvent, *args, **kwargs) -> Any:
|
|
if not await self.user_permitted(evt.sender, min_level, room_id):
|
|
await evt.reply("You don't have permission to use this command")
|
|
return
|
|
return await func(self, evt, *args, **kwargs)
|
|
|
|
return wrapper
|
|
|
|
return decorator
|
|
|
|
|
|
def require_parent_room(func: Callable) -> Callable:
|
|
"""Decorator to require parent room to be configured."""
|
|
|
|
@functools.wraps(func)
|
|
async def wrapper(self, evt: MessageEvent, *args, **kwargs) -> Any:
|
|
if not await self.check_parent_room(evt):
|
|
return
|
|
return await func(self, evt, *args, **kwargs)
|
|
|
|
return wrapper
|
|
|
|
|
|
def handle_errors(error_message: str = "An error occurred"):
|
|
"""Decorator to handle common errors in command execution.
|
|
|
|
Args:
|
|
error_message: Default error message to show to user
|
|
"""
|
|
|
|
def decorator(func: Callable) -> Callable:
|
|
@functools.wraps(func)
|
|
async def wrapper(self, evt: MessageEvent, *args, **kwargs) -> Any:
|
|
try:
|
|
return await func(self, evt, *args, **kwargs)
|
|
except Exception as e:
|
|
self.log.error(f"Error in {func.__name__}: {e}")
|
|
await evt.reply(f"{error_message}: {e}")
|
|
|
|
return wrapper
|
|
|
|
return decorator
|