
     h                        S SK r S SKrS SKrS SKJr  S SKrS SKrS SKr	S SK
r	S SKJrJrJrJr  S SKJrJrJrJrJr  S SKJrJrJr  S SK	Jr  S SKJr  S SKJr  S S	KJr  S S
K J!r!J"r"  S SK#J$r$J%r%  S SK&J'r'  S SK(J)r)J*r*  S SK+J,r,  S SK-J.r.J/r/J0r0J1r1J2r2  \ Rf                  " \45      r5Sq6S r7SSSS\%-  SS4S jr8S/S jr9 " S S5      r: " S S\*5      r; " S S\)5      r< " S S5      r= " S S\=5      r> " S S \5      r? " S! S"5      r@ " S# S$5      rA " S% S&5      rB " S' S(5      rC " S) S*5      rD " S+ S,5      rE " S- S.5      rFg)0    N)BytesIO)AwsCredentialsAwsCredentialsProviderAwsSigningAlgorithmAwsSigningConfig)ClientBootstrapClientTlsContextDefaultHostResolverEventLoopGroupTlsContextOptions)S3ClientS3RequestTlsModeS3RequestType)UNSIGNED)urlsplit)Config)NoCredentialsError)	ArnParserInvalidArnException)FULL_OBJECT_CHECKSUM_ARGSMB)TransferNotDoneError)BaseTransferFutureBaseTransferMeta)TransferManager)CallArgsOSUtilscreate_nested_clientget_callbacksis_s3express_bucketc                     [         c2  [        R                  R                  U 5      n UR	                  5         Uq [         $ ! [
         a     g f = fN)CRT_S3_PROCESS_LOCKawscrts3CrossProcessLockacquireRuntimeError)namecrt_locks     l/root/racknerd_01/codex-conversation/amazon-q-terraform/.venv/lib/python3.13/site-packages/s3transfer/crt.pyacquire_crt_s3_process_lockr,   ;   sS     "99--d3	 '  	 		s   A   
