Fix MessageService not handled on .delete_messages (closes #681)

This commit is contained in:
Lonami Exo 2018-03-11 09:43:55 +01:00
parent 70ef93a62e
commit 055aa7fe43

View File

@ -8,7 +8,7 @@ import re
import sys import sys
import time import time
import warnings import warnings
from collections import OrderedDict, UserList from collections import UserList
from datetime import datetime, timedelta from datetime import datetime, timedelta
from io import BytesIO from io import BytesIO
from mimetypes import guess_type from mimetypes import guess_type
@ -83,7 +83,8 @@ from .tl.types import (
InputMessageEntityMentionName, DocumentAttributeVideo, InputMessageEntityMentionName, DocumentAttributeVideo,
UpdateEditMessage, UpdateEditChannelMessage, UpdateShort, Updates, UpdateEditMessage, UpdateEditChannelMessage, UpdateShort, Updates,
MessageMediaWebPage, ChannelParticipantsSearch, PhotoSize, PhotoCachedSize, MessageMediaWebPage, ChannelParticipantsSearch, PhotoSize, PhotoCachedSize,
PhotoSizeEmpty) PhotoSizeEmpty, MessageService
)
from .tl.types.messages import DialogsSlice from .tl.types.messages import DialogsSlice
from .extensions import markdown, html from .extensions import markdown, html
@ -903,10 +904,13 @@ class TelegramClient(TelegramBareClient):
Returns: Returns:
The affected messages. The affected messages.
""" """
if not utils.is_list_like(message_ids):
message_ids = (message_ids,)
if not isinstance(message_ids, list): message_ids = [
message_ids = [message_ids] m.id if isinstance(m, (Message, MessageService, MessageEmpty))
message_ids = [m.id if isinstance(m, Message) else int(m) for m in message_ids] else int(m) for m in message_ids
]
if entity is None: if entity is None:
return self(messages.DeleteMessagesRequest(message_ids, revoke=revoke)) return self(messages.DeleteMessagesRequest(message_ids, revoke=revoke))