Warning

This document is for an in-development version of Galaxy. You can alternatively view this page in the latest release if it exists or view the top of the latest release's documentation.

Source code for galaxy.util.image_util

"""Provides utilities for working with image files."""

import logging

try:
    from PIL import Image
except ImportError:
    Image = None  # type: ignore[assignment, unused-ignore]

log = logging.getLogger(__name__)




def image_type(filename: str) -> str | None:
    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





def check_image_type(filename: str, types: list[str]) -> bool:
    fmt = image_type(filename)
    if fmt in types:
        return True
    return False