AA   Tc                 T   [        U5      n[        U5      n[        Xx5      n	Sn
U(       a  [        R                  O[        R
                  nUbC  [        5       nU(       a  UR                  US9  OSUl        [        U5      nUR                  5       n
[        US9n[        U	U UUUU
USS9$ )a  
:type region: str
:param region: The region used for signing

:type crt_credentials_provider:
    Optional[awscrt.auth.AwsCredentialsProvider]
:param crt_credentials_provider: CRT AWS credentials provider
    to use to sign requests. If not set, requests will not be signed.

:type num_threads: Optional[int]
:param num_threads: Number of worker threads generated. Default
    is the number of processors in the machine.

:type target_throughput: Optional[int]
:param target_throughput: Throughput target in bytes per second.
    By default, CRT will automatically attempt to choose a target
    throughput that matches the system's maximum network throughput.
    Currently, if CRT is unable to determine the maximum network
    throughput, a fallback target throughput of ``1_250_000_000`` bytes
    per second (which translates to 10 gigabits per second, or 1.16
    gibibytes per second) is used. To set a specific target
    throughput, set a value for this parameter.

:type part_size: Optional[int]
:param part_size: Size, in Bytes, of parts that files will be downloaded
    or uploaded in.

:type use_ssl: boolean
:param use_ssl: Whether or not to use SSL.  By default, SSL is used.
    Note that not all services support non-ssl connections.

:type verify: Optional[boolean/string]
:param verify: Whether or not to verify SSL certificates.
    By default SSL certificates are verified.  You can provide the
    following values:

    * False - do not validate SSL certificates.  SSL will still be
        used (unless use_ssl is False), but SSL certificates
        will not be verified.
    * path/to/cert/bundle.pem - A filename of the CA cert bundle to
        use. Specify this argument if you want to use a custom CA cert
        bundle instead of the default one on your system.
N)ca_filepathF) provided_throughput_target_bytesT)	bootstrapregioncredential_provider	part_sizetls_modetls_connection_optionsthroughput_target_gbpsenable_s3express)r   r
   r   r   ENABLEDDISABLEDr   &override_default_trust_store_from_pathverify_peerr	   new_connection_options_get_crt_throughput_target_gbpsr   )r2   crt_credentials_providernum_threadstarget_throughputr4   use_sslverifyevent_loop_grouphost_resolverr1   r6   r5   tls_ctx_optionsclient_tls_optiontarget_gbpss                  r+   create_s3_crt_clientrI   T   s    h &k2'(89M 0@I! %,  1A1J1J  +-BB" C  +0O',_=!2!I!I!K1):K 45*	 	    c                     U c:  [         R                  R                  5       n[        R	                  SU5        Uc  SnOU S-  S-  n[        R	                  SU5        U$ )Nz-Recommended CRT throughput target in gbps: %sg      $@r-   i ʚ;z'Using CRT throughput target in gbps: %s)r$   r%   &get_recommended_throughput_target_gbpsloggerdebug)r0   rH   s     r+   r>   r>      s_    '/iiFFH;[	
 K 7:]J
LL:KHrJ   c                       \ rS rSr\R
                  r\R                  r\R                  rSr\R                  r	SS jr
S rS r SS jrSS jrSS	 jrSS
 jrS rS rS rS rS rS rSS jrS rS rSrg)CRTTransferManager   TNc                     Uc  [        5       U l        Xl        [        X R                  5      U l        UR
                  U l        / U l        [        R                  " S5      U l
        SU l        g)a  A transfer manager interface for Amazon S3 on CRT s3 client.

:type crt_s3_client: awscrt.s3.S3Client
:param crt_s3_client: The CRT s3 client, handling all the
    HTTP requests and functions under then hood

:type crt_request_serializer: s3transfer.crt.BaseCRTRequestSerializer
:param crt_request_serializer: Serializer, generates unsigned crt HTTP
    request.

:type osutil: s3transfer.utils.OSUtils
:param osutil: OSUtils object to use for os-related behavior when
    using with transfer manager.
N   r   )r   _osutil_crt_s3_clientS3ClientArgsCreator_s3_args_creatortranslate_crt_exception_crt_exception_translator_future_coordinators	threading	Semaphore
_semaphore_id_counter)selfcrt_s3_clientcrt_request_serializerosutils       r+   __init__CRTTransferManager.__init__   sc     >"9DL+ 3"LL!
 #:: 	& %'!#--c2rJ   c                     U $ r"    r_   s    r+   	__enter__CRTTransferManager.__enter__   s    rJ   c                 <    SnU(       a  SnU R                  U5        g )NFT	_shutdown)r_   exc_type	exc_valueargscancels        r+   __exit__CRTTransferManager.__exit__   s    FvrJ   c                     Uc  0 nUc  0 nU R                  X@R                  5        U R                  U5        [        UUUUUS9nU R	                  SU5      $ )Nbucketkeyfileobj
extra_argssubscribers
get_object)_validate_all_known_argsALLOWED_DOWNLOAD_ARGS_validate_if_bucket_supportedr   _submit_transfer)r_   ru   rv   rw   rx   ry   callargss          r+   downloadCRTTransferManager.download   sj     JK%%j2L2LM**62!#
 $$\8<<rJ   c                     Uc  0 nUc  0 nU R                  X@R                  5        U R                  U5        U R                  U5        [	        UUUUUS9nU R                  SU5      $ )Nrt   
put_object)r{   ALLOWED_UPLOAD_ARGSr}   &_validate_checksum_algorithm_supportedr   r~   )r_   rw   ru   rv   rx   ry   r   s          r+   uploadCRTTransferManager.upload   sv    JK%%j2J2JK**6233J?!#
 $$\8<<rJ   c                     Uc  0 nUc  0 nU R                  X0R                  5        U R                  U5        [        UUUUS9nU R	                  SU5      $ )N)ru   rv   rx   ry   delete_object)r{   ALLOWED_DELETE_ARGSr}   r   r~   )r_   ru   rv   rx   ry   r   s         r+   deleteCRTTransferManager.delete  se    JK%%j2J2JK**62!#	
 $$_h??rJ   c                 &    U R                  U5        g r"   rk   r_   rp   s     r+   shutdownCRTTransferManager.shutdown  s    vrJ   c                     U R                   (       aK  U R                  R                  5        H,  u  p#UR                  U5      nU(       d  M  [	        SU S35      e   g g )Nz'TransferManager methods do not support z+ resource. Use direct client calls instead.) VALIDATE_SUPPORTED_BUCKET_VALUES_UNSUPPORTED_BUCKET_PATTERNSitemsmatch
