Support downloading web documents

This commit is contained in:
Lonami Exo
2018-08-01 00:37:25 +02:00
parent 6d1bc227aa
commit 76c7217000
2 changed files with 64 additions and 13 deletions

View File

@@ -92,7 +92,8 @@ def get_extension(media):
# Documents will come with a mime type
if isinstance(media, types.MessageMediaDocument):
media = media.document
if isinstance(media, types.Document):
if isinstance(media, (
types.Document, types.WebDocument, types.WebDocumentNoProxy)):
if media.mime_type == 'application/octet-stream':
# Octet stream are just bytes, which have no default extension
return ''