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

Fix file name could be lost when uploading files

Browse Source 
Leading to invalid extension when sending photos.
This commit is contained in:
Lonami Exo
2023-10-17 20:31:58 +02:00
parent eb67ef1b15
commit fda6840449
3 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
telethon/helpers.py
View File

@@ -348,21 +348,22 @@ class _FileStream(io.IOBase):
self._size = os.path.getsize(self._file)
self._stream = open(self._file, 'rb')
self._close_stream = True
return self
elif isinstance(self._file, bytes):
if isinstance(self._file, bytes):
self._size = len(self._file)
self._stream = io.BytesIO(self._file)
self._close_stream = True
return self
elif not callable(getattr(self._file, 'read', None)):
if not callable(getattr(self._file, 'read', None)):
raise TypeError('file description should have a `read` method')
elif self._size is not None:
self._name = getattr(self._file, 'name', None)
self._stream = self._file
self._close_stream = False
self._name = getattr(self._file, 'name', None)
self._stream = self._file
self._close_stream = False
else:
if self._size is None:
if callable(getattr(self._file, 'seekable', None)):
seekable = await _maybe_await(self._file.seekable())
else:
@@ -373,8 +374,6 @@ class _FileStream(io.IOBase):
await _maybe_await(self._file.seek(0, os.SEEK_END))
self._size = await _maybe_await(self._file.tell())
await _maybe_await(self._file.seek(pos, os.SEEK_SET))
self._stream = self._file
self._close_stream = False
else:
_log.warning(
'Could not determine file size beforehand so the entire '