ValueError)r_   ru   resourcepatternr   s        r+   r}   0CRTTransferManager._validate_if_bucket_supported  sc     00%)%F%F%L%L%N!f-5$A( LE E  &O 1rJ   c           	      `    U H(  nX2;  d  M
  [        SU SSR                  U5       35      e   g )NzInvalid extra_args key 'z', must be one of: z, )r   join)r_   actualallowedkwargs       r+   r{   +CRTTransferManager._validate_all_known_args,  s?    E# .ug 6''+yy'9&:<  rJ   c                     UR                  S5      nUc  g [        [        R                  R                  R
                  5      nUR                  5       U;  a  [        SU SU 35      eg )NChecksumAlgorithmzChecksumAlgorithm: z* not supported. Supported algorithms are: )getlistr$   r%   S3ChecksumAlgorithm__members__upperr   )r_   rx   checksum_algorithmsupported_algorithmss       r+   r   9CRTTransferManager._validate_checksum_algorithm_supported4  su    '^^,?@%#FII$A$A$M$MN##%-AA%&8%9 :--A,BD  BrJ   c                 x    U R                    H*  nUR                  5       (       a  M  UR                  5         M,     g r"   )rZ   donerp   r_   coordinators     r+   _cancel_transfers$CRTTransferManager._cancel_transfers?  s-    44K##%%""$ 5rJ   c                 J    U R                    H  nUR                  5         M     g r"   )rZ   resultr   s     r+   _finish_transfers$CRTTransferManager._finish_transfersD  s    44K  5rJ   c                 J    U R                    H  nUR                  5         M     g r"   )rZ   %wait_until_on_done_callbacks_completer   s     r+   _wait_transfers_done'CRTTransferManager._wait_transfers_doneH  s    44K==? 5rJ   c                     U(       a  U R                  5          U R                  5         U R	                  5         g ! [         a    U R                  5          N-[         a     N8f = f! U R	                  5         f = fr"   )r   r   KeyboardInterrupt	Exceptionr   r   s     r+   rl   CRTTransferManager._shutdownL  sg    ""$	(""$ %%' ! 	%""$ 		 %%'s-   : A"A% 	A"A% !A""A% %A7c                 8    U R                   R                  5         g r"   )r]   releaser_   kwargss     r+   _release_semaphore%CRTTransferManager._release_semaphoreY  s    !rJ   c                    U R                   /n[        U R                  U R                  S9n[	        U R                  U5      US.n[        S	0 UD6n[        U5      nUR                  U5         U R                  R                  5         U R                  R                  US5      nU" 5         U R                  R                  UUUUU5      n	U R                  R                  " S	0 U	D6n
UR                  U
5        U R$                  R                  U5        U =R                  S-  sl        U$ ! [          a=  nUR#                  US5        U R                  R                  USUS9nU" US9   S nANtS nAff = f)
N)transfer_idexception_translator)metar   queuedTr   )after_subscribers)error   rf   )r   CRTTransferCoordinatorr^   rY   CRTTransferMetaCRTTransferFutureAfterDoneHandlerappendr]   r'   rW   get_crt_callbackget_make_request_argsrU   make_requestset_s3_requestr   set_exceptionrZ   )r_   request_type	call_argson_done_after_callsr   
componentsfuture	afterdone	on_queuedcrt_callargscrt_s3_requesteon_dones                r+   r~   #CRTTransferManager._submit_transfer\  sp   #667,((!%!?!?

 $D$4$4i@&

 #0Z0$[1	""9-	7OO##%-->>I K00FF#L "00==MMN &&~6!!((5A  	%%a.++<<2E = G !	s   -A8D( (
E/23E**E/)rY   rU   rZ   r^   rT   rW   r]   r"   NNF)__name__
__module____qualname____firstlineno__r   r|   r   r   r   r   rc   rh   rq   r   r   r   r   r}   r{   r   r   r   r   rl   r   r~   __static_attributes__rf   rJ   r+   rP   rP      s    +AA)==)=='+$#2#O#O : BF=$="@	%
!@("'rJ   rP   c                   R    \ rS rSrSrS	S jr\S 5       r\S 5       r\S 5       r	Sr
g)
r   i  z*Holds metadata about the CRTTransferFutureNc                 *    Xl         X l        0 U l        g r"   )_transfer_id
_call_args_user_context)r_   r   r   s      r+   rc   CRTTransferMeta.__init__  s    '#rJ   c                     U R                   $ r"   )r   rg   s    r+   r   CRTTransferMeta.call_args  s    rJ   c                     U R                   $ r"   )r   rg   s    r+   r   CRTTransferMeta.transfer_id  s       rJ   c                     U R                   $ r"   )r   rg   s    r+   user_contextCRTTransferMeta.user_context  s    !!!rJ   )r   r   r   r   )r   r   r   r   __doc__rc   propertyr   r   r   r   rf   rJ   r+   r   r     sF    4 
   ! ! " "rJ   r   c                   J    \ rS rSrS
S jr\S 5       rS rSS jrS r	S r
S	rg)r   i  Nc                 @    Xl         Uc  [        5       U l         X l        g)a2  The future associated to a submitted transfer request via CRT S3 client

:type meta: s3transfer.crt.CRTTransferMeta
:param meta: The metadata associated to the transfer future.

:type coordinator: s3transfer.crt.CRTTransferCoordinator
:param coordinator: The coordinator associated to the transfer future.
N)_metar   _coordinator)r_   r   r   s      r+   rc   CRTTransferFuture.__init__  s     
<(*DJ'rJ   c                     U R                   $ r"   )r   rg   s    r+   r   CRTTransferFuture.meta  s    zzrJ   c                 6    U R                   R                  5       $ r"   )r   r   rg   s    r+   r   CRTTransferFuture.done  s      %%''rJ   c                 :    U R                   R                  U5        g r"   )r   r   r_   timeouts     r+   r   CRTTransferFuture.result  s      )rJ   c                 8    U R                   R                  5         g r"   )r   rp   rg   s    r+   rp   CRTTransferFuture.cancel  s      "rJ   c                 x    U R                  5       (       d  [        S5      eU R                  R                  USS9  g)z!Sets the exception on the future.z?set_exception can only be called once the transfer is complete.T)overrideN)r   r   r   r   r_   	exceptions     r+   r   CRTTransferFuture.set_exception  s:    yy{{&  	''	D'ArJ   )r   r   r   r"   )r   r   r   r   rc   r   r   r   r   rp   r   r   rf   rJ   r+   r   r     s0    (  (*#BrJ   r   c                        \ rS rSrS rS rSrg)BaseCRTRequestSerializeri  c                     [        S5      e)a4  Serialize CRT HTTP requests.

:type transfer_type: string
:param transfer_type: the type of transfer made,
    e.g 'put_object', 'get_object', 'delete_object'

:type future: s3transfer.crt.CRTTransferFuture

:rtype: awscrt.http.HttpRequest
:returns: An unsigned HTTP request to be used for the CRT S3 client
zserialize_http_request()NotImplementedError)r_   transfer_typer   s      r+   serialize_http_request/BaseCRTRequestSerializer.serialize_http_request  s     ""<==rJ   c                     [        S5      e)Nztranslate_crt_exception()r  r  s     r+   rX   0BaseCRTRequestSerializer.translate_crt_exception  s    !"=>>rJ   rf   N)r   r   r   r   r  rX   r   rf   rJ   r+   r  r    s    >?rJ   r  c                   `    \ rS rSrSS jrS rS rS rS rS r	S	 r
S
 rS rS rS rS rSrg)BotocoreCRTRequestSerializeri  Nc                 .   Xl         Uc  0 nU R                  X5        [        U40 UD6U l        U R                  R                  R
                  R                  SU R                  5        U R                  R                  R
                  R                  SU R                  5        U R                  R                  R
                  R                  SU R                  5        U R                  R                  R
                  R                  SU R                  5        g)a  Serialize CRT HTTP request using botocore logic
