"""Provides utilities for working with image files."""
import logging
from typing import (
List,
Optional,
)
try:
from PIL import Image
except ImportError:
Image = None # type: ignore[assignment, unused-ignore]
log = logging.getLogger(__name__)
[docs]
def image_type(filename: str) -> Optional[str]:
fmt = None
if Image is not None:
try:
with Image.open(filename) as im:
fmt = im.format
mode = im.mode
except Exception:
pass
if fmt:
if fmt == "PPM":
if mode in ["I", "L"]:
return "PGM"
if mode == "1":
return "PBM"
return fmt.upper()
else:
return None
[docs]
def check_image_type(filename: str, types: List[str]) -> bool:
fmt = image_type(filename)
if fmt in types:
return True
return False