wehub-resource-sync
e904b667c6
PaddleOCR PR Tests GPU / test-pr-gpu (push) Blocked by required conditions
PaddleOCR PR Tests / test-pr (push) Blocked by required conditions
PaddleOCR PR Tests / test-pr-python (3.8) (push) Waiting to run
Build/Publish Develop Docs / deploy (push) Failing after 1s
PaddleOCR Code Style Check / check-code-style (push) Failing after 1s
PaddleOCR PR Tests GPU / detect-changes (push) Failing after 1s
PaddleOCR PR Tests GPU / test-pr-gpu-impl (push) Waiting to run
PaddleOCR PR Tests / detect-changes (push) Failing after 1s
PaddleOCR PR Tests / test-pr-python (3.13) (push) Waiting to run
PaddleOCR PR Tests / test-pr-python (3.9) (push) Waiting to run
87 lines
2.9 KiB
Python
87 lines
2.9 KiB
Python
# Copyright (c) 2026 PaddlePaddle Authors. All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
import asyncio
|
|
from typing import Any
|
|
|
|
from ._core import (
|
|
job_status_from_data,
|
|
parse_batch_status,
|
|
validate_result_json_url,
|
|
validate_state,
|
|
)
|
|
from .errors import (
|
|
JobFailedError,
|
|
PollTimeoutError,
|
|
)
|
|
from .results import BatchStatus, JobStatus
|
|
|
|
DEFAULT_INITIAL_INTERVAL = 3.0
|
|
DEFAULT_MULTIPLIER = 1.5
|
|
DEFAULT_MAX_INTERVAL = 15.0
|
|
DEFAULT_MAX_WAIT_TIME = 600.0
|
|
|
|
|
|
class AsyncPoller:
|
|
def __init__(
|
|
self,
|
|
http_client,
|
|
initial_interval: float = DEFAULT_INITIAL_INTERVAL,
|
|
multiplier: float = DEFAULT_MULTIPLIER,
|
|
max_interval: float = DEFAULT_MAX_INTERVAL,
|
|
max_wait_time: float = DEFAULT_MAX_WAIT_TIME,
|
|
):
|
|
self._http = http_client
|
|
self._initial_interval = initial_interval
|
|
self._multiplier = multiplier
|
|
self._max_interval = max_interval
|
|
self._max_wait_time = max_wait_time
|
|
|
|
async def poll_until_done(self, job_id: str) -> Any:
|
|
interval = self._initial_interval
|
|
loop = asyncio.get_running_loop()
|
|
start = loop.time()
|
|
deadline = start + self._max_wait_time
|
|
|
|
while True:
|
|
now = loop.time()
|
|
if now >= deadline:
|
|
raise PollTimeoutError(job_id, now - start)
|
|
|
|
data = await self._http.get_job_status(job_id)
|
|
state = validate_state(data)
|
|
|
|
if state == "done":
|
|
json_url = validate_result_json_url(data)
|
|
jsonl_data = await self._http.fetch_jsonl(json_url)
|
|
return jsonl_data, data
|
|
|
|
if state == "failed":
|
|
error_msg = data.get("errorMsg", "Unknown error")
|
|
raise JobFailedError(job_id, error_msg)
|
|
|
|
remaining = deadline - loop.time()
|
|
if remaining <= 0:
|
|
raise PollTimeoutError(job_id, loop.time() - start)
|
|
await asyncio.sleep(min(interval, remaining))
|
|
interval = min(interval * self._multiplier, self._max_interval)
|
|
|
|
async def get_status(self, job_id: str) -> JobStatus:
|
|
data = await self._http.get_job_status(job_id)
|
|
return job_status_from_data(job_id, data)
|
|
|
|
async def get_batch_status(self, batch_id: str) -> BatchStatus:
|
|
data = await self._http.get_batch_status(batch_id)
|
|
return parse_batch_status(batch_id, data)
|