It also takes into account configuration from both the session
and any keyword arguments that could be passed to
`Session.create_client()` when serializing the request.

:type session: botocore.session.Session

:type client_kwargs: Optional[Dict[str, str]])
:param client_kwargs: The kwargs for the botocore
    s3 client initialization.
Nzrequest-created.s3.*zafter-call.s3.*zbefore-send.s3.*zbefore-call.s3.*)_session_resolve_client_configr   _clientr   eventsregister_capture_http_request+_change_response_to_serialized_http_request_make_fake_http_response_remove_checksum_context)r_   sessionclient_kwargss      r+   rc   %BotocoreCRTRequestSerializer.__init__  s       M##G;+GE}E  ))"D$>$>	
 	  ))tOO	
 	  )) = =	
 	  )) = =	
rJ   c                     S nUR                  5       (       a  UR                  5       nSU;   a  US   n[        [        S9nU(       a  UR                  U5      nXBS'   SUS'   g )Nconfig)signature_versionr%   service_name)get_default_client_configr   r   merge)r_   r   r!  user_provided_configclient_configs        r+   r  3BotocoreCRTRequestSerializer._resolve_client_config  sh    #,,..#*#D#D#F }$#0#: :066}EM"/h(,n%rJ   c                    [        UR                  5      nUR                  nUR                  (       a  U SUR                   3n/ nUR                  R                  5        HK  u  pV[        U[        5      (       a  UR                  XV45        M.  UR                  U[        US5      45        MM     [        R                  R                  U5      n[        R                  R                  UR                  UUUR                  S9nU$ )N?zutf-8)methodpathheadersbody_stream)r   urlr/  queryr0  r   
isinstancestrr   r$   httpHttpHeadersHttpRequestr.  body)	r_   aws_request	url_partscrt_pathheaders_listr)   valuecrt_headerscrt_requests	            r+   _crt_request_from_aws_request:BotocoreCRTRequestSerializer._crt_request_from_aws_request   s    [__-	>>??"1Y__$56H&..446KD%%%##TM2##T3ug+>$?@	 7 kk--l;kk--%%#((	 . 
 rJ   c                 <   U R                  U5      nUR                  R                  S5      c;  [        UR                  5      nUR                  R                  SUR                  5        UR                  R                  S5      b  UR                  R                  S5        UR                  R                  S5      c)  UR                  c  UR                  R                  SS5        UR                  R                  S5      b  UR                  R                  S5        U$ )NhostzContent-MD5zContent-Length0zTransfer-Encoding)
