PUh8$ddlmZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z m Z ddlmZddlmZddlmZdd lmZej,eZGd d ej2 ZGd deZddZddZddZGdde Zy)) annotationsN) HTTPStatus)Any)ResponseSession)ConnectionError)CaseInsensitiveDict) TypeGuard) JIRAErrorc>eZdZdZej ddZy)PrepareRequestForRetryzThis class allows for the manipulation of the Request keyword arguments before a retry. The :py:meth:`.prepare` handles the processing of the Request keyword arguments. c|S)a)Process the Request's keyword arguments before retrying the Request. Args: original_request_kwargs (CaseInsensitiveDict): The keyword arguments of the Request. Returns: CaseInsensitiveDict: The new keyword arguments to use in the retried Request. )selforiginal_request_kwargss M/var/www/timesheet/venv/lib/python3.12/site-packages/jira/resilientsession.pypreparezPrepareRequestForRetry.prepares '&Nrr returnr )__name__ __module__ __qualname____doc__abcabstractmethodrrrrr r s1   '': '  ' 'rr ) metaclassc,eZdZdZ dfd ZxZS)PassthroughRetryPreparezbReturns the Request's keyword arguments unchanged, when no change needs to be made before a retry.c"t||S)N)superr)rr __class__s rrzPassthroughRetryPrepare.prepare-sw677rr)rrrrr __classcell__r"s@rrr*s l8':8 88rrc |jdd}| tdd|i||js1t|}t|f|j|j ||d|y)a.Handle errors from a Jira Request. Args: resp (Optional[Response]): Response from Jira request Raises: JIRAError: If Response is None JIRAError: for unhandled 400 status codes. Returns: TypeGuard[Response]: True if the passed in Response is all good. requestNresponse)resp) status_codeurlr&r'T)zEmpty Response!)getr okparse_error_msgr)r*)r(kwargsr&errors rraise_on_errorr03swjjD)G |CDCFCC 77T*  ((      rc,i}g}|jdk(rd|jvr|jdgS|jr |j}d|vr|dg}|Sd|vr|dg}|Sd|vr;|d}t |dkDr&t |ttfr t|}|S|g}|Sd|vrI|d}t |dkDr6t |tr&|jDcgc] }t|}}|S#t$r|jgcYSwxYwcc}w)a;Parse a Jira Error messages from the Response. https://developer.atlassian.com/cloud/jira/platform/rest/v2/intro/#status-codes Args: resp (Response): The Jira API request's response. Returns: List[str]: The error messages list parsed from the Response. An empty list if no error. izx-authentication-denied-reasonmessage errorMessage errorMessagesrerrors) r)headerstextjson ValueErrorlen isinstancelisttupledictvaluesstr)r( resp_data parsed_errorserror_messages resp_errorserrs r parse_errorsrFTs[!#I!M 3#Ct||#S =>??   II"9-. * ) 9 $#>23 " ! I %#?3 ~  ".4-8 $^ 4  "0 0   Y ) { a J{D$A2=1C1C1EF#SXFMF 5 II;  0GsC5D5D Dc:t|}dj|S)aEParse a Jira Error messages from the Response and join them by comma. https://developer.atlassian.com/cloud/jira/platform/rest/v2/intro/#status-codes Args: resp (Response): The Jira API request's response. Returns: str: The error message parsed from the Response. An empty str if no error. z, )rFjoin)r(r5s rr-r-s$ F 99V rceZdZdZdd fd Zd dZef d fd Zd dZ d ddZ d dZ xZ S)ResilientSessionzThis class is supposed to retry requests that do return temporary errors. :py:meth:`__recoverable` handles all retry-able errors. c||_||_||_t||j j dditjddtdzy)aA Session subclass catered for the Jira API with exponential delaying retry. Args: timeout (Optional[Union[Union[float, int], Tuple[float, float]]]): Connection/read timeout delay. Defaults to None. max_retries (int): Max number of times to retry a request. Defaults to 3. max_retry_delay (int): Max delay allowed between retries. Defaults to 60. Acceptzapplication/json,*/*;q=0.9z@WARNING: On error, will dump Response headers and body to logs. zLog level debug in 'z"' is not safe for production code!N) timeout max_retriesmax_retry_delayr!__init__r6updateLOGdebugr)rrMrNrOr"s rrPzResilientSession.__init__se &.  X'CDE N$XJ.PQ R rc D|j}||jj}|j|jdi||d<|jdd}t |t rt j||d<d|vr|j|d<|S)z?Do any pre-processing of our own and return the updated kwargs.r6dataNverify) copyr6rQr+r;r>r8dumpsrV)roriginal_kwargsprepared_kwargsrequest_headersrUs r _jira_preparezResilientSession._jira_prepares)..0 ,,++-229bAB%4 """640 dD !'+jj&6OF # ? *(, OH %rc  d d}d}jdi|}d fd }|rd}d} t ||fdji|}|jrj ||S ||n|} dz |r4j| ||j r|j|njdz |r||t|fi|r|Std#t $r } | }Yd} ~ d} ~ wwxYw) a This is an intentional override of `Session.request()` to inject some error handling and retry logic. Raises: Exception: Various exceptions as defined in py:method:`raise_on_error`. Returns: Response: The response. rNc"jkS)z4Helper method to say if we should still be retrying.)rN) retry_numberrsris_allowed_to_retryz5ResilientSession.request..is_allowed_to_retrys4#3#33 3rrMz*Expected a Response or Exception to raise!r)rbool) r\r!r&rMr,2_ResilientSession__handle_known_ok_response_errorsr_ResilientSession__recoverableupperrrNr0 RuntimeError) rmethodr*_prepare_retry_classr. exceptionr'processed_kwargsr`eresponse_or_exceptionr_r"s ` @rr&zResilientSession.requests9 &* $(.4--77 4"#HI  7?C)-9I;;::8D#O190DH) ! A L"$););%sFLLNL*%,,-=>#//!3 /"#2  O H 9(8 9OKL L+#   s;C&& C;/C66C;c|jsyt|jdk(r7d|jvr(d|jdvrtj dyyyy)zResponses that report ok may also have errors. We can either log the error or raise the error as appropriate here. Args: response (Response): The response. NrzX-Seraph-LoginReasonAUTHENTICATED_FAILEDz;Atlassian's bug https://jira.atlassian.com/browse/JRA-41559)r,r:contentr6rRwarning)rr's r!__handle_known_ok_response_errorsz2ResilientSession.__handle_known_ok_response_errorss`{{    !Q &&(*:*::&(*:*:;Q*RR KKU VS; 'rcd}t|}t|trndd|zz}tj d|d|j d|d|dztj tjkDrtj d nt|trtjtjg}|j|vr|jjd }|rdt!t#|d z}n%|jtjk(rdd|zz}|jtjk(r,|jd|j$}|j'||d kD} | rt)|j*|t-j.d dz} tj d|d|d|d|j0d| d| t|tr^tj3dt5j6t9|jdtj3d|j:t=j>| | S)aReturn whether the request is recoverable and hence should be retried. Exponentially delays if recoverable. At this moment it supports: 429, 503 Args: response (Optional[Union[ConnectionError, Response]]): The response or exception. Note: the response here is expected to be ``not response.ok``. url (Union[str, bytes]): The URL. request_method (str): The request method. counter (int, optional): The retry counter to use when calculating the exponential delay. Defaults to 1. Returns: bool: True if the request should be retried. zGot ConnectionError [z] errno:z on   zJResponse headers for ConnectionError are only printed for log level DEBUG. Retry-Afterrarg?g?zGot recoverable error from z, will retry [/z] in zs. Err: zresponse.headers: %s)indentzresponse.body: %s) r@r;rrRrperrnolevelloggingDEBUGrrTOO_MANY_REQUESTSSERVICE_UNAVAILABLEr)r6r+maxintreason(_ResilientSession__log_http_429_responseminrOrandomuniformrNrSr8rXr>rotimesleep) rr'r*request_methodcountersuggested_delaymsgrecoverable_error_codes retry_afteris_recoverabledelays r __recoverablezResilientSession.__recoverables8.(m h 0 1g:oO KK'z(..9InM]]^_E* yy7==( `( +,,..' # ##'>>&..22=A &'#K(!+'O))Z-I-II&(1g:oO'':+G+GG%112!HOO3DEC00:(1, ,,o>SBE KK-n-=Qse>RYQZZ[\`\l\l[mmrsxryzBCFBGH (H- +JJtH$4$45a@ .0@0@A JJu rch|jjd}|jjd}|jjd}|jjd}d}||d|dndz }| | ||d |dz }| |d |d z }|d zd z}tj|y)NrxzX-RateLimit-FillRatezX-RateLimit-Interval-SecondszX-RateLimit-LimitzRequest rate limited by Jira.z! Request should be retried after z seconds. rwz tokens are issued every zYou can accumulate up to z tokens. z;Consider adding an exemption for the user as explained in: zohttps://confluence.atlassian.com/adminjiraserver/improving-instance-stability-with-rate-limiting-983794911.html)r6r+rRrp)rr'r$number_of_tokens_issued_per_interval#token_issuing_rate_interval_secondsmaximum_number_of_tokens warning_msgs r__log_http_429_responsez(ResilientSession.__log_http_429_response[s&&**=9 /7/?/?/C/C "0 ,/7.>.>.B.B */ +$,#3#3#7#78K#L 5 &0 }K H  1 <3? BCC\^A]BBMN NK # / +,D+EZP K K L @  K r)N<)rNrrOr)rr>)rgr@r* str | bytesrhr rr)r'r)ra)r'z!ConnectionError | Response | Noner*rrr@rr) rrrrrPr\rr&rcrdrr#r$s@rrJrJs  ,.8O7P ;M;M;M5 ;M  ;MzW, J3JJ J  JX"!rrJ)r(zResponse | NonerzTypeGuard[Response])r(rrz list[str])r(rrr@) __future__rrr8r~rrhttprtypingrrequestsrrrequests.exceptionsrrequests.structuresr typing_extensionsr jira.exceptionsr getLoggerrrRABCMetar rr0rFr-rJrrrrsz"  &/3'%g!'s{{'*848B,^ k!wk!r