mirror of
https://github.com/aiogram/aiogram.git
synced 2026-04-08 16:37:47 +00:00
40 lines
973 B
Python
40 lines
973 B
Python
from abc import ABC, abstractmethod
|
|
from typing import TYPE_CHECKING, Any, Dict, Generic, TypeVar, cast
|
|
|
|
from aiogram import Bot
|
|
from aiogram.api.types import Update
|
|
|
|
T = TypeVar("T")
|
|
|
|
|
|
class BaseHandlerMixin(Generic[T]):
|
|
if TYPE_CHECKING: # pragma: no cover
|
|
event: T
|
|
data: Dict[str, Any]
|
|
|
|
|
|
class BaseHandler(BaseHandlerMixin[T], ABC):
|
|
"""
|
|
Base class for all class-based handlers
|
|
"""
|
|
|
|
def __init__(self, event: T, **kwargs: Any) -> None:
|
|
self.event: T = event
|
|
self.data: Dict[str, Any] = kwargs
|
|
|
|
@property
|
|
def bot(self) -> Bot:
|
|
if "bot" in self.data:
|
|
return cast(Bot, self.data["bot"])
|
|
return Bot.get_current(no_error=False)
|
|
|
|
@property
|
|
def update(self) -> Update:
|
|
return cast(Update, self.data["update"])
|
|
|
|
@abstractmethod
|
|
async def handle(self) -> Any: # pragma: no cover
|
|
pass
|
|
|
|
def __await__(self) -> Any:
|
|
return self.handle().__await__()
|