From 343f407fc5177847eec1e7d9e01105e76b4d02ec Mon Sep 17 00:00:00 2001 From: Darskiy Date: Tue, 11 Mar 2025 02:49:39 +0200 Subject: [PATCH] Add type hints for `message` in `NewMessage.Event` and integrate `SenderGetter` for better IDE support Inherit from `SenderGetter` in `NewMessage.Event` to explicitly expose sender-related properties for IDE autocompletion. Annotates `self.message` with `types.Message` to further clarify type expectations. --- telethon/events/newmessage.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/telethon/events/newmessage.py b/telethon/events/newmessage.py index d2077a71..c888191c 100644 --- a/telethon/events/newmessage.py +++ b/telethon/events/newmessage.py @@ -2,7 +2,8 @@ import re from .common import EventBuilder, EventCommon, name_inner_event, _into_id_set from .. import utils -from ..tl import types +from ..tl import types, custom +from ..tl.custom.sendergetter import SenderGetter @name_inner_event @@ -163,7 +164,7 @@ class NewMessage(EventBuilder): return super().filter(event) - class Event(EventCommon): + class Event(EventCommon, SenderGetter): """ Represents the event of a new message. This event can be treated to all effects as a `Message `, @@ -202,7 +203,7 @@ class NewMessage(EventBuilder): msg_id=message.id, broadcast=bool(message.post)) self.pattern_match = None - self.message = message + self.message: custom.Message = message def _set_client(self, client): super()._set_client(client)