def getMedia(self, mediaMessageProtocolEntity, message):
media = Media(
type=MediaType.get_mediatype(mediaMessageProtocolEntity.getMediaType()),
preview=mediaMessageProtocolEntity.getPreview())
if mediaMessageProtocolEntity.getMediaType() in (
MediaMessageProtocolEntity.MEDIA_TYPE_IMAGE,
MediaMessageProtocolEntity.MEDIA_TYPE_AUDIO,
MediaMessageProtocolEntity.MEDIA_TYPE_VIDEO
):
self.setDownloadableMediaData(mediaMessageProtocolEntity, media)
media.encoding = mediaMessageProtocolEntity.encoding
if mediaMessageProtocolEntity.getMediaType() != MediaMessageProtocolEntity.MEDIA_TYPE_AUDIO:
message.content = mediaMessageProtocolEntity.getCaption()
elif mediaMessageProtocolEntity.getMediaType() == MediaMessageProtocolEntity.MEDIA_TYPE_LOCATION:
message.content = mediaMessageProtocolEntity.getLocationName()
self.setLocationMediaData(mediaMessageProtocolEntity, media)
elif mediaMessageProtocolEntity.getMediaType() == MediaMessageProtocolEntity.MEDIA_TYPE_VCARD:
message.content = mediaMessageProtocolEntity.getName()
self.setVCardMediaData(mediaMessageProtocolEntity, media)
return media
