utils.py

import re
from pathlib import Path

IMAGE_EXTENSIONS = {'.png', '.jpg', '.jpeg', '.webp', '.tiff', '.tif', '.bmp', '.gif', '.avif', '.heic', '.heif', '.svg'}

SIZE_PATTERNS = [
    (r'^(\d+)x$', 'width'),
    (r'^x(\d+)$', 'height'),
    (r'^(\d+)x(\d+)$', 'box'),
    (r'^(\d+)x(\d+)!$', 'stretch'),
    (r'^宽(\d+)$', 'width'),
    (r'^宽度(\d+)$', 'width'),
    (r'^高(\d+)$', 'height'),
    (r'^高度(\d+)$', 'height'),
    (r'^(\d+)$', 'width'),
]

OPERATION_ALIASES = {
    '右旋': 'rot_right',
    '右': 'rot_right',
    '左旋': 'rot_left',
    '左': 'rot_left',
    '左右翻转': 'flip_h',
    '左右': 'flip_h',
    '上下翻转': 'flip_v',
    '上下': 'flip_v',
    '批量': 'batch',
    '批': 'batch',
    '动画': 'keep_animation',
    '动': 'keep_animation',
    '拉伸': 'stretch',
    '拉': 'stretch',
}


def is_image_file(path: str) -> bool:
    return Path(path).suffix.lower() in IMAGE_EXTENSIONS


def parse_size(size_str: str):
    for pattern, size_type in SIZE_PATTERNS:
        match = re.match(pattern, size_str)
        if match:
            if size_type == 'width':
                return ('width', int(match.group(1)))
            elif size_type == 'height':
                return ('height', int(match.group(1)))
            elif size_type == 'box':
                return ('box', (int(match.group(1)), int(match.group(2))))
            elif size_type == 'stretch':
                return ('stretch', (int(match.group(1)), int(match.group(2))))
    return None


def get_output_path(input_path: str, output_dir: str, target_ext: str):
    input_name = Path(input_path).stem
    output_dir_path = Path(output_dir)
    output_dir_path.mkdir(parents=True, exist_ok=True)

    base_output = output_dir_path / f"{input_name}.{target_ext}"

    if not base_output.exists():
        return str(base_output)

    counter = 0
    while True:
        candidate = output_dir_path / f"{input_name}-{counter}.{target_ext}"
        if not candidate.exists():
            return str(candidate)
        counter += 1