6i~FddlZddlZddlZddlZddlZddlZddlmZmZddl m Z ddl m Z m Z m Z mZddlmZmZmZmZddlmZddlmZmZmZddlmZmZmZmZmZm Z d d l!m"Z"m#Z#d d l$m%Z%d d l&m'Z'm(Z(erdd l)m*Z*ddl+m,Z,m-Z-ej\e/Z0dZ1dZ2dZ3dZ4dZ5dZ6Gdde Z7GddeZ8Gdde Z9GddeZ:Gdde Z;Gdd eZ<Gd!d"e Z=Gd#d$eZ>Gd%d&e>Z?Gd'd(e>Z@Gd)d*eZAy)+N)SEEK_SETUnsupportedOperation)time)AnyDictOptional TYPE_CHECKING) parse_qsl urlencodeurlparse urlunparse)format_date_time) AzureErrorServiceRequestErrorServiceResponseError)BearerTokenCredentialPolicy HeadersPolicy HTTPPolicyNetworkTraceLoggingPolicyRequestHistorySansIOHTTPPolicy)AzureSigningErrorStorageHttpChallenge)DEFAULT_OAUTH_SCOPE) LocationModeStorageErrorCode)TokenCredential)PipelineRequestPipelineResponsect|tr|jd}tj|}|j dS)Nzutf-8) isinstancestrencodebase64 b64encodedecode)dataencodeds f/home/azureuser/techstart-app/venv/lib/python3.12/site-packages/azure/storage/blob/_shared/policies.py encode_base64r+1s9${{7#t$G >>' ""cr|d|d|d|df}ttd|}|syt|dkS)NtotalconnectreadstatusFr)listfiltermin)settings retry_countss r* is_exhaustedr79sLW%x ':HV|dr|dd|ddz |dd|yy)Nhookcountrmode) retry_count location_moder>)r5kwargss r* retry_hookr@As9eXg%6%:(SYJZe^der,cZ|jj}d|cxkrdkrnn||dk(r|tjk(ry|dk(ry|dk\rW|jjj d}|t jt jt jfvryy|dk\r|d vryyy) Ni,iiTiizx-ms-copy-source-error-codeF)ii) http_response status_coder SECONDARYheadersgetrOPERATION_TIMED_OUTINTERNAL_ERROR SERVER_BUSY)responser;r1 error_codes r*is_retryrLKs  # # / /F fs S=T\%;%;; S= S=!//77;;_ ?F ==?r,ceZdZdZy)QueueMessagePolicyc|jjjdd}|r0t|jj ||j_yy)Nqueue_message_id)rPoptionspopr_rQurl)selfrequest message_ids r* on_requestzQueueMessagePolicy.on_request{sH__,,001CTJ '.w/C/C/G/G'TG $ r,N)__name__ __module__ __qualname__rjr>r,r*raraysUr,rac$eZdZdZdfd ZxZS)StorageHeadersPolicyx-ms-client-request-idc@tt| |tt }||j j d<|jjjdd}|xsttj|j j d<y)Nz x-ms-dateclient_request_idrp) superrorjrrrQrErPrdrer#uuiduuid1)rgrh current_time custom_id __class__s r*rjzStorageHeadersPolicy.on_requests} "D4W='/ 4@$$[1OO++//0CTJ AJA_cRVR\R\R^N_$$%=>r,rhrreturnN)rkrlrmrequest_id_header_namerj __classcell__rxs@r*roros5``r,roc(eZdZdfd ZddZxZS) StorageHostsc 8||_tt|yN)hostsrsr__init__)rgrr?rxs r*rzStorageHosts.__init__s  lD*,r,c|j|jjd<t|jj }t j}|jjD]\}}|j|k(s|}|jjjdd}|rzd|jjd<||jvrtd|||k7r@|j|j|}|j|j_|}||jjd<y)Nr use_locationFretry_to_secondaryz*Attempting to use undefined host location netlocr=)rrPrdr rQrfrPRIMARYitemsrre ValueErrorrZr[)rgrh parsed_urlr=keyvaluerupdateds r*rjzStorageHosts.on_requests+/::(g22667 %,, ****, $JC  E) #  $ ..22>4H >+;$$( , 3@0r,rryrkrlrmrrjr|r}s@r*rrs-Ar,rc>eZdZdZddeddffd Zd dZd dZxZS) StorageLoggingPolicyzA policy that logs HTTP request and response to the DEBUG logger. This accepts both global configuration, and per-request level with "enable_http_logger" logging_enablerzNc ^|jdd|_tt|dd|i|y)N logging_bodyFrr>)rerrsrr)rgrr?rxs r*rzStorageLoggingPolicy.__init__s/"JJ~u= "D2[.[TZ[r,c |j}|jj}|jxs|j dd|_|j d|j rd|jd<t jtjsy |j}|j}d|vr|j|dd}t jd|t jd|jt jd |jj!D]\}}|j#d k(rd }n\|j#d k(rId|vrEt%|\}} } } } } t't)| }d |d<t+|| | | t-|| f}t jd ||t jd|jr)t jt/|j0yt jdyy#t2$r }t jd|Yd}~yd}~wwxYw)NrFrTsigz sig=*****zRequest URL: %rzRequest method: %rzRequest headers: authorizationz*****zx-ms-copy-source %r: %rz Request body:z1Hidden body, please use logging_body to show bodyzFailed to log request: %r)rQrPrdrreenable_http_logger_LOGGER isEnabledForloggingDEBUGrfqueryreplacedebugmethodrErlowerr dictr r r r#rT Exception)rgrhrQrdlog_url query_paramsheaderrschemerrYparamsrfragment parsed_qserrs r*rjzStorageLoggingPolicy.on_requests++ //)) --S^U1S ;;')@)@ A04GOO, -'' 6 @&**+11 L(%ool5.A;OG /9 2L4G4GH 01%1%9%9%?%?%A ?MFE||~8 '+==%5.HPQVEfeX$(5)9$: +2 %(!+FFD&)T]J^`h+i jMM,> ? o.$$MM#l&7&7"89MM"UVA BB @ 93?? @sE0H! H!! I *II c~|jjd|jr=tj t j sy tjd|jjtjd|jjjD]\}}tjd||tjdtjdtj}|jjjd}|jjjdd }|r<|j!|r+|j#d d }tjd |nM|j%d rtjdn&|j'drtjd|j(r?|j'dr.tj|jj+y|j(r/ tj|jj-yyy#t.$rtjdYywxYw#t0$r)} tjdt3| Yd} ~ yd} ~ wwxYw)NrzResponse status: %rzResponse headers:rzResponse content:zattachment; ?filename=["\w.]+zcontent-dispositionz content-type=zFile attachments: %sz octet-streamzBody contains binary data.imagezBody contains image data.textzBody is streamablezFailed to log response: %s)rPrerrrrrrrBrCrErrecompile IGNORECASErFmatch partitionendswith startswithrrrTrrrepr) rgrhrJ res_headerrpatternrresp_content_typefilenamers r* on_responsez StorageLoggingPolicy.on_responses      0$2I2I J'' 6 G 3X5K5K5W5WX 12)1)?)?)G)G)M)M)OC%JMM, EBC 12**%Er}}U!//77;;?&11':MM"=>$$):)E)Ef)MMM("8"8"="="?@&&< h&<&<&A&A&CD'5 K:&< &:;< G :DIFF Gs= GJ ) J 6-I&&JJ JJ J<J77J<)FryrhrrJr rzN) rkrlrm__doc__boolrrjrr|r}s@r*rrs+ \t\$\&@P"Gr,rc&eZdZfdZddZxZS)StorageRequestHookc V|jd|_tt|yNraw_request_hook)rF_request_callbackrsrrrgr?rxs r*rzStorageRequestHook.__init__s#!',>!?  $02r,cz|jjjd|j}|r ||yyr)rPrdrer)rgrhrequest_callbacks r*rjzStorageRequestHook.on_requests6"??22667I4KaKab  W % r,ryrr}s@r*rrs 3&r,rc&eZdZfdZddZxZS)StorageResponseHookc V|jd|_tt|y)Nraw_response_hook)rF_response_callbackrsrrrs r*rzStorageResponseHook.__init__!s#"(**-@"A !413r,cl|jjd}|&|jjjdd}|jjd}|&|jjjdd}|jjd}|&|jjjdd}|jjdxs0|jjjd|j}|j j |}t||jjjdxs t|}|jjdk(}|xs| } | r||t|jjjdd z }||jjjd } | r2t| jd d d jd d d }n9|}n6| r4|2|t|jjjdd z }||fD]<} t| ds|| jd<|| jd<|| jd<>|r||||jd<|S)Ndata_stream_totaldownload_stream_currentupload_stream_currentresponse_callbackrr;izContent-Lengthrz Content-Range rrXrP)rPrFrdrernextsendrLrVrBrCintrEsplitrQhasattr) rgrhrrrrrJ will_retry is_auth_errorshould_update_counts content_range pipeline_objs r*rzStorageResponseHook.send%s#OO//0CD  $ ' 7 7 ; ;>'*h(?(?(C(CF(KLkPabjPk  ..::cA $.$?-@ $;$G #s8+A+A+I+I+M+MN^`a+b'c c # ( ( 6 6 > > B B? S  (+M,?,?Q,G,J,P,PQTVW,XYZ,[(\%(?% !&;&G !S)>)>)F)F)J)JK[]^)_%` ` !$h/ VL|Y/?@U $$%<=  V  h '3DGOO/ 0r,)rhrrzr )rkrlrmrrr|r}s@r*rrs 4)r,rcPeZdZdZdZdeddffd ZedZd dZ d d Z xZ S) rRzA simple policy that sends the given headers with the request. This will overwrite any headers already defined in the request. z Content-MD5r?rzNc *tt| yr)rsrRrrs r*rz!StorageContentValidation.__init__Zs &68r,cxsdtj}ttr!|j |jSt dr_d} j }tfddD]}|j | j|t|jStd#Y[xYw#ttf$r}td|d}~wwxYw)Nr,r0rc&jdS)Ni)r0)r(sr*z:StorageContentValidation.get_content_md5..ksdiior,z4Data should be bytes or a seekable file-like object.)hashlibmd5r"bytesupdatertelliterseekrAttributeErrorIOErrorrdigest)r(rposchunkexcs` r*rSz(StorageContentValidation.get_content_md5]s{skkm dE " JJt  zz|T6 "C iik5s; " 5! " b #x( zz|ST T  #G, b !WX^aa bs$CC C C* C%%C*cl|jjjdd}|r}|jjdk7rdt t j|jj}||jj|j<||jd<||jd<y)NrNFGETvalidate_content_md5) rPrdrerQrr+rRrSr(rE header_name)rgrhrNrUs r*rjz#StorageContentValidation.on_requestvs"??22667I5Q  4 4 ; ;u D()A)Q)QRYRfRfRkRk)lmL=IG ( ()9)9 :6BGOO2 3.>*+r,c|jjddr|jjjdr|jjdxs6t t j |jj}|jjd|k7r5td|jjdd|d|jyyy) NrNFrOrz!MD5 mismatch. Expected value is 'z', computed value is 'z'.)rJ) rPrFrBrEr+rRrSrTr)rgrhrJrUs r*rz$StorageContentValidation.on_response~s      2E :x?U?U?]?]?a?abo?p"??../EF-(889O9O9T9T9VWKL%%--m< L ;H > @ //))[[0B0BC{{?D4H4HIKK d.?.?@kk.$2E2EF&{{+?AXAXYKK1E1EF[[$/KK d3*  #$89  s$D,,D>=D>cy)zFormula for computing the current backoff. Should be calculated by child class. :param Dict[str, Any] settings: The configurable values pertaining to the backoff time. :return: The backoff time. :rtype: float rr>)rgr5s r*get_backoff_timez#StorageRetryPolicy.get_backoff_timesr,cX|j|}|r|dkry|j|y)abSleep for the backoff time. :param Dict[str, Any] settings: The configurable values pertaining to the sleep operation. :param transport: The transport to use for sleeping. :type transport: ~azure.core.pipeline.transport.AsyncioBaseTransport or ~azure.core.pipeline.transport.BaseTransport rN)rsleep)rgr5 transportbackoffs r*rzStorageRetryPolicy.sleeps,''1'A+  r,rJr errorc|dxxdzcc<|r=t|tr-|dxxdzcc<|djt||nm|r=t|tr-|dxxdzcc<|djt||n.|r,|dxxdzcc<|djt||t |s|j d vr|d r|j|||jr@t|jdr*|d y |jj|d t|d xxdz cc<yy #ttf$rYy wxYw)aIncrement the retry counters. :param Dict[str, Any] settings: The configurable values pertaining to the increment operation. :param request: A pipeline request object. :type request: ~azure.core.pipeline.PipelineRequest :param response: A pipeline response object. :type response: ~azure.core.pipeline.PipelineResponse or None :param error: An error encountered during the request, or None if the response was received successfully. :type error: ~azure.core.exceptions.AzureError or None :return: Whether the retry attempts are exhausted. :rtype: bool r.rr/r)rr0r1)rB)PUTr r Fr:T)r"rappendrrr7rr rTrrrrr)rgr5rhrJrs r* incrementzStorageRetryPolicy.incrementsU( Q Z':; Y 1 $  Y  & &~gU'K L z%)=> V  !  Y  & &~gU'K L "a'"#**>'QY+Z[H%~~W,:K1L,,Xw?|| f =O,4 !LL%%h&?J W  "  -j9! !s#EEEcd}d}|j|}|r |jj|}t||ds t |rt|j ||j |j}|rJt||j |jd|j||jj |dr|d|jd<|d|j_|S#t$r}}t|tr|j ||j |}|rEt||j d||j||jjYd}~Y|d}~wwxYw)aSend the request with retry logic. :param request: A pipeline request object. :type request: ~azure.core.pipeline.PipelineRequest :return: A pipeline response object. :rtype: ~azure.core.pipeline.PipelineResponse TNr;)rhrJ)rhrJr)rhrr)rrrrLrVrrQrBr@rrPrrr"rr=)rgrhretries_remainingrJretry_settingsrs r*rzStorageRetryPolicy.send)sh!//8 99>>'2HnV&<=ARS[A\(,&0D0DxOeOe)7)%)"*G4H4HS[SiSiqu >7??3L3LM  ) $*8*CH  Y '/=f/E, c#45$(NN>7K_K_gjN$k!$~w7K7KVZbefJJ~w/H/HI  sB(C00 E69A0E1/E11E6)NN)rkrlrmrr__annotations__rrrrr#r rfloatrrrrrrr|r}s@r*rrs$,O)+;333+S#X+IZ+_c+" ): tCH~ >c3hE !$26&* :sCx.:#:-. :  # :  :x%r,rceZdZUdZeed< eed< eed< d dededededededd ffd Zd e e efde fd Z xZ S)ExponentialRetryzExponential retry.initial_backoffincrement_baserandom_jitter_rangerrr?rzNc \||_||_||_tt|d||d|y)a Constructs an Exponential retry object. The initial_backoff is used for the first retry. Subsequent retries are retried after initial_backoff + increment_power^retry_count seconds. :param int initial_backoff: The initial backoff interval, in seconds, for the first retry. :param int increment_base: The base, in seconds, to increment the initial_backoff by after the first retry. :param int retry_total: The maximum number of retry attempts. :param bool retry_to_secondary: Whether the request should be retried to secondary, if able. This should only be enabled of RA-GRS accounts are used and potentially stale data can be handled. :param int random_jitter_range: A number in seconds which indicates a range to jitter/randomize for the back-off interval. For example, a random_jitter_range of 3 results in the back-off interval x to vary between x+3 and x-3. rrNr>)r"r#r$rsr!r)rgr"r#rrr$r?rxs r*rzExponentialRetry.__init__\s9: /,#6  .x;[mxqwxr,r5c tj}|j|ddk(rdnt|j|dz}||j kDr||j z nd}||j z}|j ||S)aR Calculates how long to sleep before retrying. :param Dict[str, Any] settings: The configurable values pertaining to get backoff time. :return: A float indicating how long to wait before retrying the request, or None to indicate no retry should be performed. :rtype: float r:r)randomRandomr"powr#r$uniform)rgr5random_generatorrrandom_range_startrandom_range_ends r*rz!ExponentialRetry.get_backoff_time~s"==?&&x/@A/E!3tObObdlmtduKvwCJTMeMeCeWt'?'??kl"T%=%==''(:)rr$rsr2r)rgrrrr$r?rxs r*rzLinearRetry.__init__s/. #6  k4)skVhslrsr,r5ctj}|j|jkDr|j|jz nd}|j|jz}|j ||S)aR Calculates how long to sleep before retrying. :param Dict[str, Any] settings: The configurable values pertaining to the backoff time. :return: A float indicating how long to wait before retrying the request, or None to indicate no retry should be performed. :rtype: float r)r(r)rr$r+)rgr5r,r-r.s r*rzLinearRetry.get_backoff_timesh"==?IM W[WoWoHoT\\D,D,DDuv<<$*B*BB''(:r,r*rNs1 -55 .WWD*26 '  H %#!f: U)U `= `6A#A@TG4TGn &) &/*/d9/9xAAH;N);N|2N$2Nj)Dr,