From f7a071cb0bf05b86ccfdd4ed3d2b2febb5b0a9a2 Mon Sep 17 00:00:00 2001 From: Alex Root Junior Date: Thu, 26 Jul 2018 22:53:56 +0300 Subject: [PATCH] Added the field animation to the Message object. --- aiogram/types/message.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/aiogram/types/message.py b/aiogram/types/message.py index 75b86291..c107220a 100644 --- a/aiogram/types/message.py +++ b/aiogram/types/message.py @@ -5,6 +5,7 @@ import typing from . import base from . import fields +from .animation import Animation from .audio import Audio from .chat import Chat from .contact import Contact @@ -49,6 +50,7 @@ class Message(base.TelegramObject): caption_entities: typing.List[MessageEntity] = fields.ListField(base=MessageEntity) audio: Audio = fields.Field(base=Audio) document: Document = fields.Field(base=Document) + animation: Animation = fields.Field(base=Animation) game: Game = fields.Field(base=Game) photo: typing.List[PhotoSize] = fields.ListField(base=PhotoSize) sticker: Sticker = fields.Field(base=Sticker) @@ -81,6 +83,8 @@ class Message(base.TelegramObject): return ContentType.TEXT[0] elif self.audio: return ContentType.AUDIO[0] + elif self.animation: + return ContentType.ANIMATION[0] elif self.document: return ContentType.DOCUMENT[0] elif self.game: @@ -748,6 +752,7 @@ class ContentType(helper.Helper): TEXT = helper.ListItem() # text AUDIO = helper.ListItem() # audio DOCUMENT = helper.ListItem() # document + ANIMATION = helper.ListItem() # animation GAME = helper.ListItem() # game PHOTO = helper.ListItem() # photo STICKER = helper.ListItem() # sticker