from typing import List

from mypy_boto3_mturk.type_defs import WorkerBlockTypeDef

from jb.decorators import AMT_CLIENT


class WorkerManager:

    @staticmethod
    def fetch_worker_blocks() -> List[WorkerBlockTypeDef]:
        p = AMT_CLIENT.get_paginator("list_worker_blocks")
        res: List[WorkerBlockTypeDef] = []
        for item in p.paginate():
            res.extend(item["WorkerBlocks"])
        return res