Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@ keywords = [
"scraping",
]
dependencies = [
"async-timeout>=5.0.1",
"async-timeout>=5.0.1; python_full_version < '3.11'",
"cachetools>=5.5.0",
"colorama>=0.4.0",
"impit>=0.13.2",
Expand Down
16 changes: 12 additions & 4 deletions src/crawlee/_utils/time.py
Original file line number Diff line number Diff line change
@@ -1,17 +1,25 @@
from __future__ import annotations

import asyncio
import sys
import time
from contextlib import contextmanager
from dataclasses import dataclass
from datetime import timedelta
from typing import TYPE_CHECKING

from async_timeout import Timeout, timeout

if TYPE_CHECKING:
from collections.abc import Iterator
from types import TracebackType

if sys.version_info >= (3, 11):
timeout_ctx = asyncio.timeout
else:
# TODO: `async-timeout` backports `asyncio.timeout` for Python 3.10; drop once the
# project floor bump to 3.11
# https://github.com/apify/crawlee-python/issues/2052
from async_timeout import timeout as timeout_ctx # type: ignore[no-redef]

_SECONDS_PER_MINUTE = 60
_SECONDS_PER_HOUR = 3600

Expand Down Expand Up @@ -46,15 +54,15 @@ class SharedTimeout:

def __init__(self, timeout: timedelta) -> None:
self._remaining_timeout = timeout
self._active_timeout: Timeout | None = None
self._active_timeout: asyncio.Timeout | None = None
self._activation_timestamp: float | None = None

async def __aenter__(self) -> timedelta:
if self._active_timeout is not None or self._activation_timestamp is not None:
raise RuntimeError('A shared timeout context cannot be entered twice at the same time')

self._activation_timestamp = time.monotonic()
self._active_timeout = new_timeout = timeout(self._remaining_timeout.total_seconds())
self._active_timeout = new_timeout = timeout_ctx(self._remaining_timeout.total_seconds())
await new_timeout.__aenter__()
return self._remaining_timeout

Expand Down
4 changes: 2 additions & 2 deletions uv.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.