diff options
| author | pukkandan <pukkandan.ytdlp@gmail.com> | 2022-07-31 01:29:02 +0530 |
|---|---|---|
| committer | pukkandan <pukkandan.ytdlp@gmail.com> | 2022-07-31 02:20:12 +0530 |
| commit | b4daacb4ecd1f686d1a4e204ade6a9b1bb75a5d3 (patch) | |
| tree | 341a6270d4c43c5b3e5f788b0be87b49621c3c87 /yt_dlp/compat/imghdr.py | |
| parent | 6a7d3a0a0981d05903e70bcb31fc3f9438eedf22 (diff) | |
| download | hypervideo-pre-b4daacb4ecd1f686d1a4e204ade6a9b1bb75a5d3.tar.lz hypervideo-pre-b4daacb4ecd1f686d1a4e204ade6a9b1bb75a5d3.tar.xz hypervideo-pre-b4daacb4ecd1f686d1a4e204ade6a9b1bb75a5d3.zip | |
[mhtml, cleanup] Use imghdr
Diffstat (limited to 'yt_dlp/compat/imghdr.py')
| -rw-r--r-- | yt_dlp/compat/imghdr.py | 12 |
1 files changed, 7 insertions, 5 deletions
diff --git a/yt_dlp/compat/imghdr.py b/yt_dlp/compat/imghdr.py index 734b0d876..5d64ab07b 100644 --- a/yt_dlp/compat/imghdr.py +++ b/yt_dlp/compat/imghdr.py @@ -2,13 +2,15 @@ tests = { 'webp': lambda h: h[0:4] == b'RIFF' and h[8:] == b'WEBP', 'png': lambda h: h[:8] == b'\211PNG\r\n\032\n', 'jpeg': lambda h: h[6:10] in (b'JFIF', b'Exif'), + 'gif': lambda h: h[:6] in (b'GIF87a', b'GIF89a'), } -def what(path): - """Detect format of image (Currently supports jpeg, png, webp only) +def what(file=None, h=None): + """Detect format of image (Currently supports jpeg, png, webp, gif only) Ref: https://github.com/python/cpython/blob/3.10/Lib/imghdr.py """ - with open(path, 'rb') as f: - head = f.read(12) - return next((type_ for type_, test in tests.items() if test(head)), None) + if h is None: + with open(file, 'rb') as f: + h = f.read(12) + return next((type_ for type_, test in tests.items() if test(h)), None) |