K.h8*ddlmZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z m Z ddlmZddlmZddlmZdd lmZejeZGd d ej ZGd deZddZddZddZGdde ZdS)) annotationsN) HTTPStatus)Any)ResponseSession)ConnectionError)CaseInsensitiveDict) TypeGuard) JIRAErrorc4eZdZdZejddZdS)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. original_request_kwargsr returnc|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. )selfrs g/Users/user/workspace/sujinbaek/cqa-test-app/venv/lib/python3.11/site-packages/jira/resilientsession.pypreparezPrepareRequestForRetry.prepares '&Nrr rr )__name__ __module__ __qualname____doc__abcabstractmethodrrrrr r sB   ' ' ' ' ' 'rr ) metaclassc$eZdZdZdfd ZxZS)PassthroughRetryPreparezbReturns the Request's keyword arguments unchanged, when no change needs to be made before a retry.rr rcFt|S)N)superr)rr __class__s rrzPassthroughRetryPrepare.prepare-sww6777rr)rrrrr __classcell__r"s@rrr*sCll8888888888rrrespResponse | NonerTypeGuard[Response]c |dd}|tdd|i||js,t|}t|f|j|j||d|dS) 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. requestNEmpty Response!response)r%) status_codeurlr)r+T)r*)getr okparse_error_msgr,r-)r%kwargsr)errors rraise_on_errorr33sjjD))G |CCDCFCCC 7  T***  (        4rr list[str]c2i}g}|jdkrd|jvr|jdgS|jr. |}n#t$r |jgcYSwxYwd|vr |dg}d|vr |dg}d|vrK|d}t |dkr0t |ttzrt|}n|g}d|vrN|d}t |dkr3t |trd| D}|S) 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 errorMessagesrerrorsc,g|]}t|Sr)str).0errs r z parse_errors..~sFFF#SXXFFFr) r,headerstextjson ValueErrorlen isinstancelisttupledictvalues)r% resp_data parsed_errorserror_messages resp_errorss r parse_errorsrMTsm!#I!M 3#Ct|#S#S =>??   II   I;    I"9-. ""#>23 )###?3 ~   " ".$,77 1 $^ 4 4 !/ 0 9) {  a  J{D$A$A GF1C1C1E1EFFFM sAAAr;cJt|}d|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, )rMjoin)r%r9s rr0r0s#$  F 99V  rcjeZdZdZddfd Zdd Zefdfd Zd dZ d!d"dZ d dZ xZ S)#ResilientSessionzThis class is supposed to retry requests that do return temporary errors. :py:meth:`__recoverable` handles all retry-able errors. N< max_retriesintmax_retry_delayc||_||_||_t|jdditddtdzdS)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) timeoutrTrVr!__init__r?updateLOGdebugr)rrYrTrVr"s rrZzResilientSession.__init__s &.  X'CDEEE NQXQQQ R     rrrGc d|}||j}||di||d<|dd}t |t r|rt j||d<d|vr |j|d<|S)z?Do any pre-processing of our own and return the updated kwargs.r?dataNverify) copyr?r[r.rDrGrAdumpsr`)roriginal_kwargsprepared_kwargsrequest_headersr_s r _jira_preparezResilientSession._jira_prepares)..00 ),++--229bAABBB%4 """6400 dD ! ! 7d 7'+j&6&6OF # ? * *(, OH %rmethodr;r- str | bytes_prepare_retry_classr rc $ d d}d}jd i|}d fd }|rd}d} tj||fdji|}|jr||Sn#t $r } | }Yd} ~ nd} ~ wwxYw||n|} dz |r@| || r| |n j dz |||t|fi|r|Std) 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. rNrboolcjkS)z4Helper method to say if we should still be retrying.)rT) retry_numberrsris_allowed_to_retryz5ResilientSession.request..is_allowed_to_retrys4#33 3rrYz*Expected a Response or Exception to raise!r)rrk) rfr!r)rYr/2_ResilientSession__handle_known_ok_response_errorsr_ResilientSession__recoverableupperrrTr3 RuntimeError) rrgr-rir1 exceptionr+processed_kwargsrneresponse_or_exceptionrmr"s ` @rr)zResilientSession.requests &* $(.4-7777 4 4 4 4 4 4 4"!## 4HI *577?C)-9I;$::8DDD#O$#     190DHH) ! A L""$$ 4););%sFLLNNL** 4%,,-=>>>>#/!3 /"!## 42  O H 9 9(8 9 9 MOKLL Ls?A.. B8A??Br+c|jsdSt|jdkr4d|jvr-d|jdvr tddSdSdSdS)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/rCcontentr?r\warning)rr+s r!__handle_known_ok_response_errorsz2ResilientSession.__handle_known_ok_response_errorssx{  F  ! !Q & &&(*:::&(*:;Q*RRR KKU V V V V V ' &::RRrro!ConnectionError | Response | Nonerequest_methodcountercd}t|}t|trldd|zz}td|d|jd|d|dztjtjkrtd nt|trtj tj g}|j |vr|jd }|r!dt!t#|d z}n|j tj krdd|zz}|j tj kr&|j d|j}|||d k} | rt)|j|t-jd dz} td|d|d|d|jd| d| t|tratdt5jt9|jdtd|jt=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-Afterrorg?g?zGot recoverable error from z, will retry [/z] in zs. Err: zresponse.headers: %s)indentzresponse.body: %s) r;rDrr\r{errnolevelloggingDEBUGrrTOO_MANY_REQUESTSSERVICE_UNAVAILABLEr,r?r.maxrUreason(_ResilientSession__log_http_429_responseminrVrandomuniformrTr]rArbrGrztimesleep) rr+r-r~rsuggested_delaymsgrecoverable_error_codes retry_afteris_recoverabledelays r __recoverablezResilientSession.__recoverables.(mm h 0 0 ; 1g:oO KK___(.__n___***   y7=(( `( + + ;,.' # #'>>>&.22=AA 6&'#K((!++'OO)Z-III&(1g:oO':+GGG%1EEHOEEC00:::(1,  ,o>>SBBE KKHnHHsHHRYHH\`\lHHsxHHCFHH   (H-- B +JtH$455a@@@ .0@AAA Ju   rcl|jd}|jd}|jd}|jd}d}||d|dndz }| | ||d |dz }| |d |d z }|d zd z}t|dS)NrzX-RateLimit-FillRatezX-RateLimit-Interval-SecondszX-RateLimit-LimitzRequest rate limited by Jira.z! Request should be retried after z seconds. rz 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)r?r.r\r{)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.&**=99 /7/?/C/C "0 0 ,/7.>.B.B */ / +$,#3#7#78K#L#L 5 & I H H H H  1 <3? BNN^ANNN NK # / P,DPPP K K L @  K     r)NrRrS)rTrUrVrU)rrG)rgr;r-rhrir rr)r+r)ro)r+r}r-rhr~r;rrU) rrrrrZrfrr)rprqrr#r$s@rrQrQs        ,.8O7N7P7P ;M;M;M;M;M;M;MzWWWW, JJJJJX"!"!"!"!"!"!"!"!rrQ)r%r&rr')r%rrr4)r%rrr;) __future__rrrArrrhttprtypingrrequestsrrrequests.exceptionsrrequests.structuresr typing_extensionsr jira.exceptionsr getLoggerrr\ABCMetar rr3rMr0rQrrrrs""""""  &&&&&&&&//////333333''''''%%%%%%g!!'''''s{''''*888884888B,,,,^    k!k!k!k!k!wk!k!k!k!k!r