rA  r0  r   r   r2  setnetlocremover9  add)r_   botocore_http_requestr@  r;  s       r+   _convert_to_crt_http_request9BotocoreCRTRequestSerializer._convert_to_crt_http_request  s    889NO""6*2 !6!:!:;I##FI,<,<=""=1=&&}5 ""#34<$))1##''(8#> ""#67C&&':;rJ   c                      XR                   S'   g )Nhttp_request)contextr_   requestr   s      r+   r  2BotocoreCRTRequestSerializer._capture_http_request4  s    *1'rJ   c                 4    US   nUR                  5       US'   g )NrN  HTTPRequest)prepare)r_   rO  parsedr   rQ  s        r+   r  HBotocoreCRTRequestSerializer._change_response_to_serialized_http_request7  s     .) ' 1}rJ   c                 X    [         R                  R                  S S0 [        S5      5      $ )N   rJ   )botocore
awsrequestAWSResponseFakeRawResponserP  s      r+   r  5BotocoreCRTRequestSerializer._make_fake_http_response=  s,    ""..C 	
 	
rJ   c                     [        U R                  U5      " SUR                  UR                  S.UR                  D6S   $ )N)BucketKeyrT  rf   )getattrr  ru   rv   rx   )r_   client_methodr   s      r+   _get_botocore_http_request7BotocoreCRTRequestSerializer._get_botocore_http_requestE  sD    t||]3 
##
:C:N:N

 	rJ   c                 r    U R                  XR                  R                  5      nU R                  U5      nU$ r"   )rd  r   r   rK  )r_   r  r   rJ  r@  s        r+   r  3BotocoreCRTRequestSerializer.serialize_http_requestJ  s9     $ ? ?;;00!
 778MNrJ   c                 x    [        U[        R                  R                  5      (       a  U R	                  U5      $ g r"   )r4  r$   r%   S3ResponseError _translate_crt_s3_response_errorr  s     r+   rX   4BotocoreCRTRequestSerializer.translate_crt_exceptionQ  s-    i!:!:;;88CCrJ   c                 F   UR                   nUS:  a  g UR                   VVs0 s H  u  p4X4_M	     nnnUR                  nUb<  U R                  R                  R
                  nUR                  U5      R                  nOS n[        R                  R                  U5      UUR                  S.n	U R                  R                  R                  XS9n
U
R                  S0 5      R                  S5      nU R                  R                  R!                  U5      nU" XS9$ s  snnf )Ni-  )r0  status_coder9  )shapeErrorCode)operation_name)rm  r0  rq  r  r   service_modeloperation_modeloutput_shaperZ  r[  HeadersDictr9  _response_parserparser   
exceptions	from_code)r_   s3_response_errorrm  kvr0  rq  rr  rn  response_dictparsed_response
error_codeerror_classs                r+   rj  =BotocoreCRTRequestSerializer._translate_crt_s3_response_errorW  s   '33 $5$=$=>$=DA14$=>*99% LL--;;M!11.ANNEE  **66w?&%**

 ,,77== > 
 %(("599&A
ll--77
C?JJ' ?s   Dc                 <    UR                  S0 5      nSU;   a  US	 g g )NrO  checksum)r   )r_   paramsr   request_contexts       r+   r  5BotocoreCRTRequestSerializer._remove_checksum_contexts  s'     **Y3(
+ )rJ   )r  r  r"   )r   r   r   r   rc   r  rA  rK  r  r  r  rd  r  rX   rj  r  r   rf   rJ   r+   r  r    s?    
<-,<22

K8,rJ   r  c                       \ rS rSrSS jrSrg)r]  iy  Nc              #   J   #     U R                  U5      nU(       d  g Uv   M  7fr"   )read)r_   amtdecode_contentchunks       r+   streamFakeRawResponse.streamz  s%     IIcNEK	 s   !#rf   )i   N)r   r   r   r   r  r   rf   rJ   r+   r]  r]  y  s    rJ   r]  c                   ,    \ rS rSrS rS rS rS rSrg)BotocoreCRTCredentialsWrapperi  c                     Xl         g r"   _resolved_credentials)r_   resolved_botocore_credentialss     r+   rc   &BotocoreCRTCredentialsWrapper.__init__  s    %B"rJ   c                     U R                  5       R                  5       n[        UR                  UR                  UR
                  5      $ r"   )_get_credentialsget_frozen_credentialsr   
access_key
secret_keytoken)r_   credentialss     r+   __call__&BotocoreCRTCredentialsWrapper.__call__  s?    ++-DDF""K$:$:K<M<M
 	
rJ   c                 .    [         R                  " U 5      $ r"   )r   new_delegaterg   s    r+   to_crt_credentials_provider9BotocoreCRTCredentialsWrapper.to_crt_credentials_provider  s    %22488rJ   c                 H    U R                   c
  [        5       eU R                   $ r"   )r  r   rg   s    r+   r  .BotocoreCRTCredentialsWrapper._get_credentials  s#    %%-$&&)))rJ   r  N)	r   r   r   r   rc   r  r  r  r   rf   rJ   r+   r  r    s    C
9*rJ   r  c                   p    \ rS rSrSr SS jr\S 5       rS rSS jr	SS jr
S	 rSS
 jrS rS rS rSrg)r   i  z-A helper class for managing CRTTransferFutureNc                     Xl         X0l        X l        [        R                  " 5       U l        S U l        S U l        [        R                  " 5       U l	        g r"   )
r   _exception_translator_s3_requestr[   Lock_lock
_exception_crt_futureEvent_done_event)r_   r   
s3_requestr   s       r+   rc   CRTTransferCoordinator.__init__  sB     '%9"%^^%
$??,rJ   c                     U R                   $ r"   )r  rg   s    r+   r  !CRTTransferCoordinator.s3_request  s    rJ   c                 8    U R                   R                  5         g r"   )r  rF  rg   s    r+   set_done_callbacks_complete2CRTTransferCoordinator.set_done_callbacks_complete  s    rJ   c                 :    U R                   R                  U5        g r"   )r  waitr   s     r+   r   <CRTTransferCoordinator.wait_until_on_done_callbacks_complete  s    g&rJ   c                     U R                      U R                  5       (       a  U(       a  Xl        S S S 5        g ! , (       d  f       g = fr"   )r  r   r  )r_   r  r  s      r+   r   $CRTTransferCoordinator.set_exception  s%    ZZ99;;("+ ZZs	   #9
Ac                 \    U R                   (       a  U R                   R                  5         g g r"   )r  rp   rg   s    r+   rp   CRTTransferCoordinator.cancel  s"    ##% rJ   c                    U R                   (       a  U R                   e U R                  R                  U5        U R                  (       a  S U l        g g ! [         a-    U R	                  5         U R                  R                  U5        e [
         a  nU R                  U5         S nANoS nAff = f! U R                  (       a  S U l        f f = fr"   )r  r  r   r   rp   r   handle_exceptionr  )r_   r  r   s      r+   r   CRTTransferCoordinator.result  s    ??//!
	(##G, #'    ! 	KKM##G, 	%!!!$$	% #'   s)   A ?B.B)$B1 )B..B1 1Cc                     S nU R                   (       a   U R                  U5      nUb  X!eUe! [         a  n[        R                  SUS9   S nAN*S nAff = f)NzUnable to translate exception.)exc_info)r  r   rM   rN   )r_   exctranslated_excr   s       r+   r  'CRTTransferCoordinator.handle_exception  s`    %%!%!;!;C!@ % )I   =J	s   - 
AAAc                 R    U R                   c  gU R                   R                  5       $ )NF)r  r   rg   s    r+   r   CRTTransferCoordinator.done  s&    #$$&&rJ   c                 F    Xl         U R                   R                  U l        g r"   )r  finished_futurer  )r_   r  s     r+   r   %CRTTransferCoordinator.set_s3_request  s    %++;;rJ   )r  r  r  r  r  r  r   )NNNr"   r   )r   r   r   r   r   rc   r   r  r  r   r   rp   r   r  r   r   r   rf   rJ   r+   r   r     sM    7 GK	-    ',
&('
<rJ   r   c                   @    \ rS rSrS rS r  S
S jrS rS rS r	S	r
g)rV   i  c                     Xl         X l        g r"   )_request_serializer	_os_utils)r_   ra   os_utilss      r+   rc   S3ClientArgsCreator.__init__  s    #9 !rJ   c           	      L    [        U SU 3U R                  5      nU" UUUU/ US9$ )N_get_make_request_args_r   r   r   r   on_done_before_callsr   )rb  _default_get_make_request_args)r_   r   r   r   r   r   request_args_handlers          r+   r   )S3ClientArgsCreator.get_make_request_args  sF      '%l^4// 

 $%#!# 3
 	
