Skip to content

Commit c30f80d

Browse files
author
Andrei Neagu
committed
removed some constants
1 parent 08a65e7 commit c30f80d

File tree

2 files changed

+7
-6
lines changed

2 files changed

+7
-6
lines changed

packages/service-library/src/servicelib/aiohttp/long_running_tasks/client.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,11 @@
11
import asyncio
22
import logging
33
from collections.abc import AsyncGenerator
4-
from typing import Any
4+
from datetime import timedelta
5+
from typing import Any, Final
56

67
from aiohttp import ClientConnectionError, ClientSession
8+
from pydantic import PositiveFloat
79
from tenacity import TryAgain, retry
810
from tenacity.asyncio import AsyncRetrying
911
from tenacity.before_sleep import before_sleep_log
@@ -12,7 +14,7 @@
1214
from tenacity.wait import wait_random_exponential
1315
from yarl import URL
1416

15-
from ...long_running_tasks.constants import DEFAULT_POLL_INTERVAL_S, HOUR
17+
from ...long_running_tasks.constants import DEFAULT_POLL_INTERVAL_S
1618
from ...long_running_tasks.models import (
1719
LRTask,
1820
RequestBody,
@@ -26,6 +28,7 @@
2628

2729
_logger = logging.getLogger(__name__)
2830

31+
_DEFAULT_CLIENT_TIMEOUT_S: Final[PositiveFloat] = timedelta(hours=1).total_seconds()
2932

3033
_DEFAULT_AIOHTTP_RETRY_POLICY: dict[str, Any] = {
3134
"retry": retry_if_exception_type(ClientConnectionError),
@@ -49,7 +52,7 @@ async def _wait_for_completion(
4952
session: ClientSession,
5053
task_id: TaskId,
5154
status_url: URL,
52-
client_timeout: int,
55+
client_timeout: PositiveFloat,
5356
) -> AsyncGenerator[TaskProgress, None]:
5457
try:
5558
async for attempt in AsyncRetrying(
@@ -98,7 +101,7 @@ async def long_running_task_request(
98101
session: ClientSession,
99102
url: URL,
100103
json: RequestBody | None = None,
101-
client_timeout: int = 1 * HOUR,
104+
client_timeout: PositiveFloat = _DEFAULT_CLIENT_TIMEOUT_S,
102105
) -> AsyncGenerator[LRTask, None]:
103106
"""Will use the passed `ClientSession` to call an oSparc long
104107
running task `url` passing `json` as request body.

packages/service-library/src/servicelib/long_running_tasks/constants.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,6 @@
11
from datetime import timedelta
22
from typing import Final
33

4-
MINUTE: Final[int] = 60 # in secs
5-
HOUR: Final[int] = 60 * MINUTE # in secs
64
DEFAULT_POLL_INTERVAL_S: Final[float] = 1
75

86
DEFAULT_STALE_TASK_CHECK_INTERVAL: Final[timedelta] = timedelta(minutes=1)

0 commit comments

Comments
 (0)