hoelee/Telethon
1
0
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-08-08 04:52:30 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Enhance a lot choosing a valid path when downloading files

Browse Source
This commit is contained in:
Lonami Exo
2017-08-24 17:44:38 +02:00
parent 8cfc4d0fda
commit d5323c5506
2 changed files with 125 additions and 91 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
telethon/utils.py
View File

@@ -9,7 +9,8 @@ from .tl.types import (
ChatPhoto, InputPeerChannel, InputPeerChat, InputPeerUser, InputPeerEmpty,
MessageMediaDocument, MessageMediaPhoto, PeerChannel, InputChannel,
UserEmpty, InputUser, InputUserEmpty, InputUserSelf, InputPeerSelf,
PeerChat, PeerUser, User, UserFull, UserProfilePhoto)
PeerChat, PeerUser, User, UserFull, UserProfilePhoto, Document
)
def get_display_name(entity):
@@ -42,12 +43,17 @@ def get_extension(media):
isinstance(media, MessageMediaPhoto)):
return '.jpg'
# Documents will come with a mime type, from which we can guess their mime type
# Documents will come with a mime type
if isinstance(media, MessageMediaDocument):
extension = guess_extension(media.document.mime_type)
return extension if extension else ''
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 ''
return None
return ''
def get_input_peer(entity):