Function _calculate_img_number

ui/easydiffusion/utils/save_utils.py:324–362

(save_dir_path: str, task_data: RenderTaskData)

def _calculate_img_number(save_dir_path: str, task_data: RenderTaskData):
    def get_highest_img_number(accumulator: int, file: os.DirEntry) -> int:
        if not file.is_file:
            return accumulator

        if len(list(filter(lambda e: file.name.endswith(e), app.IMAGE_EXTENSIONS))) == 0:
            return accumulator

        get_highest_img_number.number_of_images = get_highest_img_number.number_of_images + 1

        number_match = img_number_regex.match(file.name)
        if not number_match:
            return accumulator

        file_number = number_match.group().lstrip("0")

        # Handle 00000
        return int(file_number) if file_number else 0

    get_highest_img_number.number_of_images = 0

    highest_file_number = -1

    if os.path.isdir(save_dir_path):
        existing_files = list(os.scandir(save_dir_path))
        highest_file_number = reduce(get_highest_img_number, existing_files, -1)

    calculated_img_number = max(highest_file_number, get_highest_img_number.number_of_images - 1)

    if task_data.session_id in _calculate_img_number.session_img_numbers:
        calculated_img_number = max(
            _calculate_img_number.session_img_numbers[task_data.session_id],
            calculated_img_number,
        )

    calculated_img_number = calculated_img_number + 1

    _calculate_img_number.session_img_numbers[task_data.session_id] = calculated_img_number
    return calculated_img_number


_calculate_img_number.session_img_numbers = {}