standard_imghdr_image_type_detection_from_file_or_bytes.py

1import imghdr
2
3# Identify the image type from a file on disk
4image_type = imghdr.what('example.png')
5print(f"Image type: {image_type}")
6
7# Identify the image type from a byte stream
8with open('example.jpg', 'rb') as f:
9    data = f.read(32)  # Only the first 32 bytes are usually needed
10    image_type_from_bytes = imghdr.what(None, h=data)
11    print(f"Image type from bytes: {image_type_from_bytes}")