o zsgB@sddlmZddlZddlZddlZddlmZmZddl m Z m Z ddl m Z ddlmZmZmZmZmZddlmZmZdd lmZmZdd lmZdd lmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&dd l'm(Z(m)Z)d gZ*GdddZ+eddGdd d eZ,eGddde+Z-dS)) annotationsN) dataclassfield)BufferedReaderFileIO)Path)AnyLiteralOptionalUnioncast)HTTPStatusErrorResponse)DEFAULT_FILE_OPTIONSDEFAULT_SEARCH_OPTIONS)StorageApiError) BaseBucketCreateSignedURLsOptionsDownloadOptions FileOptionsListBucketFilesOptions RequestMethodSignedUploadURL UploadDataUploadResponse URLOptions) AsyncClientStorageException AsyncBucketc@seZdZUdZded<ded<   dQdRddZdSddZ dTdUd d!ZifdVd'd(ZifdWd-d.Z ifdXd/d0Z dYd3d4Z dYd5d6Z dZd9d:Z d[d;d<Zd\d>d?Z  d]d^dBdCZifd_dFdGZ dTd`dIdJZ dTdadKdLZ dTdadMdNZdbdOdPZdS)cAsyncBucketActionsMixinz(Functions needed to access the file API.stridr_clientNmethodrurlheadersOptional[dict[str, Any]]jsonOptional[dict[Any, Any]]files Optional[Any]kwargsrreturnrc sz|jj||f|p i||d|IdH}|Wnty8}z|j} t| d| d| dd}~ww|rPd|vrPt|ddtrP|dd |S)N)r&r(r*messageerror statusCodefile) r#requestraise_for_statusr responser(r isinstancerclose) selfr$r%r&r(r*r,r5excrespr;T/var/www/html/minecraft/env/lib/python3.10/site-packages/storage3/_async/file_api.py_request&s$    z AsyncBucketActionsMixin._requestpathrcs||}|dd|IdH}|}tjt|jj|d}tj |j }| ds4t 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/Nr%tokenzNo token sent by the APIr) signed_urlrAr>)_get_final_pathr=r(urllibparseurlparser!r#base_urlparse_qsquerygetrgeturl)r8r>_pathr5datafull_url query_paramsr;r;r<create_signed_upload_url>s   z0AsyncBucketActionsMixin.create_signed_upload_urlrAr1/Union[BufferedReader, bytes, FileIO, str, Path] file_optionsOptional[FileOptions]rcs||}tjd|}tjd|i}|d|}|dur&i}|d} i} | r:d| |d<d| i} i|jjt |} |j dd d d } t |t s\t |t s\t |trgd | || d fi} n d | t|d| d fi} |jd|| | | dIdH}|}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 r@rA?N cache-controlmax-age= cacheControl/r2maxsplitr1 content-typerbPUTr*r&rMKeyr>r`)rCrDrErF urlencoderKrJr#r&rrsplitr6rbytesrpopopenr=r(r)r8r>rAr1rRrL_urlrO final_url cache_control_datar&filename_filer5rMr;r;r<upload_to_signed_urlVsL    z,AsyncBucketActionsMixin.upload_to_signed_url expires_inintoptionsrdict[str, str]c sdt|i}d}|dr(|d|di|ddur dnd|d}|dr6|d|di||}|jdd||d Id H}|}tj|d }tj |j d |j }|j j tt|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. expiresIndownloadT &download= transformr? /object/sign/r(N signedURLrTrX)r!rJupdaterCr=r(rDrErFquoter>rIr#rGr lstrip) r8r>rnrpr(download_queryr5rMr%r;r;r<create_signed_urls.     z)AsyncBucketActionsMixin.create_signed_urlpaths list[str]rlist[dict[str, str]]c s|t|d}d}|dr*|d|di|ddur"dnd|d}|jdd|j|dId H}|}|D])}tj|d } tj | j d | j } |j j tt| 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. )rrrrsrtTrur?rwrxNryrTrX)r!rJrzr=r"r(rDrErFr{r>rIr#rGr r|) r8rrnrpr(r}r5rMitemr%r;r;r<create_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`. rsrtTrurvz render/imageobjectN&rTz/public/) rJappendrDrErbjoinrCr#rG) r8r>rp _query_stringr} render_pathtransformation_query query_stringrLr;r;r<get_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`. r?z /object/movebucketId sourceKeydestinationKeyrxNr=r"r(r8rrresr;r;r<move   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`. r?z /object/copyrrxNrrr;r;r<copy$rzAsyncBucketActionsMixin.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/prefixesrxNr)r8rr5r;r;r<remove:s   zAsyncBucketActionsMixin.removecs*|dd|jd|IdH}|S)z Lists info for a particular file. Parameters ---------- path The path to the file. GET /object/info/rXNrr8r>r5r;r;r<infoJs   zAsyncBucketActionsMixin.infoboolcs,|dd|jd|IdH}|jdkS)z Returns True if the file exists, False otherwise. Parameters ---------- path The path to the file. HEADrrXN)r=r" status_coderr;r;r<exists\s   zAsyncBucketActionsMixin.exists 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`, `sortBy` and `search`. z Content-Typezapplication/jsonprefixrsr?z /object/list/)r(r&N)rr=r"r()r8r>rp extra_options extra_headersbodyr5r;r;r<rns"  zAsyncBucketActionsMixin.listrrdcsl|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`. rvzrender/image/authenticatedrrTrsrrXN)rJrDrErbrCr=content)r8r>rprrrrLr5r;r;r<rts   z AsyncBucketActionsMixin.downloadLiteral['POST', 'PUT']csz|duri}|dd}i}|dd}|r|d|i|dd}|dd} i|jjt|} |rKt|} t| | d<|d| i| rT| i| |dkr[| d=|j d d d d } |rtd || d<|d|it |t st |t st |trd| || dfi} n d| t|d| dfi} ||}|j|d|| | |dIdH}|}t||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. NrUupsertzx-upsertmetadatar&z x-metadatar?rXr2rYr[rVrWr1r\r]rr_r`ra)rerzr#r&rr(dumpsbase64 b64encodeencodercr6rrdrrfrCr=rrJ)r8r$r>r1rRrirjrrfile_opts_headersr& metadata_strrkr*rLr5rMr;r;r<_upload_or_updates`       z)AsyncBucketActionsMixin._upload_or_updatecs|d|||IdHS)rr?Nrr8r>r1rRr;r;r<uploadszAsyncBucketActionsMixin.uploadcs|d|||IdHS)Nr^rrr;r;r<rz szAsyncBucketActionsMixin.updatecCs|jd|S)NrX)r")r8r>r;r;r<rCsz'AsyncBucketActionsMixin._get_final_path)NNN)r$rr%r!r&r'r(r)r*r+r,rr-r)r>r!r-r)N) r>r!rAr!r1rQrRrSr-r)r>r!rnrorprr-rq)rrrnrorprr-r)r>r!rprr-r!)rr!rr!r-rq)rrr-r)r>r!r-r)r>r!r-r)NN)r>rrprr-r)r>r!rprr-rd) r$rr>r!r1rQrRrSr-r)r>r!r1rQrRrSr-r)r>r!r-r!)__name__ __module__ __qualname____doc____annotations__r=rPrmr~rrrrrrrrrtrrrzrCr;r;r;r<r s@   C , ) "      W  r Freprc@seZdZdZdS)rzRepresents a storage bucket.N)rrrrr;r;r;r<rsc@s,eZdZUdZded<eddZded<dS) AsyncBucketProxyzPA bucket proxy, this contains the minimum required fields to query the File API.r!r"Frrr#N)rrrrrrr#r;r;r;r<rs r). __future__rrr( urllib.parserD dataclassesrriorrpathlibrtypingrr r r r httpxr r constantsrr exceptionsrtypesrrrrrrrrrrutilsrr__all__r rrr;r;r;r<s,   0 y