Ej5dZddlmZddlZddlZddlZddlmZmZddl m Z ddl m Z ddl mZddlmZmZmZmZdd lmZddlZdd lmZdd lmZdd lmZmZdd lm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-ddl.m/Z/ddl0m1Z1erddl2m3Z3ddl4m5Z5ddl6m7Z7ddl8m9Z9ejte;Z<GddeZ=e dGdde=Z>GddZ?ded ddZ@y)zBprawcore.sessions: Provides prawcore.Session and prawcore.session.) annotationsN)ABCabstractmethod)deepcopy) dataclass)pformat)IO TYPE_CHECKINGAnyClassVar)urljoin)codes)BaseAuthorizer)TIMEOUT WINDOW_SIZE)BadJSON BadRequestConflictInvalidInvocationNotFoundRedirectRequestExceptionResponseException ServerError SpecialErrorTooLargeTooManyRequestsUnavailableForLegalReasons URITooLong) RateLimiter)authorization_error_class)Mapping)Response)Self) RequestorcNeZdZdZeddZeddZed dZd dZy) RetryStrategyzAn abstract class for scheduling request retries. The strategy controls both the number and frequency of retry attempts. Instances of this class are immutable. cy)Nselfs O/root/workspace/ytshorts/venv/lib/python3.12/site-packages/prawcore/sessions.py_sleep_secondszRetryStrategy._sleep_seconds;s cy)Allow one fewer retry.Nr)r*s r,consume_available_retryz%RetryStrategy.consume_available_retry?r.cy)z*Return ``True`` when a retry should occur.Nr)r*s r,should_retry_on_failurez%RetryStrategy.should_retry_on_failureCr2r.c|jx}2d|dd}tj|tj|yy)z0Sleep until we are ready to attempt the request.Nz Sleeping: z0.2fz seconds prior to retry)r-logdebugtimesleep)r+ sleep_secondsmessages r,r9zRetryStrategy.sleepGsF!002 2M ?"="66MNG IIg  JJ} % @r.Nreturnz float | None)r=r'r=boolr=None) __name__ __module__ __qualname____doc__rr-r1r4r9r)r.r,r'r'2sH  %%99&r.r'T)frozencFeZdZUdZdZded<eZded<d dZd dZd d Z y )FiniteRetryStrategyzIA :class:`.RetryStrategy` that retries requests a finite number of times.z ClassVar[int]DEFAULT_RETRIESintretriesc|j|jkr-|jdkDrdnd}|dtjzzSy)NrrI)rLrJrandom)r+bases r,r-z"FiniteRetryStrategy._sleep_secondsWs? <<$.. . q(1aD!fmmo-- -r.c@t||jdz S)r0rL)typerLr*s r,r1z+FiniteRetryStrategy.consume_available_retry]stDz$,,"233r.c |jdkDS)zEReturn ``True`` if and only if the strategy will allow another retry.rrRr*s r,r4z+FiniteRetryStrategy.should_retry_on_failureas||ar.Nr<)r=rHr>) rBrCrDrErJ__annotations__rLr-r1r4r)r.r,rHrHOs(S%&O]&"GS" 4 r.rHc eZdZUdZej j ej jej jfZ dde de de de de dhZ d e d <ie de e d ee d ee d ee dee de e de e dee dee dee dee dee de e dee dee dede de iZd e d<e de de dhZd e d<ed-dZe d.dZed/dZed0dZd1d Z d2d!Z!e"d" d3d#Z# d4d$Z$ d5d%Z%d&d' d6d(Z&d7d)Z'd8d*Z(d&d&d&d&e)d+ d9d,Z*y&):Sessionz3The low-level connection interface to Reddit's API.ii  bad_gatewaygateway_timeoutinternal_server_errorrequest_timeoutservice_unavailabler RETRY_STATUSES bad_requestconflictfound forbidden media_typemoved_permanently not_foundrequest_entity_too_largerequest_uri_too_largetoo_many_requests unauthorizedunavailable_for_legal_reasonsSTATUS_EXCEPTIONSacceptedcreatedokSUCCESS_STATUSESc|jS)z?Return the :class:`.BaseAuthorizer` used to authorize requests.) _authorizerr*s r, authorizerzSession.authorizersr.ctjd||tjtjdt |tjdt |y)NzFetching: %s %s at %szData: %sz Params: %s)r6r7r8 monotonicrdatamethodparamsurls r, _log_requestzSession._log_requestsB )638HI *gdm, ,0r.c|jS)z9Return the :class:`.RateLimiter` that throttles requests.) _rate_limiterr*s r, rate_limiterzSession.rate_limiters!!!r.cB|jjjS)z;Return the :class:`.Requestor` used to issue HTTP requests.)rp authenticator requestorr*s r,rzSession.requestors--777r.c|S)2Allow this object to be used as a context manager.r)r*s r, __enter__zSession.__enter__s r.c$|jy)rN)close)r+_argss r,__exit__zSession.__exit__s  r. window_sizect|tsd|}t|||_t ||_t |_y)zPrepare the connection to Reddit's API. :param authorizer: An instance of :class:`.Authorizer`. :param window_size: The size of the rate limit reset window in seconds. zinvalid Authorizer: rN) isinstancerrrpr r{rH_retry_strategy_class)r+rqrmsgs r,__init__zSession.__init__sB*n5( 5C#C( (%([A%8"r.c tjd||| |j||||||j|| S)NzRetrying due to %s: %s %s)rufilesjsonrvrwretry_strategy_statetimeoutrx)r6warning_request_with_retriesr1) r+rurrrvrwrstatusrrxs r, _do_retryzSession._do_retrysP /E))!5!M!M!O*  r.c |jjd||||||jj|j|| }t j d|j|jjd|jjd|jjd|jjdtj|S)NF) allow_redirectsrurrrvrwrequest_functionset_header_callbackrrxz?Response: %s (%s bytes) (rst-%s:rem-%s:used-%s ratelimit) at %scontent-lengthzx-ratelimit-resetzx-ratelimit-remainingzx-ratelimit-used) r{callrrequest_set_header_callbackr6r7 status_codeheadersgetr8rs) r+rurrrvrwrrxresponses r, _make_requestzSession._make_requests%%**!!^^33 $ 9 9+   M     !1 2    !4 5    !8 9    !3 4 NN  r.N)rc T||j}|j|j|||| |j|||||||} d} | jtdk(r@|jjt|jdr7| jd} n'| j|j vr| jd} | *|j r|j||||||| || S| jtdk(ry| j|j"vr|j"| j| | j|j$vr t'| | j(j+dd k(ry | j-S#t$rh} |j rRt | j|jr2|j||||||t| j|| cYd} ~ Sd} ~ wwxYw#t.$r t1| dwxYw) Nrtrurrrvrwrrx) rurrrvrwrrrrxrhrefreshz status no_contentr0)rr9ryrrr4roriginal_exceptionRETRY_EXCEPTIONSrreprrrrp_clear_access_tokenhasattrr]rjrnrrrr ValueErrorr) r+rurrrvrwrrrxr exception retry_statuss r,rzSession._request_with_retriessU '#'#=#=#? ""$ tF6sK ))*H6   5#8 8    0 0 2t''3"*"6"6!7w?  ! !T%8%8 8&2237;L  #(<(T(T(V>>%9#"     5#6 6   4#9#9 9>$(()=)=>xH H   t'<'< <#H- -      0 1S 8 .==? "a  #;;=*,,%%C~~!!)= < <=#&   ! b .(# - .s0F H H&AH HH  HH'ct|jdd}|jjs ||dd|jjiS)Nr Authorizationzbearer )getattrrpis_valid access_token)r+refresh_methods r,rzSession._set_header_callbackPsR !1!19dC((*~/I  74+;+;+H+H*I!JKKr.c8|jjy)z+Close the session and perform any clean up.N)rrr*s r,rz Session.closeVs r.)rurrrwrc d|rtt|ni}d|d<t|tr*t|}d|d<t|j }n|}t|trt|}d|d<t |j j|} |j||||||| S)aReturn the json content from the resource at ``path``. :param data: Dictionary, bytes, or file-like object to send in the body of the request. :param files: Dictionary, mapping ``filename`` to file-like object. :param json: Object to be serialized to JSON in the body of the request. :param method: The request verb. E.g., ``"GET"``, ``"POST"``, ``"PUT"``. :param params: The query parameters to send with the request. :param path: The path of the request. This path will be combined with the ``oauth_url`` of the Requestor. :param timeout: Specifies a particular timeout, in seconds. Automatically refreshes the access token if it becomes invalid and a refresh token is available. :raises: :class:`.InvalidInvocation` in such a case if a refresh token is not available. rQraw_jsonrapi_typer) rdictrsorteditemsr r oauth_urlr) r+rurrrvrwpathr data_listrxs r,rzSession.requestZs<,2$v,'rz dD !D>D%D tzz|,II dD !D>D%D dnn..5))*  r.)r=r) ru7list[tuple[str, object]] | bytes | IO[Any] | str | Nonervstrrwdict[str, object]rxrr=rA)r=r )r=r%)r=r$)robjectr=rA)rqBaseAuthorizer | NonerrKr=rA)rurrdict[str, IO[Any]] | Noner'dict[str, object] | list[object] | NonervrrwrrrHrrrfloatrxrr=dict[str, object] | str | None)rurrrrrrvrrwrrrrxrr=r#)rurrrrrrvrrwrrzFiniteRetryStrategy | Nonerrrxrr=r)r=zdict[str, str]r@)ruz0dict[str, object] | bytes | IO[Any] | str | NonerrrrrvrrwzMapping[str, object] | Nonerrrrr=r)+rBrCrDrErequests exceptionsChunkedEncodingErrorConnectionError ReadTimeoutrrr]rUrrrrr!rrrrrrrjrnpropertyrq staticmethodryr|rrrrrrrrrrrrr)r.r,rWrWfs4= 00++''   m   %&   #$ NH# mk# mj# j8# g # k5 #  + # %& # l\# !"H# kH# ()8# %& # #$k# !"O# n8#  + %%#& ['#( [)#x,#( "3U95EuT{!ShS    1E 1 1" 1  1  1 1""88' 9*9 9  9& F )  6    " 2     ( 2 F )  6   "     T<@N.FN.) N. 6 N.  N."N.9N.N.N. (N.`L BF+/8<.2 2 ?2 ) 2 6 2  2 ,2 2 2  (2 r.rWrqrct||S)zReturn a :class:`.Session` instance. :param authorizer: An instance of :class:`.BaseAuthorizer`. :param window_size: The size of the rate limit reset window in seconds. r)rWrs r,sessionrs jk BBr.)rqrrrKr=rW)ArE __future__rloggingrNr8abcrrcopyr dataclassesrpprintrtypingr r r r urllib.parser requests.exceptionsrrequests.status_codesr prawcore.authrprawcore.constrrprawcore.exceptionsrrrrrrrrrrrrrrprawcore.rate_limitr prawcore.utilr!collections.abcr"requests.modelsr#typing_extensionsr$prawcore.requestorr% getLogger __package__r6r'rHrWrr)r.r,rsH" #!33 '(/ ,3'(&,g $&C&: $ -  ,f f V )-" C% C C Cr.