rJ   Nc                     ^^^^ UUUU4S jnU$ )Nc                     > / nTb  UT-  nU[        TT5      -  nTb  UT-  nU H  nTS:X  a  U" U S   S9  M  U" U 0 UD6  M     g )Nprogressr   )bytes_transferred)r   )ro   r   callbacks_listcallbackr   before_subscriberscallback_typer   s       r+   invoke_all_callbacksBS3ClientArgsCreator.get_crt_callback.<locals>.invoke_all_callbacks  sh    N!-"44mFMBBN ,"33* !J.tAw7d-f- +rJ   rf   )r_   r   r  r  r   r  s    ```` r+   r   $S3ClientArgsCreator.get_crt_callback  s    	. 	.  $#rJ   c           	        ^ S n[        TR                  [        5      (       a7  TR                  nU R                  R	                  U5      nUTR
                  S'   OTR                  TR
                  S'   S n	[        U4S j[         5       5      (       d  TR
                  R                  SS5      R                  5       n
[        R                  R                  [        R                  R                  U
   [        R                  R                  R                  S9n	STR
                  S'   U R!                  UTUUUUS	9nX{S
'   XS'   U$ )NContentLengthBodyc              3   B   >#    U  H  nUTR                   ;   v   M     g 7fr"   )rx   ).0checksum_argr   s     r+   	<genexpr>HS3ClientArgsCreator._get_make_request_args_put_object.<locals>.<genexpr>  s#      
 9 I000 9s   r   CRC32)	algorithmlocationzoverride-to-be-removed
ContentMD5r  send_filepathchecksum_config)r4  rw   r5  r  get_file_sizerx   anyr   popr   r$   r%   S3ChecksumConfigr   S3ChecksumLocationTRAILERr  )r_   r   r   r   r   r  r   r  data_lenr  r   make_request_argss     `         r+   !_get_make_request_args_put_object5S3ClientArgsCreator._get_make_request_args_put_object  s>    i''--%--M~~33MBH4<I  1+4+<+<I  ( 
 9
 
 
 "+!5!5!9!9#W"eg  %ii88 ))778JK55== 9 O .F	\* ??%#!5 3 @ 
 .;/*/>+,  rJ   c           	         S nS n[         R                  R                  SS9n	[        UR                  [
        5      (       aM  UR                  n
U R                  R                  U
5      nUR                  [        X:XpR                  5      5        O[        UR                  5      nU R                  UUUUUUS9nX{S'   XS'   XS'   U$ )NT)validate_responser  recv_filepathon_bodyr  )r$   r%   r  r4  rw   r5  r  get_temp_filenamer   RenameTempFileHandlerOnBodyFileObjWriterr  )r_   r   r   r   r   r  r   r  r  r  final_filepathr  s               r+   !_get_make_request_args_get_object5S3ClientArgsCreator._get_make_request_args_get_object<  s      ))44t4Li''--&..N NN<<^LM ''% *)*;*;<G ??%#!5 3 @ 
 .;/*'.)$/>+,  rJ   c                    U R                   R                  X5      [        [        UR	                  5       [        R
                  5      U R                  USXV5      U R                  US5      S.nUS   [        R
                  :X  a*  SR                  S UR                  S5       5       5      US'   [        5       nUR                  UR                  5      =n	(       a2  U	S	   S:X  a)  [        [        R                  S
SSS9US'   U	S   Ul        U$ [        UR                  5      (       a  [        [        R                   SSS9US'   U$ )Nr   r  )rQ  typer   on_progressr   c              3   @   #    U  H  oR                  5       v   M     g 7fr"   )title)r  xs     r+   r  ES3ClientArgsCreator._default_get_make_request_args.<locals>.<genexpr>y  s      :#:a		#:s   _rq  r2   *F)r  r2   use_double_uri_encodeshould_normalize_uri_pathsigning_configresource_name)r  r  r  )r  r  rb  r   r   DEFAULTr   r   split_S3ArnParamHandler
handle_arnru   r   r   V4_ASYMMETRICr    V4_S3EXPRESS)
r_   r   r   r   r   r  r   r  arn_handleraccesspoint_arn_detailss
             r+   r  2S3ClientArgsCreator._default_get_make_request_args`  s^    //FF |113]5J5J ,, 4  00D
 V$(=(==24'' :#/#5#5c#:: 3./ )*'2'='=i>N>N'OO#O%h/25
 3C-;;&+*/	3./  7GI !  !!1!122 3C-::&+*/3./
 ! rJ   )r  r  r   )r   r   r   r   rc   r   r   r  r  r  r   rf   rJ   r+   rV   rV     s*    "
