o ˳iL@s^ddlmZddlZddlZddlZddlmZmZddl m Z m Z ddl m Z ddlmZmZmZmZmZmZmZddlmZmZmZmZddlmZd d lmZmZd d l m!Z!d d l"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/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6d d l7m8Z8dgZ9dddZ:GdddZ;Gddde#Zr=requestr/raise_for_statusrresponserCrKeyErrortext isinstancerclose)selfr?r.rArCrErGrIurl_pathrUexcresperrrJr7r7r8_request:sD       z AsyncBucketActionsMixin._requestoptions&Optional[CreateSignedUploadUrlOptions]r"c st}|dur|jr|d|jit|}|jdddd|jg||dIdH}|}tj t |j t t |d d }tj|j}|d sPtd |||d d |d S)a Creates a signed upload URL. Parameters ---------- path The file path, including the file name. For example `folder/image.png`. options Additional options for the upload url creation. Nx-upsertPOSTobjectuploadsign)rAr6r2tokenzNo token sent by the APIr) signed_url signedUrlrgr.)rQupsertrRr9r_r;rCurllibparseurlparser/r<rlstripparse_qsquerygetr,geturl) rZr.r`rA path_partsrUdatafull_urlrGr7r7r8create_signed_upload_urlas(   z0AsyncBucketActionsMixin.create_signed_upload_urlrgrN/Union[BufferedReader, bytes, FileIO, str, Path] file_options$Optional[UploadSignedUrlFileOptions]r(cst|}d|i}ddd|jg|}|pi}|d} i} | r*d| |d<d| i} i|jjt|} |d} t|tsGt|tsGt|t rRd | || d fi} n d | t |d | d fi} |j d || | | |d IdH}| }t||ddS)a Upload a file with a token generated from :meth:`.create_signed_url` Parameters ---------- path The file path, including the file name token The token generated from :meth:`.create_signed_url` file The file contents or a file-like object to upload file_options Additional options for the uploaded file rgrdrerf cache-controlmax-age= cacheControlrN content-typerbPUT)rErArtrGNKeyr.r)r9r;rqr=rArrXrbytesrpopopenr_rCr()rZr.rgrNrxrsrG final_urlr` cache_control_datarAfilename_filerUrtr7r7r8upload_to_signed_urlsR  z,AsyncBucketActionsMixin.upload_to_signed_urlrhdownload_querydict[str, str]r$cCs4t|dd}|j||}t|t|dS)Nr3) signedURLri)rr<join extend_queryr/)rZrhrr6rr7r7r8_make_signed_urlsz(AsyncBucketActionsMixin._make_signed_url expires_inintOptional[URLOptions]c sdt|i}i}|p i}|d}r%|d|id|dur"dn|i}|d}r3|d|it|} |jddd|jg| |d Id H} t| j} | | j |S) a/ Parameters ---------- path file path to be downloaded, including the current file name. expires_in number of seconds until the signed URL expires. options options to be passed for downloading or transforming the file. expiresIndownloadT transformrcrdrfrCN) r/rqrRr9r_r;r#model_validate_jsoncontentrr) rZr.rr`rCr url_optionsrrrsrUrtr7r7r8create_signed_urls"  z)AsyncBucketActionsMixin.create_signed_urlpaths List[str]!Optional[CreateSignedURLsOptions]List[CreateSignedUrlResponse]cs|t|d}i}|p i}|d}r&|d|id|dur#dn|i}|jddd|jg|dId H}t|j} g} | D]} || j |} | j | j | d | d d } | | q?| S) a Parameters ---------- path file path to be downloaded, including the current file name. expires_in number of seconds until the signed URL expires. options options to be passed for downloading the file. )rrrTrrcrdrfrNr)rKr.rri) r/rqrRr_r;r% validate_jsonrrrrKr.append)rZrrr`rCrrrrUrt signed_urlsitemr6 signed_itemr7r7r8create_signed_urlss2   z*AsyncBucketActionsMixin.create_signed_urlsc si}|pi}|d}rd|durdn|i}|dr!ddgndg}|d}r/t|nt}t|} |jjg|d|j| R||} t | S) z Parameters ---------- path file path, including the path and file name. For example `folder/image.png`. rTrrrenderimagerdpublic) rqr+rQr9r<rOr;rPrr/) rZr.r`rrr render_pathttransformationrsr6r7r7r8get_public_urls z&AsyncBucketActionsMixin.get_public_url from_pathto_pathc.|jdddg|j||ddIdH}|S)aa Moves an existing file, optionally renaming it at the same time. Parameters ---------- from_path The original file path, including the current file name. For example `folder/image.png`. to_path The new file path, including the new file name. For example `folder/image-copy.png`. rcrdmovebucketId sourceKeydestinationKeyrNr_r;rCrZrrresr7r7r8r<   zAsyncBucketActionsMixin.movecr)aZ Copies an existing file to a new path in the same bucket. Parameters ---------- from_path The original file path, including the current file name. For example `folder/image.png`. to_path The new file path, including the new file name. For example `folder/image-copy.png`. rcrdcopyrrNrrr7r7r8rRrzAsyncBucketActionsMixin.copylist[dict[str, Any]]cs*|jdd|jgd|idIdH}|S)z Deletes files within the same bucket Parameters ---------- paths An array or list of files to be deletes, including the path and file name. For example [`folder/image.png`]. DELETErdprefixesrNr)rZrrUr7r7r8removehs  zAsyncBucketActionsMixin.removedict[str, Any]cs0t|}|ddd|jg|IdH}|S)z Lists info for a particular file. Parameters ---------- path The path to the file. GETrdinfoN)r9r_r;rCrZr.rsrUr7r7r8rxs  zAsyncBucketActionsMixin.infoboolcsJzt|}|dd|jg|IdH}|jdkWStjy$YdSw)z Returns True if the file exists, False otherwise. Parameters ---------- path The path to the file. HEADrdNF)r9r_r; status_coderCJSONDecodeErrorrr7r7r8existss   zAsyncBucketActionsMixin.exists Optional[str] Optional[ListBucketFilesOptions]csR|pi}ddi}it|d|pdi}|jddd|jg||dId H}|S) z Lists all the files within a bucket. Parameters ---------- path The folder path. options Search options, including `limit`, `offset`, `sortBy` and `search`. z Content-Typezapplication/jsonprefixrrcrdlist)rCrAN)rr_r;rC)rZr.r` extra_options extra_headersbodyrUr7r7r8rs"  zAsyncBucketActionsMixin.listOptional[SearchV2Options]r!cs<|ri|ni}|jddd|jg|dIdH}t|jS)Nrcrdzlist-v2r)r_r;r!rr)rZr`rrUr7r7r8list_v2s  zAsyncBucketActionsMixin.list_v2Optional[DownloadOptions]Optional[Dict[str, str]]rc sv|pt}|drgdndg}|dpt}t|}|jdg||j|it||p1idIdH}|jS)z Downloads a file. Parameters ---------- path The file path to be downloaded, including the path and file name. For example `folder/image.png`. r)rr authenticatedrdr)rGN)rrqr&r9r_r;r+r) rZr.r`rGrrtransform_optionsrsrUr7r7r8rs$   z AsyncBucketActionsMixin.downloadLiteral['POST', 'PUT']r1Optional[FileOptions]csn|duri}|dd}i}|dd}|r|d|i|dd}|dd} i|jjt|} |rKt|} t| | d<|d| i| rT| i| |dkr[| d=|d } |rod || d<|d |it |t s~t |t s~t |t rd | || d fi} n d | t|d| d fi} |j|d|jg|| | |dIdH}|}td||ddS) Uploads a file to an existing bucket. Parameters ---------- path The relative file path including the bucket ID. Should be of the format `bucket/folder/subfolder/filename.png`. The bucket must already exist before attempting to upload. file The File object to be stored in the bucket. or a async generator of chunks file_options HTTP headers. NrzrjrbmetadatarAz x-metadatarcr}r{r|rNr~rrd)rErArtr2rr)rrRr=rArrCdumpsbase64 b64encodeencoderXrrrrr_r;r(r)rZr?r.rNrxrrrjrfile_opts_headersrA metadata_strrrErUrtr7r7r8_upload_or_updates^      z)AsyncBucketActionsMixin._upload_or_updatecs t|}|d|||IdHS)rrcNr9rrZr.rNrxrsr7r7r8re>szAsyncBucketActionsMixin.uploadcs t|}|d|||IdHS)Nrrrr7r7r8rRTszAsyncBucketActionsMixin.update)NNNN)r?rr.r@rArBrCrDrErFrGrHrIrr0r)N)r.r/r`rar0r") r.r/rgr/rNrwrxryr0r()rhr/rrr0r$)r.r/rrr`rr0r$)rrrrr`rr0r)r.r/r`rr0r/)rr/rr/r0r)rr@r0r)r.r/r0r)r.r/r0r)NN)r.rr`rr0r)r`rr0r!)r.r/r`rrGrr0r) r?rr.r1rNrwrxrr0r()r.r/rNrwrxrr0r()__name__ __module__ __qualname____doc____annotations__r_rvrrrrrrrrrrrrrrrerRr7r7r7r8r:2sR  * * E $ .        ' U r:c@seZdZdZdS)r-zRepresents a storage bucket.N)rrrrr7r7r7r8r-^sc@s<eZdZUdZded<ded<ded<edd Zd ed <d S) AsyncBucketProxyzPA bucket proxy, this contains the minimum required fields to query the File API.r/r;rr<rr>F)reprrr=N)rrrrrrr=r7r7r7r8rbs r)r.r/r0r1)> __future__rrrC urllib.parserk dataclassesrriorrpathlibrtypingrr r r r r rhttpxrrrryarlr constantsrr exceptionsrtypesrrrrrrrrr r!r"r#r$r%r&r'r(r)r*r+utilsr,__all__r9r:r-rr7r7r7r8s0  $  X  0