o _j%@sdZddlmZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z mZmZmZmZedZe d Ze d d Gd d d ZddddddZd dgZdS)uZHTTP layer — retry policy with exponential backoff and jitter. Key design decisions vs. original code: - ``_MAX_RETRIES = 1000`` → configurable, default 5, max 10. - ``2**attempt`` unbounded sleep → ``min(base**attempt + jitter, cap)`` capped at 30s. - Session is NOT recreated on every retry — only recreated on explicit session error. ) annotationsN)Callable) dataclass)TypeVar)DEFAULT_BACKOFF_BASEDEFAULT_BACKOFF_CAP_SDEFAULT_BACKOFF_JITTERDEFAULT_MAX_RETRIESRATE_LIMIT_STATUS_CODESzindiaopt.retryTT)frozenc@szeZdZUdZeZded<eZded<e Z ded<e Z ded<e Zded <dd d ZdddZdddZedddZdS) RetryPolicyaImmutable retry policy configuration. Attributes: max_retries: Maximum number of attempts (including the first). backoff_base: Base for exponential backoff formula. backoff_cap: Maximum sleep duration in seconds. jitter: Maximum random jitter added to each sleep. retry_statuses: HTTP status codes that trigger a retry. Backoff formula:: sleep = min(base ** attempt + uniform(0, jitter), cap) int max_retriesfloat backoff_base backoff_capjitterzfrozenset[int]retry_statusesattemptreturncCs@t|j|td|j|j}td||dt ||S)zCompute and sleep for the backoff duration on *attempt* (0-indexed). Returns: Actual sleep duration in seconds. z!Backoff sleep %.2fs (attempt %d).) minrrandomuniformrrloggerdebugtimesleep)selfrdurationr"N/home/dinkstrade/pdmp/venv/lib/python3.10/site-packages/indiaopt/http/retry.py sleep_for5s zRetryPolicy.sleep_forcCs t|j|td|j|jS)zmReturn the backoff duration for *attempt* without sleeping. Useful for logging and testing. r)rrrrrr)r rr"r"r# compute_sleepCszRetryPolicy.compute_sleep status_codeboolcCs ||jvS)z2Return ``True`` if *status_code* warrants a retry.)r)r r&r"r"r#should_retry_statusMs zRetryPolicy.should_retry_statussettingsobjectcCs0|t|dtt|dtt|dtt|dtdS)z[Construct a :class:`RetryPolicy` from a :class:`~indiaopt.config.settings.Settings` object.rrbackoff_cap_secondsbackoff_jitter)rrrr)getattrr rrr)clsr)r"r"r# from_settingsQs     zRetryPolicy.from_settingsN)rrrr)r&rrr')r)r*rr )__name__ __module__ __qualname____doc__r r__annotations__rrrrrrr rr$r%r( classmethodr/r"r"r"r#r s         r operation)labelon_retryfnCallable[[], T]policyr7strr8'Callable[[int, Exception], None] | Nonerc Csd}t|jD]=}z|WStyD}z)|}|dur"|||td|d|j||||jdkr:||WYd}~qd}~ww|pRtd|jd|d)aExecute *fn* with retry logic defined by *policy*. Args: fn: Zero-argument callable to execute. policy: :class:`RetryPolicy` governing retry behaviour. label: Human-readable name for logging. on_retry: Optional callback invoked with ``(attempt, exc)`` before sleep. Returns: The return value of *fn* on success. Raises: The last exception raised by *fn* after all attempts are exhausted. Example:: result = with_retry( lambda: session.get(url, timeout=15), RetryPolicy(max_retries=5), label=f"fetch {symbol}", ) Nz!Attempt %d/%d for '%s' failed: %srzAll z attempts for 'z ' failed.)ranger Exceptionrwarningr$ RuntimeError)r9r;r7r8last_excrexcr"r"r# with_retry\s(   rD) r9r:r;r r7r<r8r=rr )r3 __future__rloggingrrcollections.abcr dataclassesrtypingrindiaopt.constantsrrrr r getLoggerrr r rD__all__r"r"r"r#s"     @ 3