o iK@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:sB       zSyncBucketActionsMixin._requestoptions&Optional[CreateSignedUploadUrlOptions]r"c Cst}|dur|jr|d|jit|}|jdddd|jg||d}|}tj t |j t t |d d }tj|j}|d sLtd |||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&  z/SyncBucketActionsMixin.create_signed_upload_urlrgrN/Union[BufferedReader, bytes, FileIO, str, Path] file_options$Optional[UploadSignedUrlFileOptions]r(cCst|}d|i}ddd|jg|}|pi}|d} i} | r)d| |d<d| i} i|jjt|} |d} t|tsFt|tsFt|t rQd | || d fi} n d | t |d | d fi} |j d || | | |d }| }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)rErArtrGKeyr.r)r9r;rqr=rArrXrbytesrpopopenr_rCr()rZr.rgrNrxrsrG final_urlr` cache_control_datarAfilename_filerUrtr7r7r8upload_to_signed_urlsP z+SyncBucketActionsMixin.upload_to_signed_urlrhdownload_querydict[str, str]r$cCs4t|dd}|j||}t|t|dS)Nr3) signedURLri)rr<join extend_queryr/)rZrhrr6rr7r7r8_make_signed_urlsz'SyncBucketActionsMixin._make_signed_url expires_inintOptional[URLOptions]c Csdt|i}i}|p i}|d}r$|d|id|dur!dn|i}|d}r2|d|it|} |jddd|jg| |d } 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 transformrcrdrfrC) r/rqrRr9r_r;r#model_validate_jsoncontentrr) rZr.rr`rCr url_optionsrrrsrUrtr7r7r8create_signed_urls  z(SyncBucketActionsMixin.create_signed_urlpaths List[str]!Optional[CreateSignedURLsOptions]List[CreateSignedUrlResponse]cCs|t|d}i}|p i}|d}r%|d|id|dur"dn|i}|jddd|jg|d}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. )rrrTrrcrdrfrr)rKr.rri) r/rqrRr_r;r% validate_jsonrrrrKr.append)rZrrr`rCrrrrUrt signed_urlsitemr6 signed_itemr7r7r8create_signed_urlss0   z)SyncBucketActionsMixin.create_signed_urlsc Csi}|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_urlsz%SyncBucketActionsMixin.get_public_url from_pathto_pathcC&|jdddg|j||dd}|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 sourceKeydestinationKeyrr_r;rCrZrrresr7r7r8r:  zSyncBucketActionsMixin.movecCr)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`. rcrdcopyrrrrr7r7r8rPrzSyncBucketActionsMixin.copylist[dict[str, Any]]cCs"|jdd|jgd|id}|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`]. DELETErdprefixesrr)rZrrUr7r7r8removefs  zSyncBucketActionsMixin.removedict[str, Any]cCs(t|}|ddd|jg|}|S)z Lists info for a particular file. Parameters ---------- path The path to the file. GETrdinfo)r9r_r;rCrZr.rsrUr7r7r8rvs  zSyncBucketActionsMixin.infoboolcCsBzt|}|dd|jg|}|jdkWStjy YdSw)z Returns True if the file exists, False otherwise. Parameters ---------- path The path to the file. HEADrdF)r9r_r; status_coderCJSONDecodeErrorrr7r7r8existss   zSyncBucketActionsMixin.exists Optional[str] Optional[ListBucketFilesOptions]cCsJ|pi}ddi}it|d|pdi}|jddd|jg||d}|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)rCrA)rr_r;rC)rZr.r` extra_options extra_headersbodyrUr7r7r8rs  zSyncBucketActionsMixin.listOptional[SearchV2Options]r!cCs4|ri|ni}|jddd|jg|d}t|jS)Nrcrdzlist-v2r)r_r;r!rr)rZr`rrUr7r7r8list_v2s  zSyncBucketActionsMixin.list_v2Optional[DownloadOptions]Optional[Dict[str, str]]rc Csn|pt}|drgdndg}|dpt}t|}|jdg||j|it||p0id}|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)rG)rrqr&r9r_r;r+r) rZr.r`rGrrtransform_optionsrsrUr7r7r8rs"  zSyncBucketActionsMixin.downloadLiteral['POST', 'PUT']r1Optional[FileOptions]cCsf|duri}|dd}i}|dd}|r|d|i|dd}|dd} i|jjt|} |rJt|} t| | d<|d| i| rS| i| |dkrZ| d=|d } |rnd || d<|d |it |t s}t |t s}t |t rd | || d fi} n d | t|d| d fi} |j|d|jg|| | |d}|}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(SyncBucketActionsMixin._upload_or_updatecCt|}|d|||S)rrcr9rrZr.rNrxrsr7r7r8re<szSyncBucketActionsMixin.uploadcCr)Nrrrr7r7r8rRRszSyncBucketActionsMixin.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:2sP  * * E $ -        ' U r:c@seZdZdZdS)r-zRepresents a storage bucket.N)rrrrr7r7r7r8r-\sc@s<eZdZUdZded<ded<ded<edd Zd ed <d S) SyncBucketProxyzPA bucket proxy, this contains the minimum required fields to query the File API.r/r;rr<rr>F)reprrr=N)rrrrrrr=r7r7r7r8r`s 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  .