aboutsummaryrefslogtreecommitdiffstats
path: root/yt_dlp/compat/imghdr.py
diff options
context:
space:
mode:
Diffstat (limited to 'yt_dlp/compat/imghdr.py')
-rw-r--r--yt_dlp/compat/imghdr.py14
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)