*  $2-!^"!H4!rJ   rV   c                        \ rS rSrS rS rSrg)r  i  c                 4    Xl         X l        X0l        X@l        g r"   )r   _final_filename_temp_filenamerT   )r_   r   final_filenametemp_filenamerb   s        r+   rc   RenameTempFileHandler.__init__  s    '-+rJ   c                 x   US   nU(       a&  U R                   R                  U R                  5        g  U R                   R                  U R                  U R                  5        g ! [
         aJ  nU R                   R                  U R                  5        U R                  R                  U5         S nAg S nAff = f)Nr   )rT   remove_filer  rename_filer  r   r   r   )r_   r   r   r   s       r+   r  RenameTempFileHandler.__call__  s    wLL$$T%8%893(('')=)=  3(()<)<=!!//223s   0A% %
B9/A B44B9)r   r  rT   r  Nr   r   r   r   rc   r  r   rf   rJ   r+   r  r    s    3rJ   r  c                        \ rS rSrS rS rSrg)r   i  c                     Xl         g r"   r   r   s     r+   rc   AfterDoneHandler.__init__  s    'rJ   c                 8    U R                   R                  5         g r"   )r   r  r   s     r+   r  AfterDoneHandler.__call__  s    557rJ   r)  Nr&  rf   rJ   r+   r   r     s    (8rJ   r   c                        \ rS rSrS rS rSrg)r   i  c                     Xl         g r"   _fileobj)r_   rw   s     r+   rc   OnBodyFileObjWriter.__init__  s    rJ   c                 :    U R                   R                  U5        g r"   )r0  write)r_   r  r   s      r+   r  OnBodyFileObjWriter.__call__  s    E"rJ   r/  Nr&  rf   rJ   r+   r   r     s     #rJ   r   c                   T    \ rS rSrSr\R                  " S5      rS rS r	S r
S rSrg	)
r  i  zPartial port of S3ArnParamHandler from botocore.

