Add .video_note and .gif convenience properties to NewMessage

This commit is contained in:
Lonami Exo 2018-03-08 20:21:56 +01:00
parent cf650e061e
commit 2fb42772c6

View File

@ -522,6 +522,23 @@ class NewMessage(_EventBuilder):
""" """
return self._document_by_attribute(types.DocumentAttributeVideo) return self._document_by_attribute(types.DocumentAttributeVideo)
@property
def video_note(self):
"""
If the message media is a document with a Video attribute,
this returns the (:obj:`Document`) object.
"""
return self._document_by_attribute(types.DocumentAttributeVideo,
lambda attr: attr.round_message)
@property
def gif(self):
"""
If the message media is a document with an Animated attribute,
this returns the (:obj:`Document`) object.
"""
return self._document_by_attribute(types.DocumentAttributeAnimated)
@property @property
def sticker(self): def sticker(self):
""" """