LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-08-02 11:10:18 +03:00
Code Issues Packages Projects Releases Wiki Activity

switched to returning None for non-seekable files

Browse Source
This commit is contained in:
Allerter 2020-09-08 00:39:42 +04:30
parent 92beac2e35
commit 8f3de80064
1 changed files with 4 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
telethon/utils.py
View File

@ -611,6 +611,7 @@ def _get_metadata(file):
# The parser may fail and we don't want to crash if # The parser may fail and we don't want to crash if
# the extraction process fails. # the extraction process fails.
try: try:
# Note: aiofiles are intentionally left out for simplicity
if isinstance(file, str): if isinstance(file, str):
stream = open(file, 'rb') stream = open(file, 'rb')
elif isinstance(file, bytes): elif isinstance(file, bytes):
@ -624,22 +625,18 @@ def _get_metadata(file):
seekable = False seekable = False
if not seekable: if not seekable:
_log.warning( return None
'Could not read metadata since the file is not '
'seekable so the entire file will be read in-memory')
data = file.read()
stream = io.BytesIO(data)
close_stream = True
pos = stream.tell() pos = stream.tell()
filename = getattr(file, 'name', '') filename = getattr(file, 'name', '')
parser = hachoir.parser.guess.guessParser(hachoir.stream.InputIOStream( parser = hachoir.parser.guess.guessParser(hachoir.stream.InputIOStream(
stream, stream,
source='file:' + filename, source='file:' + filename,
tags=[], tags=[],
filename=filename filename=filename
)) ))
return hachoir.metadata.extractMetadata(parser) return hachoir.metadata.extractMetadata(parser)
except Exception as e: except Exception as e: