o zsgA@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)StorageException SyncClient SyncBucketc@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)cSyncBucketActionsMixinz(Functions needed to access the file API.stridr_clientNmethodrurlheadersOptional[dict[str, Any]]jsonOptional[dict[Any, Any]]files Optional[Any]kwargsrreturnrc Ksz|jj||f|p i||d|}|Wnty4}z|j} t| d| d| dd}~ww|rLd|vrLt|ddtrL|dd |S)N)r&r(r*messageerror statusCodefile) r#requestraise_for_statusr responser(r isinstancerclose) selfr$r%r&r(r*r,r5excrespr;S/var/www/html/minecraft/env/lib/python3.10/site-packages/storage3/_sync/file_api.py_request&s"    zSyncBucketActionsMixin._requestpathrcCsx||}|dd|}|}tjt|jj|d}tj |j }| ds0t 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) signed_urlrAr>)_get_final_pathr=r(urllibparseurlparser!r#base_urlparse_qsquerygetrgeturl)r8r>_pathr5datafull_url query_paramsr;r;r<create_signed_upload_url>s   z/SyncBucketActionsMixin.create_signed_upload_urlrAr1/Union[BufferedReader, bytes, FileIO, str, Path] file_optionsOptional[FileOptions]rcCs||}tjd|}tjd|i}|d|}|dur%i}|d} i} | r9d| |d<d| i} i|jjt |} |j dd d d } t |t s[t |t s[t |trfd | || d fi} n d | t|d| d fi} |jd|| | | 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 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_urlVsJ   z+SyncBucketActionsMixin.upload_to_signed_url expires_inintoptionsrdict[str, str]c Csdt|i}d}|dr'|d|di|ddurdnd|d}|dr5|d|di||}|jdd||d }|}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( signedURLrTrX)r!rJupdaterCr=r(rDrErFquoter>rIr#rGr lstrip) r8r>rnrpr(download_queryr5rMr%r;r;r<create_signed_urls,    z(SyncBucketActionsMixin.create_signed_urlpaths list[str]rlist[dict[str, str]]c Cs|t|d}d}|dr)|d|di|ddur!dnd|d}|jdd|j|d}|}|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?rwrxryrTrX)r!rJrzr=r"r(rDrErFr{r>rIr#rGr r|) r8rrnrpr(r}r5rMitemr%r;r;r<create_signed_urlss(   z)SyncBucketActionsMixin.create_signed_urlsc Csg}d}|dr|ddurdnd|d}|r!|||dr(dnd}|dr8tj|dnd}|rA||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%SyncBucketActionsMixin.get_public_url from_pathto_pathcC"|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`. r?z /object/movebucketId sourceKeydestinationKeyrxr=r"r(r8rrresr;r;r<move  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`. r?z /object/copyrrxrrr;r;r<copy$rzSyncBucketActionsMixin.copylistlist[dict[str, Any]]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/prefixesrxr)r8rr5r;r;r<remove:s   zSyncBucketActionsMixin.removecCs"|dd|jd|}|S)z Lists info for a particular file. Parameters ---------- path The path to the file. GET /object/info/rXrr8r>r5r;r;r<infoJs  zSyncBucketActionsMixin.infoboolcCs$|dd|jd|}|jdkS)z Returns True if the file exists, False otherwise. Parameters ---------- path The path to the file. HEADrrX)r=r" status_coderr;r;r<exists\s   zSyncBucketActionsMixin.exists Optional[str] Optional[ListBucketFilesOptions]cCsJ|pi}ddi}it|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`, `sortBy` and `search`. z Content-Typezapplication/jsonprefixrsr?z /object/list/)r(r&)rr=r"r()r8r>rp extra_options extra_headersbodyr5r;r;r<rns  zSyncBucketActionsMixin.listrrdcCsd|drdnd}tj|dpi}|rd|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`. rvzrender/image/authenticatedrrTrsrrX)rJrDrErbrCr=content)r8r>rprrrrLr5r;r;r<rts  zSyncBucketActionsMixin.downloadLiteral['POST', 'PUT']cCsr|duri}|dd}i}|dd}|r|d|i|dd}|dd} i|jjt|} |rJt|} t| | d<|d| i| rS| i| |dkrZ| d=|j d d d d } |rsd || d<|d|it |t st |t st |trd| || dfi} n d| t|d| dfi} ||}|j|d|| | |d}|}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(SyncBucketActionsMixin._upload_or_updatecC|d|||S)rr?rr8r>r1rRr;r;r<uploadszSyncBucketActionsMixin.uploadcCr)Nr^rrr;r;r<rz szSyncBucketActionsMixin.updatecCs|jd|S)NrX)r")r8r>r;r;r<rCsz&SyncBucketActionsMixin._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) SyncBucketProxyzPA 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