o ~jh:@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) AsyncClientStorageException AsyncBucketc@seZdZUdZded<ded<   dKdLddZdMddZ dNdOdd ZifdPd&d'ZifdQd,d-Z ifdRd.d/Z dSd2d3Z dSd4d5Z dTd8d9Z  dUdVd urlencoderCrBr!r$rrsplit isinstancerbytesrpopopenr5)r0r6r9rIrKrD_urlrG final_url cache_control_datar$filename_filer3r3r4upload_to_signed_urlQsH    z,AsyncBucketActionsMixin.upload_to_signed_url expires_inintoptionsrdict[str, str]csdt|i}|dr|d|di|dr#|d|di||}|jdd||dIdH}|}|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 transformr7 /object/sign/r&N signedURLrQ) rrBupdater;r5r&r!r?r lstrip)r0r6rgrir&r1rEr3r3r4create_signed_urls     z)AsyncBucketActionsMixin.create_signed_urlpaths list[str]rlist[dict[str, str]]cs|t|d}|dr|d|di|jdd|j|dIdH}|}|D]}|jjtt|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. )rtrkrlr7rnroNrprQ) rrBrqr5r r&r!r?r rr)r0rtrgrir&r1rEitemr3r3r4create_signed_urlss   z*AsyncBucketActionsMixin.create_signed_urlsc sg}d}|dr|ddurdnd|d}|r"|||dr)dnd}|dr9tj|dnd}|rB||d|}d |}||}|jj|d ||S) z Parameters ---------- path file path, including the path and file name. For example `folder/image.png`. NrlTz download=rmz render/imageobject&rMz/public/) rBappendr<r=rZjoinr;r!r?) r0r6ri _query_stringdownload_query render_pathtransformation_query query_stringrDr3r3r4get_public_urls*      z&AsyncBucketActionsMixin.get_public_url from_pathto_pathc*|jdd|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`. r7z /object/movebucketId sourceKeydestinationKeyroNr5r r&r0rrresr3r3r4move   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`. r7z /object/copyrroNrrr3r3r4copyrzAsyncBucketActionsMixin.copylistlist[dict[str, Any]]cs,|jdd|jd|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`]. DELETE/object/prefixesroNr)r0rtr1r3r3r4removes   zAsyncBucketActionsMixin.remove Optional[str] Optional[ListBucketFilesOptions]csR|pi}ddi}it|d|pdi}|jdd|j||dIdH}|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/jsonprefixr7z /object/list/)r&r$N)rr5r r&)r0r6ri extra_options extra_headersbodyr1r3r3r4r*s"  zAsyncBucketActionsMixin.listrr]csl|drdnd}tj|dpi}|rd|nd}||}|d|d||IdH}|jS) z Downloads a file. Parameters ---------- path The file path to be downloaded, including the path and file name. For example `folder/image.png`. rmzrender/image/authenticatedryrMrGETrQN)rBr<r=rZr;r5content)r0r6rirrrrDr1r3r3r4rlHs   z AsyncBucketActionsMixin.downloadLiteral['POST', 'PUT']c s|duri}|d}i}|dr |d|di|d=i|jjt|}|dkr1|d=|jdddd }|rGd ||d<d |i}t|tsVt|tsVt|t rad ||| d fi} n d |t |d| d fi} | |} |j |d| | ||dIdHS) 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. NrNupsertzx-upsertr7rQrRrSrUrOrPrIrVrWrrY)rBrqr!r$rr[r\rr]rr^r_r;r5) r0r"r6rIrKrbrcr$rdr(rDr3r3r4_upload_or_update^sL    z)AsyncBucketActionsMixin._upload_or_updatecs|d|||IdHS)rr7Nrr0r6rIrKr3r3r4uploadszAsyncBucketActionsMixin.uploadcs|d|||IdHS)NrXrrr3r3r4rqszAsyncBucketActionsMixin.updatecCs|jd|S)NrQ)r )r0r6r3r3r4r;sz'AsyncBucketActionsMixin._get_final_path)NNN)r"rr#rr$r%r&r'r(r)r*r r+r)r6rr+r)N) r6rr9rrIrJrKrLr+r)r6rrgrhrirr+rj)rtrurgrhrirr+rv)r6rrirr+r)rrrrr+rj)rtrr+r)NN)r6rrirr+rv)r6rrirr+r]) r"rr6rrIrJrKrLr+r)r6rrIrJrKrLr+r)r6rr+r)__name__ __module__ __qualname____doc____annotations__r5rHrfrsrxrrrrrrlrrrqr;r3r3r3r4rs<   @   "    G  rFreprc@s$eZdZUdZeddZded<dS)rzRepresents a storage bucket.Frrr!N)rrrrrr!rr3r3r3r4rs c@s,eZdZUdZded<eddZded<dS) AsyncBucketProxyzPA bucket proxy, this contains the minimum required fields to query the File API.rr Frrr!N)rrrrrrr!r3r3r3r4rs r)* __future__r urllib.parser< dataclassesrriorrr&rpathlibrtypingr r r r r httpxrr constantsrrtypesrrrrrrrrutilsrr__all__rrrr3r3r3r4s(   ( ,