search_image_by_image/pools.py

from typing import Type

from cheesechaser.datapool import IncrementIDDataPool


def quick_webp_pool(site_name: str, level: int = 3) -> Type[IncrementIDDataPool]:
    repo_id = f'deepghs/{site_name}-webp-4Mpixel'

    class _QuickWebpDataPool(IncrementIDDataPool):
        def __init__(self, revision: str = 'main'):
            IncrementIDDataPool.__init__(
                self,
                data_repo_id=repo_id,
                data_revision=revision,
                idx_repo_id=repo_id,
                idx_revision=revision,
                base_level=level,
            )

    return _QuickWebpDataPool