Skip to content

iter_get_work_units

iter_get_work_units(
    query_id: GetWorkUnitsRequestQueryIdString,
    *,
    config_overrides: Optional[LakeFormationClientConfig] = None,
    next_token: Optional[Token] = None,
    page_size: Optional[int] = None,
) -> Iterator[WorkUnitRange]
Source code in docs_src/capo_lakeformation/_services/lake_formation.py
def iter_get_work_units(
    self,
    query_id: "capo_lakeformation.types.get_work_units_request_query_id_string.GetWorkUnitsRequestQueryIdString",
    *,
    config_overrides: Optional[LakeFormationClientConfig] = None,
    next_token: Optional["capo_lakeformation.types.token.Token"] = None,
    page_size: Optional[int] = None,
) -> "Iterator[capo_lakeformation.types.work_unit_range.WorkUnitRange]":
    _token = next_token
    while True:
        _response = self.get_work_units(
            query_id,
            config_overrides=config_overrides,
            next_token=_token,
            page_size=page_size,
        )
        _page = _resolve_path(_response, ("work_unit_ranges",))
        for _item in _page or []:
            yield _item
        _token = _resolve_path(_response, ("next_token",))
        if not _token:
            break