Fix .download_media() not accepting Document

This commit is contained in:
Lonami Exo
2018-01-23 12:10:23 +01:00
parent 81c95b5a60
commit 58d90e7e34
2 changed files with 16 additions and 13 deletions

View File

@@ -61,13 +61,13 @@ def get_extension(media):
# Documents will come with a mime type
if isinstance(media, MessageMediaDocument):
if isinstance(media.document, Document):
if media.document.mime_type == 'application/octet-stream':
# Octet stream are just bytes, which have no default extension
return ''
else:
extension = guess_extension(media.document.mime_type)
return extension if extension else ''
media = media.document
if isinstance(media, Document):
if media.mime_type == 'application/octet-stream':
# Octet stream are just bytes, which have no default extension
return ''
else:
return guess_extension(media.mime_type) or ''
return ''