LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2024-09-21 03:08:49 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix downloading vcard and webdoc with Path

Browse Source 
Closes #4027.
This commit is contained in:
Lonami Exo 2023-01-23 08:48:00 +01:00
parent 1f8b59043b
commit 9f077e356b
1 changed files with 17 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
telethon/client/downloads.py
View File

@ -918,22 +918,19 @@ class DownloadMethods:
'END:VCARD\n'
).format(f=first_name, l=last_name, p=phone_number).encode('utf-8')
file = cls._get_proper_filename(
file, 'contact', '.vcard',
possible_names=[first_name, phone_number, last_name]
)
if file is bytes:
return result
elif isinstance(file, str):
file = cls._get_proper_filename(
file, 'contact', '.vcard',
possible_names=[first_name, phone_number, last_name]
)
f = open(file, 'wb')
else:
f = file
f = file if hasattr(file, 'write') else open(file, 'wb')
try:
f.write(result)
finally:
# Only close the stream if we opened it
if isinstance(file, str):
if f != file:
f.close()
return file
@ -950,18 +947,17 @@ class DownloadMethods:
)
# TODO Better way to get opened handles of files and auto-close
in_memory = file is bytes
if in_memory:
kind, possible_names = self._get_kind_and_names(web.attributes)
file = self._get_proper_filename(
file, kind, utils.get_extension(web),
possible_names=possible_names
)
if file is bytes:
f = io.BytesIO()
elif isinstance(file, str):
kind, possible_names = cls._get_kind_and_names(web.attributes)
file = cls._get_proper_filename(
file, kind, utils.get_extension(web),
possible_names=possible_names
)
f = open(file, 'wb')
else:
elif hasattr(file, 'write'):
f = file
else:
f = open(file, 'wb')
try:
async with aiohttp.ClientSession() as session:
@ -974,10 +970,10 @@ class DownloadMethods:
break
f.write(chunk)
finally:
if isinstance(file, str) or file is bytes:
if f != file:
f.close()
return f.getvalue() if in_memory else file
return f.getvalue() if file is bytes else file
@staticmethod
def _get_proper_filename(file, kind, extension,