hoelee/Telethon
1
0
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-06-17 10:36:37 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix type hinting for custom.Message

Browse Source
This commit is contained in:
Lonami Exo 2021-01-28 19:58:03 +01:00
parent 4e73577d59
commit 055643bd01
1 changed files with 36 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
telethon/tl/custom/message.py
View File

@ -1,3 +1,5 @@
from typing import Optional, List, TYPE_CHECKING
from datetime import datetime
from .chatgetter import ChatGetter from .chatgetter import ChatGetter
from .sendergetter import SenderGetter from .sendergetter import SenderGetter
from .messagebutton import MessageButton from .messagebutton import MessageButton
@ -152,29 +154,47 @@ class _Message(ChatGetter, SenderGetter):
def __init__( def __init__(
# Common to all # Common to all
self, id, self, id: int,
# Common to Message and MessageService (mandatory) # Common to Message and MessageService (mandatory)
peer_id=None, date=None, peer_id: types.TypePeer = None,
date: Optional[datetime] = None,
# Common to Message and MessageService (flags) # Common to Message and MessageService (flags)
out=None, mentioned=None, media_unread=None, silent=None, out: Optional[bool] = None,
post=None, from_id=None, reply_to=None, mentioned: Optional[bool] = None,
media_unread: Optional[bool] = None,
silent: Optional[bool] = None,
post: Optional[bool] = None,
from_id: Optional[types.TypePeer] = None,
reply_to: Optional[types.TypeMessageReplyHeader] = None,
# For Message (mandatory) # For Message (mandatory)
message=None, message: Optional[str] = None,
# For Message (flags) # For Message (flags)
fwd_from=None, via_bot_id=None, media=None, reply_markup=None, fwd_from: Optional[types.TypeMessageFwdHeader] = None,
entities=None, views=None, edit_date=None, post_author=None, via_bot_id: Optional[int] = None,
grouped_id=None, from_scheduled=None, legacy=None, media: Optional[types.TypeMessageMedia] = None,
edit_hide=None, pinned=None, restriction_reason=None, reply_markup: Optional[types.TypeReplyMarkup] = None,
forwards=None, replies=None, entities: Optional[List[types.TypeMessageEntity]] = None,
views: Optional[int] = None,
edit_date: Optional[datetime] = None,
post_author: Optional[str] = None,
grouped_id: Optional[int] = None,
from_scheduled: Optional[bool] = None,
legacy: Optional[bool] = None,
edit_hide: Optional[bool] = None,
pinned: Optional[bool] = None,
restriction_reason: Optional[types.TypeRestrictionReason] = None,
forwards: Optional[int] = None,
replies: Optional[types.TypeMessageReplies] = None,
# For MessageAction (mandatory) # For MessageAction (mandatory)
action=None): action: Optional[types.TypeMessageAction] = None
):
# Common properties to messages, then to service (in the order they're defined in the `.tl`) # Common properties to messages, then to service (in the order they're defined in the `.tl`)
self.out = out self.out = bool(out)
self.mentioned = mentioned self.mentioned = mentioned
self.media_unread = media_unread self.media_unread = media_unread
self.silent = silent self.silent = silent
@ -1119,3 +1139,7 @@ def _patch(cls):
_patch(types.MessageEmpty) _patch(types.MessageEmpty)
Message = _patch(types.Message) Message = _patch(types.Message)
_patch(types.MessageService) _patch(types.MessageService)
if TYPE_CHECKING:
Message = _Message