LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-07-31 10:19:48 +03:00
Code Issues Packages Projects Releases Wiki Activity

switched to creating parser's stream manually

Browse Source
This commit is contained in:
Allerter 2020-09-03 12:16:08 +04:30
parent e73e5c9efc
commit f21a76ebce
1 changed files with 13 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
telethon/utils.py
View File

@ -605,15 +605,20 @@ def _get_metadata(file):
# the extraction process fails.
if hachoir:
try:
original_filetype = type(file)
file = io.BytesIO(file) if isinstance(file, bytes) else file
parser = hachoir.parser.createParser(file)
if isinstance(file, str):
parser.close()
return hachoir.metadata.extractMetadata(parser)
except AssertionError:
msg = ('Your version of Hachoir can only open in-disk files.'
'Update it to read byte arrays/streams.')
_log.warning('Failed to analyze %s: %s', file, msg)
file = open(file, 'rb') if isinstance(file, str) else file
filename = getattr(file, 'name', '')
stream = hachoir.stream.InputIOStream(file,
source='file:' + filename,
tags=[],
filename=filename)
parser = hachoir.parser.guess.guessParser(stream)
metadata = hachoir.metadata.extractMetadata(parser)
if original_filetype is str:
file.close()
return metadata
except Exception as e:
_log.warning('Failed to analyze %s: %s %s', file, e.__class__, e)