gBDddlmZddlZddlZddlZddlmZmZddl 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*GddZ+edGdd eZ,eGdde+Z-y)) annotationsN) dataclassfield)BufferedReaderFileIO)Path)AnyLiteralOptionalUnioncast)HTTPStatusErrorResponse)DEFAULT_FILE_OPTIONSDEFAULT_SEARCH_OPTIONS)StorageApiError) BaseBucketCreateSignedURLsOptionsDownloadOptions FileOptionsListBucketFilesOptions RequestMethodSignedUploadURL UploadDataUploadResponse URLOptions) AsyncClientStorageException AsyncBucketcleZdZUdZded<ded< d ddZddZ d dd Zif dd Zif dd Z ifdd Z d d Z d dZ d!dZ d"dZ d#dZ d$ d%dZifd&dZ d d'dZ d d(dZ d d(dZd)dZy)*AsyncBucketActionsMixinz(Functions needed to access the file API.stridr_clientNcvK |jj||f|xsi||d|d{}|j|r0d|vr,t|ddtr|ddj|S7H#t$r5}|jj } t | d| d| dd}~wwxYww)N)headersjsonfilesmessageerror statusCodefile) r%requestraise_for_statusrresponser(r isinstancerclose) selfmethodurlr'r(r)kwargsr1excresps M/var/www/openai/venv/lib/python3.12/site-packages/storage3/_async/file_api.py_requestz AsyncBucketActionsMixin._request&s V1T\\11%,]UNTH  % % ' Vu_E&M!4Dn)U &M!  " " $ V<<$$&D!$y/4=$|BTU U Vs8B9)A8A6A84B96A88 B60B11B66B9cK|j|}|jdd|d{}|j}tjj t |jj|dz}tjj|j}|jds td|j|dd|dS7w) z Creates a signed upload URL. Parameters ---------- path The file path, including the file name. For example `folder/image.png`. POST/object/upload/sign/Nr6tokenzNo token sent by the APIr) signed_urlr?path)_get_final_pathr;r(urllibparseurlparser#r%base_urlparse_qsquerygetrgeturl)r4rA_pathr1datafull_url query_paramss r:create_signed_upload_urlz0AsyncBucketActionsMixin.create_signed_upload_url>s$$T*v1EeW/MNN}}-3\\-B-B  %% &e 4. ||,,X^^< ("#=> >"//+!'*1-  Os*C$C"B6C$cK|j|}tjjd|}tjj d|i}|j d|}|i}|j d} i} | r d| |d<d| i} i|jjt|} |jdd d } t|ts t|tst|trd | || jd fi} n d | t!|d| jd fi} |j#d|| | | d{}|j%}t'||j dS70w)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>r??N cache-controlmax-age= cacheControl/r.maxsplitr- content-typerbPUTr)r'rLKeyrAr])rBrCrDrE urlencoderJrIr%r'rrsplitr2rbytesrpopopenr;r(r)r4rAr?r- file_optionsrK_urlrN final_url cache_control_datar'filename_filer1rLs r:upload_to_signed_urlz,AsyncBucketActionsMixin.upload_to_signed_urlVs*$$T*||$$';E7%CD||--w.>? {{}oQ|n5  L$((9  .6}o,FL )#]3E ll"" "   ;;sQ;/3 t^ ,$&$'hgkk..IJKEt$KK/E 9E7'  $==?4TXXe_==  sEE:E81E:cKdt|i}d}|jdr@|jd|di|jddurdnd|jd}|jdr|jd|di|j|}|j dd|| d {}|j }t jj|d }t jj|jd |jz}|jjtt|jd ||d <|S7w)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 signedURLrQrU)r#rIupdaterBr;r(rCrDrEquoterArHr%rFr lstrip) r4rA expires_inoptionsr(download_queryr1rLr6s r:create_signed_urlz)AsyncBucketActionsMixin.create_signed_urlscS_- ;;z " DKKWZ%89 :;;z*d2!'++j"9!:;  ;;{ # DKKgk&:; <##D) D6 "'   }}ll##D$56ll  *q _<||$$%d3n&;&;C&@%A.AQ R [  sB4E&6E$7B.E&cK|t|d}d}|jdrL|jd|jdi|jddurdnd|jd}|jdd|j|d {}|j }|D]}t jj|d } t jj| jd | jz} |jjtt| jd ||d <|S7w) 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. )pathsrmrnroTrpr=rrrsNrtrQrU)r#rIrur;r$r(rCrDrErvrArHr%rFr rw) r4r}rxryr(rzr1rLitemr6s r:create_signed_urlsz*AsyncBucketActionsMixin.create_signed_urlssAS_= ;;z " DKKW[[%<= >;;z*d2!'++j"9!:;   DGG9 %'   }}D,,''[(9:C,,$$SXX.1SYYK@C<<(()$sC.*?*?*D)EnEUV     sBE E B5E c Kg}d}|jdr)|jddurdnd|jd}|r|j||jdrdnd}|jdr.tjj |jdnd}|r|j|d j |}d |}|j |}|jj|d ||Sw) z Parameters ---------- path file path, including the path and file name. For example `folder/image.png`. rnroTrprqz render/imageobjectN&rQz/public/) rIappendrCrDr_joinrBr%rF) r4rAry _query_stringrz render_pathtransformation_query query_stringrKs r:get_public_urlz&AsyncBucketActionsMixin.get_public_urls   ;;z ";;z*d2!'++j"9!:;     0(/ K(@nh {{;' LL " "7;;{#; <   !5 6xx . <.) $$T*,,''( XeW\NSSsDDcK|jdd|j||dd{}|jS7w)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 sourceKeydestinationKeyrsNr;r$r(r4 from_pathto_pathress r:movezAsyncBucketActionsMixin.moveKMM   GG&")"  xxz &?=?cK|jdd|j||dd{}|jS7w)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/copyrrsNrrs r:copyzAsyncBucketActionsMixin.copy$rrcK|jdd|jd|id{}|jS7w)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/prefixesrsNr)r4r}r1s r:removezAsyncBucketActionsMixin.remove:sL twwi e$'   }}  s'A>AcK|jdd|jd|d{}|jS7w)z Lists info for a particular file. Parameters ---------- path The path to the file. GET /object/info/rUNrr4rAr1s r:infozAsyncBucketActionsMixin.infoJsE DGG9AdV ,  }}  rcK|jdd|jd|d{}|jdk(S7w)z Returns True if the file exists, False otherwise. Parameters ---------- path The path to the file. HEADrrUN)r;r$ status_coders r:existszAsyncBucketActionsMixin.exists\sJ DGG9AdV ,  ##s**  s &><>cK|xsi}ddi}it|d|xsdi}|jdd|j||d{}|jS7w) 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/jsonprefixrnr=z /object/list/)r(r'N)rr;r$r()r4rAry extra_options extra_headersbodyr1s r:listzAsyncBucketActionsMixin.listns 2 ');<  $   djb   DGG9 %! '   }}  sAAAAc*K|jdrdnd}tjj|jdxsi}|rd|nd}|j |}|j d|d||d{}|j S7w) z Downloads a file. Parameters ---------- path The file path to be downloaded, including the path and file name. For example `folder/image.png`. rqzrender/image/authenticatedrrQrnrrUN)rIrCrDr_rBr;content)r4rAryrrrrKr1s r:roz AsyncBucketActionsMixin.downloads-4KK ,D ((  &||55gkk+6N6TRTU5I/01r $$T* m1UGL> 2    sA>BBBcK|i}|jdd}i}|jdd}|r|jd|i|jdd}|jdd} i|jjt|} |rNt j |} tj| j| d<|jd| i| r| ji| |dk7r| d=|jd d d } |rd || d<|jd|it|ts t|tst|trd| || jdfi} n d| t|d| jdfi} |j!|}|j#|d|| | |d{}|j }t%||j'dS70w) 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. NrRupsertzx-upsertmetadatar'z x-metadatar=rUr.rVrXrSrTr-rYrZrr\r]r^)rbrur%r'rr(dumpsbase64 b64encodeencoder`r2rrarrcrBr;rrI)r4r5rAr-rdrgrhrrfile_opts_headersr' metadata_strrir)rKr1rLs r:_upload_or_updatez)AsyncBucketActionsMixin._upload_or_updates(  L$(($? !!(D1    V 4 5##J5(,,Y= ll"" "   ::h/L$*$4$4\5H5H5J$KGL ! LL*l3 4  NN0/0 1 V  #;;sQ;/3 )1-'AGO $ LL.-8 9 t^ ,$&$'hgkk..IJKEt$KK/E$$T* hug&eW5'  $==?4TXXe_==  sF8G-:G+;1G-cFK|jd|||d{S7w)rr=Nrr4rAr-rds r:uploadzAsyncBucketActionsMixin.uploads%&++FD$ MMMM !!cFK|jd|||d{S7w)Nr[rrs r:ruzAsyncBucketActionsMixin.update s% ++E4|LLLLrc$|jd|S)NrU)r$)r4rAs r:rBz'AsyncBucketActionsMixin._get_final_paths''!D6"")NNN)r5rr6r#r'zOptional[dict[str, Any]]r(zOptional[dict[Any, Any]]r)z Optional[Any]r7r returnr)rAr#rr)N) rAr#r?r#r-/Union[BufferedReader, bytes, FileIO, str, Path]rdOptional[FileOptions]rr)rAr#rxintryrrdict[str, str])r}z list[str]rxrryrrlist[dict[str, str]])rAr#ryrrr#)rr#rr#rr)r}rrzlist[dict[str, Any]])rAr#rr)rAr#rbool)NN)rAz Optional[str]ryz Optional[ListBucketFilesOptions]rr)rAr#ryrrra) r5zLiteral['POST', 'PUT']rAr#r-rrdrrr)rAr#r-rrdrrr)rAr#rr#)__name__ __module__ __qualname____doc____annotations__r;rOrkr{rrrrrrrrrorrrurBrr:r"r" s42 G  -1)-# *  '    0 :/3 @>@>@>> @> , @>  @>FAC))%()3=) )XUW'',/':Q' 'REG TD,,  $++ +(#482  <DF 6/3 Q>&Q>Q>> Q> , Q>  Q>n/3 NN>N, N  N2/3 MM>M, M  M#rr"FreprceZdZdZy)r zRepresents a storage bucket.N)rrrrrrr:r r s&rc8eZdZUdZded<edZded<y) AsyncBucketProxyzPA bucket proxy, this contains the minimum required fields to query the File API.r#r$Frrr%N)rrrrrrr%rrr:rrsZ G e,G[,rr). __future__rrr( urllib.parserC dataclassesrriorrpathlibrtypingr r r r r httpxrr constantsrr exceptionsrtypesrrrrrrrrrrutilsrr__all__r"r rrrr:rs" (%66+D(   2 /s#s#l '*'' -.- -r