U #i:@sddlmZddlZddlmZmZddlmZm Z ddl m Z ddl m Z ddlmZmZmZmZmZddlmZmZd d lmZmZd d lmZmZmZmZmZm Z m!Z!m"Z"d d l#m$Z$m%Z%d gZ&GdddZ'eddGdd d ee'Z(eGddde'Z)dS)) annotationsN) dataclassfield)BufferedReaderFileIO)JSONDecodeError)Path)AnyLiteralOptionalUnioncast) HTTPErrorResponse)DEFAULT_FILE_OPTIONSDEFAULT_SEARCH_OPTIONS) BaseBucketCreateSignedURLsOptionsDownloadOptions FileOptionsListBucketFilesOptions RequestMethodSignedUploadURL URLOptions)StorageException SyncClient SyncBucketc @s^eZdZUdZded<ded<dEdddd d d d d ddZdddddZdFddddd dddZifdddddddZifddd d!d"d#d$Z ifdddd%d&d'Z dddd(d)d*Z dddd(d+d,Z d-d.d/d0d1Z dGd2d3d!d%d4d5Zifdd6d7d%d8d9ZdHd:dddd d;dd?d@ZdJdddd d>dAdBZddddCdDZdS)KSyncBucketActionsMixinz(Functions needed to access the file API.stridr_clientNrzOptional[dict[str, Any]]zOptional[dict[Any, Any]]z Optional[Any]r r)methodurlheadersjsonfileskwargsreturnc Ksz.|jj||f|pi||d|}|WnVtk rz|}t|d|jiWn"tk r~td|jiYnXYnX|S)N)r$r%r&Z statusCode)r!requestraise_for_statusrr%r status_coder) selfr"r#r$r%r&r'responserespr/;/tmp/pip-unpacked-wheel-7lsqp372/storage3/_sync/file_api.py_request"s&  zSyncBucketActionsMixin._requestr)pathr(cCsx||}|dd|}|}tjt|jj|d}tj |j }| ds`t d| |dd|dS)z Creates a signed upload URL. Parameters ---------- path The file path, including the file name. For example `folder/image.png`. POST/object/upload/sign/r#tokenzNo token sent by the APIr)Z signed_urlr5r2)_get_final_pathr1r%urllibparseurlparserr!base_urlparse_qsquerygetrgeturl)r,r2_pathr-datafull_url query_paramsr/r/r0create_signed_upload_url9s   z/SyncBucketActionsMixin.create_signed_upload_urlz/Union[BufferedReader, bytes, FileIO, str, Path]zOptional[FileOptions])r2r5file file_optionsr(cCs||}tjd|}tjd|i}|d|}|dkrJi}|d} i} | rrd| |d<d| i} |jjt |} |j dd d d } t |t st |t st |trd | || d fi} nd | t|d| d fi} |jd|| | | dS)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 r4r5?N cache-controlmax-age= cacheControl/maxsplitrD content-typerbPUTr&r$r@)r6r7r8r9 urlencoder>r=r!r$rrsplit isinstancerbytesrpopopenr1)r,r2r5rDrEr?_urlrBZ final_url cache_control_datar$filename_filer/r/r0upload_to_signed_urlQs<  z+SyncBucketActionsMixin.upload_to_signed_urlintrzdict[str, str])r2 expires_inoptionsr(cCsdt|i}|dr(|d|di|drD|d|di||}|jdd||d}|}|jjtt|d d|d<|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. expiresIndownload transformr3 /object/sign/r% signedURLrJ) rr=updater6r1r%r!r:r lstrip)r,r2r`rar%r-r@r/r/r0create_signed_urls    z(SyncBucketActionsMixin.create_signed_urlz list[str]rzlist[dict[str, str]])pathsr`rar(cCs~|t|d}|dr,|d|di|jdd|j|d}|}|D](}|jjtt|d d|d<qP|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. )rkrbrcr3rerfrgrJ) rr=rhr1r r%r!r:r ri)r,rkr`rar%r-r@itemr/r/r0create_signed_urlss   z)SyncBucketActionsMixin.create_signed_urls)r2rar(c Csg}d}|dr4|ddkr$dnd|d}|rB|||drPdnd}|drptj|dnd}|r||d|}d |}||}|jj|d ||S) z Parameters ---------- path file path, including the path and file name. For example `folder/image.png`. NrcTz download=rdz render/imageobject&rFz/public/) r=appendr7r8rSjoinr6r!r:) r,r2raZ _query_stringZdownload_query render_pathtransformation_query query_stringr?r/r/r0get_public_urls(       z%SyncBucketActionsMixin.get_public_url) from_pathto_pathr(cCs"|jdd|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`. r3z /object/moveZbucketIdZ sourceKeyZdestinationKeyrfr1r r%r,rvrwresr/r/r0moves  zSyncBucketActionsMixin.movecCs"|jdd|j||dd}|S)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`. r3z /object/copyrxrfryrzr/r/r0copys  zSyncBucketActionsMixin.copylistzlist[dict[str, Any]])rkr(cCs$|jdd|jd|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`]. DELETE/object/prefixesrfry)r,rkr-r/r/r0removes  zSyncBucketActionsMixin.removez Optional[str]z Optional[ListBucketFilesOptions]cCsD|pi}ddi}t|d|pdi}|jdd|j||d}|S)z Lists all the files within a bucket. Parameters ---------- path The folder path. options Search options, including `limit`, `offset`, and `sortBy`. z Content-Typezapplication/jsonprefixr3z /object/list/)r%r$)rr1r r%)r,r2raZ extra_options extra_headersbodyr-r/r/r0r~(s zSyncBucketActionsMixin.listrrVcCsd|drdnd}tj|dp$i}|r6d|nd}||}|d|d||}|jS)z Downloads a file. Parameters ---------- path The file path to be downloaded, including the path and file name. For example `folder/image.png`. rdzrender/image/authenticatedrnrFrGETrJ)r=r7r8rSr6r1content)r,r2rarrrsrtr?r-r/r/r0rcFs  zSyncBucketActionsMixin.downloadzLiteral[('POST', 'PUT')])r"r2rDrEr(c Cs|dkr i}|d}i}|dr>|d|di|d=|jjt|}|dkrZ|d=|jdddd }|rd ||d<d |i}t|tst|tst|t rd ||| d fi} nd |t |d| d fi} | |} |j |d| | ||dS) 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. NrGZupsertzx-upsertr3rJrKrLrNrHrIrDrOrPrrR)r=rhr!r$rrTrUrrVrrWrXr6r1) r,r"r2rDrErZr[r$r\r&r?r/r/r0_upload_or_update\sL   z(SyncBucketActionsMixin._upload_or_update)r2rDrEr(cCs|d|||S)rr3rr,r2rDrEr/r/r0uploadszSyncBucketActionsMixin.uploadcCs|d|||S)NrQrrr/r/r0rhszSyncBucketActionsMixin.updatecCs|jd|S)NrJ)r )r,r2r/r/r0r6sz&SyncBucketActionsMixin._get_final_path)NNN)N)NN)N)N)N)__name__ __module__ __qualname____doc____annotations__r1rCr^rjrmrur|r}rr~rcrrrhr6r/r/r/r0rs: > "GrFreprc@s$eZdZUdZeddZded<dS)rzRepresents a storage bucket.Frrr!N)rrrrrr!rr/r/r/r0rs c@s,eZdZUdZded<eddZded<dS) SyncBucketProxyzPA bucket proxy, this contains the minimum required fields to query the File API.rr Frrr!N)rrrrrrr!r/r/r/r0rs r)* __future__r urllib.parser7Z dataclassesrriorrr%rpathlibrtypingr r r r r Zhttpxrr constantsrrtypesrrrrrrrrutilsrr__all__rrrr/r/r/r0s&   ( *