yt-dlp/yt_dlp/compat/imghdr.py

23 lines
582 B
Python
Raw Normal View History

2022-07-30 19:59:02 +00:00
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.11/Lib/imghdr.py
Ref: https://www.w3.org/Graphics/JPEG/itu-t81.pdf
"""
2022-07-30 19:59:02 +00:00
if h is None:
with open(file, 'rb') as f:
h = f.read(12)
if h.startswith(b'RIFF') and h.startswith(b'WEBP', 8):
return 'webp'
if h.startswith(b'\x89PNG'):
return 'png'
if h.startswith(b'\xFF\xD8\xFF'):
return 'jpeg'
if h.startswith(b'GIF'):
return 'gif'
return None