diff options
| author | Jesús <heckyel@hyperbola.info> | 2022-06-27 01:25:17 +0800 |
|---|---|---|
| committer | Jesús <heckyel@hyperbola.info> | 2022-06-27 01:25:17 +0800 |
| commit | 16e8548f6a720a78679e417a20a300db2036bf6c (patch) | |
| tree | b1247bca3417ce882e4a4d80213f41c20113c1a4 /yt_dlp/compat/imghdr.py | |
| parent | 4bbf329feb5a820ac21269fa426c95ca14d7af25 (diff) | |
| parent | e08f72e6759fb6b1102521f0bdb9457038ef7c06 (diff) | |
| download | hypervideo-pre-16e8548f6a720a78679e417a20a300db2036bf6c.tar.lz hypervideo-pre-16e8548f6a720a78679e417a20a300db2036bf6c.tar.xz hypervideo-pre-16e8548f6a720a78679e417a20a300db2036bf6c.zip | |
updated from upstream | 27/06/2022 at 01:25
Diffstat (limited to 'yt_dlp/compat/imghdr.py')
| -rw-r--r-- | yt_dlp/compat/imghdr.py | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/yt_dlp/compat/imghdr.py b/yt_dlp/compat/imghdr.py new file mode 100644 index 000000000..734b0d876 --- /dev/null +++ b/yt_dlp/compat/imghdr.py @@ -0,0 +1,14 @@ +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'), +} + + +def what(path): + """Detect format of image (Currently supports jpeg, png, webp 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) |