123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266 |
- from __future__ import annotations
- import re
- from io import BytesIO
- import base64
- from .typing import ImageType, Union, Image
- try:
- from PIL.Image import open as open_image, new as new_image
- from PIL.Image import FLIP_LEFT_RIGHT, ROTATE_180, ROTATE_270, ROTATE_90
- has_requirements = True
- except ImportError:
- has_requirements = False
- from .errors import MissingRequirementsError
- ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif', 'webp', 'svg'}
- def to_image(image: ImageType, is_svg: bool = False) -> Image:
- """
- Converts the input image to a PIL Image object.
- Args:
- image (Union[str, bytes, Image]): The input image.
- Returns:
- Image: The converted PIL Image object.
- """
- if not has_requirements:
- raise MissingRequirementsError('Install "pillow" package for images')
- if isinstance(image, str):
- is_data_uri_an_image(image)
- image = extract_data_uri(image)
- if is_svg:
- try:
- import cairosvg
- except ImportError:
- raise MissingRequirementsError('Install "cairosvg" package for svg images')
- if not isinstance(image, bytes):
- image = image.read()
- buffer = BytesIO()
- cairosvg.svg2png(image, write_to=buffer)
- return open_image(buffer)
- if isinstance(image, bytes):
- is_accepted_format(image)
- return open_image(BytesIO(image))
- elif not isinstance(image, Image):
- image = open_image(image)
- image.load()
- return image
- return image
- def is_allowed_extension(filename: str) -> bool:
- """
- Checks if the given filename has an allowed extension.
- Args:
- filename (str): The filename to check.
- Returns:
- bool: True if the extension is allowed, False otherwise.
- """
- return '.' in filename and \
- filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
- def is_data_uri_an_image(data_uri: str) -> bool:
- """
- Checks if the given data URI represents an image.
- Args:
- data_uri (str): The data URI to check.
- Raises:
- ValueError: If the data URI is invalid or the image format is not allowed.
- """
- # Check if the data URI starts with 'data:image' and contains an image format (e.g., jpeg, png, gif)
- if not re.match(r'data:image/(\w+);base64,', data_uri):
- raise ValueError("Invalid data URI image.")
- # Extract the image format from the data URI
- image_format = re.match(r'data:image/(\w+);base64,', data_uri).group(1).lower()
- # Check if the image format is one of the allowed formats (jpg, jpeg, png, gif)
- if image_format not in ALLOWED_EXTENSIONS and image_format != "svg+xml":
- raise ValueError("Invalid image format (from mime file type).")
- def is_accepted_format(binary_data: bytes) -> bool:
- """
- Checks if the given binary data represents an image with an accepted format.
- Args:
- binary_data (bytes): The binary data to check.
- Raises:
- ValueError: If the image format is not allowed.
- """
- if binary_data.startswith(b'\xFF\xD8\xFF'):
- return "image/jpeg"
- elif binary_data.startswith(b'\x89PNG\r\n\x1a\n'):
- return "image/png"
- elif binary_data.startswith(b'GIF87a') or binary_data.startswith(b'GIF89a'):
- return "image/gif"
- elif binary_data.startswith(b'\x89JFIF') or binary_data.startswith(b'JFIF\x00'):
- return "image/jpeg"
- elif binary_data.startswith(b'\xFF\xD8'):
- return "image/jpeg"
- elif binary_data.startswith(b'RIFF') and binary_data[8:12] == b'WEBP':
- return "image/webp"
- else:
- raise ValueError("Invalid image format (from magic code).")
- def extract_data_uri(data_uri: str) -> bytes:
- """
- Extracts the binary data from the given data URI.
- Args:
- data_uri (str): The data URI.
- Returns:
- bytes: The extracted binary data.
- """
- data = data_uri.split(",")[1]
- data = base64.b64decode(data)
- return data
- def get_orientation(image: Image) -> int:
- """
- Gets the orientation of the given image.
- Args:
- image (Image): The image.
- Returns:
- int: The orientation value.
- """
- exif_data = image.getexif() if hasattr(image, 'getexif') else image._getexif()
- if exif_data is not None:
- orientation = exif_data.get(274) # 274 corresponds to the orientation tag in EXIF
- if orientation is not None:
- return orientation
- def process_image(image: Image, new_width: int, new_height: int) -> Image:
- """
- Processes the given image by adjusting its orientation and resizing it.
- Args:
- image (Image): The image to process.
- new_width (int): The new width of the image.
- new_height (int): The new height of the image.
- Returns:
- Image: The processed image.
- """
- # Fix orientation
- orientation = get_orientation(image)
- if orientation:
- if orientation > 4:
- image = image.transpose(FLIP_LEFT_RIGHT)
- if orientation in [3, 4]:
- image = image.transpose(ROTATE_180)
- if orientation in [5, 6]:
- image = image.transpose(ROTATE_270)
- if orientation in [7, 8]:
- image = image.transpose(ROTATE_90)
- # Resize image
- image.thumbnail((new_width, new_height))
- # Remove transparency
- if image.mode == "RGBA":
- image.load()
- white = new_image('RGB', image.size, (255, 255, 255))
- white.paste(image, mask=image.split()[-1])
- return white
- elif image.mode != "RGB":
- image = image.convert("RGB")
- return image
- def to_base64_jpg(image: Image, compression_rate: float) -> str:
- """
- Converts the given image to a base64-encoded string.
- Args:
- image (Image.Image): The image to convert.
- compression_rate (float): The compression rate (0.0 to 1.0).
- Returns:
- str: The base64-encoded image.
- """
- output_buffer = BytesIO()
- image.save(output_buffer, format="JPEG", quality=int(compression_rate * 100))
- return base64.b64encode(output_buffer.getvalue()).decode()
- def format_images_markdown(images: Union[str, list], alt: str, preview: Union[str, list] = None) -> str:
- """
- Formats the given images as a markdown string.
- Args:
- images: The images to format.
- alt (str): The alt for the images.
- preview (str, optional): The preview URL format. Defaults to "{image}?w=200&h=200".
- Returns:
- str: The formatted markdown string.
- """
- if isinstance(images, str):
- result = f"[![{alt}]({preview.replace('{image}', images) if preview else images})]({images})"
- else:
- if not isinstance(preview, list):
- preview = [preview.replace('{image}', image) if preview else image for image in images]
- result = "\n".join(
- f"[![#{idx+1} {alt}]({preview[idx]})]({image})" for idx, image in enumerate(images)
- )
- start_flag = "<!-- generated images start -->\n"
- end_flag = "<!-- generated images end -->\n"
- return f"\n{start_flag}{result}\n{end_flag}\n"
- def to_bytes(image: ImageType) -> bytes:
- """
- Converts the given image to bytes.
- Args:
- image (ImageType): The image to convert.
- Returns:
- bytes: The image as bytes.
- """
- if isinstance(image, bytes):
- return image
- elif isinstance(image, str):
- is_data_uri_an_image(image)
- return extract_data_uri(image)
- elif isinstance(image, Image):
- bytes_io = BytesIO()
- image.save(bytes_io, image.format)
- image.seek(0)
- return bytes_io.getvalue()
- else:
- return image.read()
- class ImageResponse:
- def __init__(
- self,
- images: Union[str, list],
- alt: str,
- options: dict = {}
- ):
- self.images = images
- self.alt = alt
- self.options = options
- def __str__(self) -> str:
- return format_images_markdown(self.images, self.alt, self.get("preview"))
- def get(self, key: str):
- return self.options.get(key)
- class ImageRequest:
- def __init__(
- self,
- options: dict = {}
- ):
- self.options = options
- def get(self, key: str):
- return self.options.get(key)
|