o QDi!@sddlZddlmZmZddlmZmZmZmZmZddl m Z m Z ddl m Z mZmZddlmZddlmZdd lmZmZmZmZmZGd d d eZGd d d eZeZGdddeZdS)N)datetimetimezone)OptionalSequenceTupleTypeUnion) PositiveFloat PositiveInt)RetryCallStatewait_exponentialwait_random_exponential) wait_base)HTTP_DATE_FORMAT)"get_default_http_status_exceptionsget_default_network_errorsget_default_timeoutsis_rate_limitedis_server_errorc @sdeZdZdZdefdedeeedfde e ddfdd Z d e de fd d Zd e de fd dZdS)wait_from_headeraWait strategy that derives the wait value from an HTTP header. Value may be either an integer representing the number of seconds to wait before retrying, or a future datetime in HTTP-date format, indicating when it is acceptable to retry the request. More info on HTTP-date format: https://httpwg.org/specs/rfc9110.html#http.date Args: header (str): Header to attempt to derive wait value from. wait_max (float): Maximum time to wait, in seconds. Defaults to 120.0s. If `None` is given, will wait indefinitely. Use `Non` with caution, as your program will hang if the server responds with an excessive wait value. fallback (wait_base): Wait strategy to use if `header` is not present, or unable to parse to a `float` value, or if value parsed from header exceeds `wait_max`. Defaults to `tenacity.wait_exponential`. Raises: ValueError: If `fallback` is `None`, and any one of the following is true: * header is not present; * the value cannot be parsed to a `float`; * the value exceeds `wait_max`; * the value is a date in the past. ^@headerwait_maxNfallbackreturncCs"||_|r t|nd|_||_dSN)rfloatrr)selfrrrrH/var/www/Datamplify/venv/lib/python3.10/site-packages/retryhttp/_wait.py__init__.s zwait_from_header.__init__ retry_statecCs|jr`|j}|durdSt|tr`t|dr`|jj|j}|dur-t d|jt d|r7t |St |t}|jtjd}t tj}||krYt d|jd|t ||jSt d |jd ) a=Attempts parse a wait value from header. Args: retry_state (RetryCallState): The retry call state of the request. Returns: float: Seconds to wait, as derived from `self.header`. Raises: ValueError: If unable to parse a float from `self.header`. NrresponsezHeader not present: z^\d+$)tzinfozDate provided in header "z" is in the past: z(Unable to parse wait time from header: "")outcome exception isinstancerhasattrr#headersgetr ValueErrorrematchrrstrptimerreplacerutcnowseconds)rr"excvalue retry_afterr2rrr _get_wait_value8s.     z wait_from_header._get_wait_valuecCs|jr+z|j|d}|jr||jkr|j|dWS|WSty*|j|dYSw|j|d}|jrI||jkrItd|jd|d|jd|S)N)r"zWait value parsed from header "z" (z) is greater than `wait_max` ())rr7rr,r)rr"r5rrr __call__]s"   zwait_from_header.__call__)__name__ __module__ __qualname____doc__r strrr r rrr!r rr7r9rrrr rs   %rcsDeZdZdZdefdeeedfdee ddffdd Z Z S) wait_retry_afteraWait strategy to use when the server responds with a `Retry-After` header. The header value may provide a date for when you may retry the request, or an integer, indicating the number of seconds to wait before retrying. Args: wait_max (float): Maximum time to wait, in seconds. Defaults to 120.0s. If `None` is given, will wait indefinitely. Use `Non` with caution, as your program will hang if the server responds with an excessive wait value. fallback (wait_base): Wait strategy to use if `header` is not present, or unable to parse to a `float` value, or if value parsed from header exceeds `wait_max`. Defaults to `tenacity.wait_exponential()`. Raises: ValueError: If `fallback` is `None`, and any one of the following is true: * `Retry-After` header is not present; * the value cannot be parsed to a `float`; * the value exceeds `wait_max`; * the value is a date in the past. rrNrrcstjd||ddS)Nz Retry-After)rrr)superr!)rrr __class__rr r!szwait_retry_after.__init__) r:r;r<r=r rr r rrr! __classcell__rrrAr r?ps r?c@seZdZdZeedeeeeddddfdedededed ee e e fd ee e e e e d fdfd ee e e e e d fdfd dfddZded efddZdS)wait_context_awarea Uses a different wait strategy based on the type of HTTP error. Args: wait_server_errors: Wait strategy to use with server errors. wait_network_errors: Wait strategy to use with network errors. wait_timeouts: Wait strategy to use with timeouts. wait_rate_limited: Wait strategy to use when rate limited. server_error_codes: One or more 5xx HTTP status codes that will trigger `wait_server_errors`. network_errors: One or more exceptions that will trigger `wait_network_errors`. If omitted, defaults to: - `httpx.ConnectError` - `httpx.ReadError` - `httpx.WriteError` - `requests.ConnectionError` - `requests.exceptions.ChunkedEncodingError` timeouts: One or more exceptions that will trigger `wait_timeouts`. If omitted, defaults to: - `httpx.ConnectTimeout` - `httpx.ReadTimeout` - `httpx.WriteTimeout` - `requests.Timeout` )r)iiiiNwait_server_errorswait_network_errors wait_timeoutswait_rate_limitedserver_error_codesnetwork_errors.timeoutsrcCsJ|durt}|durt}||_||_||_||_||_||_||_dSr) rrrErFrGrHrIrJrK)rrErFrGrHrIrJrKrrr r!s zwait_context_aware.__init__r"cCsj|jr3|j}t||jdr||St||jr||St||jr*| |St |r3| |SdS)N)r4 status_codesr) r&r'rrIrEr(rJrFrKrGrrH)rr"r4rrr r9s       zwait_context_aware.__call__)r:r;r<r=r?r r rrrintr BaseExceptionrr!r rr9rrrr rDsF   rD)r-rrtypingrrrrrpydanticr r tenacityr r r tenacity.waitr _constantsr_utilsrrrrrrr?rHrDrrrr s   ]