refactor using custom callback context

This commit is contained in:
2024-08-13 23:20:24 +08:00
parent 03066853de
commit 733adfc4a6
2 changed files with 68 additions and 44 deletions
+29
View File
@@ -0,0 +1,29 @@
import dataclasses
from typing import Optional
from telegram import Message
from telegram.ext import Application, CallbackContext, ExtBot
@dataclasses.dataclass
class EditMessage:
url: str
forward: tuple[Message, ...]
class ChatData:
def __init__(self):
self.forward_channel_id: Optional[int] = None
self.edit_before_forward: bool = False
self.edit_message: dict[int, EditMessage] = {}
self.template: Optional[str] = None
class CustomContext(CallbackContext[ExtBot, dict, ChatData, dict]):
def __init__(
self,
application: Application,
chat_id: Optional[int] = None,
user_id: Optional[int] = None
):
super().__init__(application=application, chat_id=chat_id, user_id=user_id)