This is used to make a determination on MRAP accesspoints for signing
purposes. This should be safe to remove once we properly integrate auth
resolution from Botocore into the CRT transfer integration.
zA^(?P<resource_type>accesspoint|outpost)[/:](?P<resource_name>.+)$c                 "    [        5       U l        g r"   )r   _arn_parserrg   s    r+   rc   _S3ArnParamHandler.__init__  s    $;rJ   c                 D    U R                  U5      nUc  g US   S:X  a  U$ g )Nresource_typeaccesspoint)_get_arn_details_from_bucketr_   ru   arn_detailss      r+   r  _S3ArnParamHandler.handle_arn  s3    77?'=8 9rJ   c                      U R                   R                  U5      nU R                  U5        U$ ! [         a     g f = fr"   )r7  	parse_arn_add_resource_type_and_namer   r=  s      r+   r<  /_S3ArnParamHandler._get_arn_details_from_bucket  sF    	**44V<K,,[9" 		s   -0 
==c                     U R                   R                  US   5      nU(       a)  UR                  S5      US'   UR                  S5      US'   g g )Nr   r:  r  )_RESOURCE_REGEXr   group)r_   r>  r   s      r+   rB  ._S3ArnParamHandler._add_resource_type_and_name  sK    $$**;z+BC+0;;+GK(+0;;+GK( rJ   )r7  N)r   r   r   r   r   recompilerE  rc   r  r<  rB  r   rf   rJ   r+   r  r    s/     jjLO'HrJ   r  r"   )GloggingrH  r[   ior   awscrt.httpr$   	awscrt.s3botocore.awsrequestrZ  botocore.sessionawscrt.authr   r   r   r   	awscrt.ior   r	   r
   r   r   r   r   r   r   botocore.compatr   botocore.configr   botocore.exceptionsr   botocore.utilsr   r   s3transfer.constantsr   r   s3transfer.exceptionsr   s3transfer.futuresr   r   s3transfer.managerr   s3transfer.utilsr   r   r   r   r    	getLoggerr   rM   r#   r,   rI   r>   rP   r   r   r  r  r]  r  r   rV   r  r   r   r  rf   rJ   r+   <module>r\     s@    	         @ ?  $ " 2 9 > 6 C .  
		8	$ 6 ""fRj$H HV"& "*#B* #BL? ?&b,#; b,Jg * *&F< F<Rv! v!r3 3,8 8# ##H #HrJ   