mirror of
https://github.com/simple-login/app.git
synced 2024-11-02 20:01:01 +01:00
29 lines
679 B
Python
29 lines
679 B
Python
|
from enum import Enum
|
||
|
|
|
||
|
|
|
||
|
|
class ImageFormat(Enum):
|
||
|
|
Png = 1
|
||
|
|
Jpg = 2
|
||
|
|
Webp = 3
|
||
|
|
Svg = 4
|
||
|
|
Unknown = 9
|
||
|
|
|
||
|
|
|
||
|
|
magic_numbers = {
|
||
|
|
ImageFormat.Png: bytes([0x89, 0x50, 0x4E, 0x47, 0x0D, 0x0A, 0x1A, 0x0A]),
|
||
|
|
ImageFormat.Jpg: bytes([0xFF, 0xD8, 0xFF, 0xE0]),
|
||
|
|
ImageFormat.Webp: bytes([0x52, 0x49, 0x46, 0x46]),
|
||
|
|
ImageFormat.Svg: bytes([0x3C, 0x3F, 0x78, 0x6D, 0x6C]), # <?xml
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
def detect_image_format(image: bytes) -> ImageFormat:
|
||
|
|
# Detect image based on magic number
|
||
|
|
for fmt, header in magic_numbers.items():
|
||
|
|
if image.startswith(header):
|
||
|
|
return fmt
|
||
|
|
# Detect if is svg
|
||
|
|
|
||
|
|
# We don't know the type
|
||
|
|
return ImageFormat.Unknown
|