mirror of
https://github.com/hyugogirubato/cbz.git
synced 2026-09-26 21:41:03 +00:00
110 lines
3.2 KiB
Python
110 lines
3.2 KiB
Python
from enum import Enum
|
|
from io import BytesIO
|
|
from pathlib import Path
|
|
|
|
from PIL import Image
|
|
from PIL.IcoImagePlugin import IcoFile
|
|
|
|
|
|
def default_attr(value: any) -> any:
|
|
"""
|
|
Provides a default value based on the expected type of attribute.
|
|
|
|
Args:
|
|
value (any): Expected type or class of the attribute.
|
|
|
|
Returns:
|
|
any: Default value appropriate for the specified type or class.
|
|
- For Enum types: Returns the 'UNKNOWN' member if available, otherwise the first member.
|
|
- For int or float: Returns -1.
|
|
- For bool: Returns False.
|
|
- For str: Returns an empty string.
|
|
- For other types: Invokes the callable (assuming it's a function or callable object).
|
|
"""
|
|
if issubclass(value, Enum):
|
|
keys = [i.name for i in list(value)]
|
|
return value['UNKNOWN' if 'UNKNOWN' in keys else 'STORY']
|
|
elif value in (int, float):
|
|
return -1
|
|
elif value == bool:
|
|
return False
|
|
elif value == str:
|
|
return ''
|
|
else:
|
|
return value()
|
|
|
|
|
|
def verify_attr(expected_type: any, key: str, value: any) -> None:
|
|
"""
|
|
Verifies if the provided value matches the expected type.
|
|
|
|
Args:
|
|
expected_type (any): Expected type of the attribute.
|
|
key (str): Name of the attribute.
|
|
value (any): Value to be verified against the expected type.
|
|
|
|
Raises:
|
|
TypeError: If the provided value does not match the expected type.
|
|
"""
|
|
if not isinstance(value, expected_type):
|
|
raise TypeError(f'Expected type {expected_type} for attribute "{key}", but got {type(value)}')
|
|
|
|
|
|
def repr_attr(value: any) -> any:
|
|
"""
|
|
Provides a representation of the attribute's value.
|
|
|
|
Args:
|
|
value (any): Value of the attribute.
|
|
|
|
Returns:
|
|
any: Representation of the attribute's value.
|
|
- For Enum types: Returns the value of the Enum.
|
|
- For other types: Returns the value itself.
|
|
"""
|
|
if isinstance(value, Enum):
|
|
return value.value
|
|
return value
|
|
|
|
|
|
def readable_size(size: int, decimal: int = 2) -> str:
|
|
"""
|
|
Converts a file size in bytes to a human-readable string format.
|
|
|
|
Args:
|
|
size (int): The size in bytes.
|
|
decimal (int): Number of decimal places to display (default is 2).
|
|
|
|
Returns:
|
|
str: Human-readable string representation of the size.
|
|
"""
|
|
for unit in ['B', 'KB', 'MB', 'GB', 'TB']:
|
|
if size < 1024:
|
|
return f'{size:.{decimal}f} {unit}'
|
|
size /= 1024
|
|
|
|
|
|
def ico_to_png(path: Path) -> BytesIO:
|
|
"""
|
|
Converts the largest icon in an ICO file to PNG format.
|
|
|
|
Args:
|
|
path (Path): Path to the ICO file.
|
|
|
|
Returns:
|
|
BytesIO: In-memory PNG file of the largest icon.
|
|
"""
|
|
# Open the ICO file and read its content
|
|
image = Image.open(BytesIO(path.read_bytes()))
|
|
assert image.format == 'ICO', 'Unsupported image format'
|
|
|
|
# Get the ICO file object and find the largest icon size
|
|
icon: IcoFile = image.ico
|
|
max_size = max(icon.sizes(), key=lambda x: x[0] + x[1])
|
|
largest_image = icon.getimage(size=max_size)
|
|
|
|
# Save the largest icon as PNG format to an in-memory BytesIO object
|
|
content = BytesIO()
|
|
largest_image.save(content, format='PNG')
|
|
return content
|