gADddlmZddlZddlZddlZddlmZmZddl 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)StorageException SyncClient SyncBucketcleZdZUdZded<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)*SyncBucketActionsMixinz(Functions needed to access the file API.stridr_clientNc Z |jj||f|xsi||d|}|j|r0d|vr,t|ddtr|ddj|S#t$r5}|jj } t | d| d| dd}~wwxYw)N)headersjsonfilesmessageerror statusCodefile) r%requestraise_for_statusrresponser(r isinstancerclose) selfmethodurlr'r(r)kwargsr1excresps L/var/www/openai/venv/lib/python3.12/site-packages/storage3/_sync/file_api.py_requestzSyncBucketActionsMixin._request&s V+t||++%,]UNTH  % % ' Vu_E&M!4Dn)U &M!  " " $ V<<$$&D!$y/4=$|BTU U Vs6A,, B*50B%%B*c|j|}|jdd|}|j}tjj t |jj|dz}tjj|j}|jds td|j|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/r6tokenzNo 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_urlz/SyncBucketActionsMixin.create_signed_upload_url>s$$T*==+?w)GH}}-3\\-B-B  %% &e 4. ||,,X^^< ("#=> >"//+!'*1-  c|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|| | | }|j%}t'||j 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 r>r?? 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+SyncBucketActionsMixin.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_==rPcdt|i}d}|jdr@|jd|di|jddurdnd|jd}|jdr|jd|di|j|}|j dd|| }|j }t jj|d }t jj|jd |jz}|jjtt|jd ||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( signedURLrRrV)r#rIupdaterBr;r(rCrDrEquoterArHr%rFr lstrip) r4rA expires_inoptionsr(download_queryr1rLr6s r:create_signed_urlz(SyncBucketActionsMixin.create_signed_urlsUS_- ;;z " DKKWZ%89 :;;z*d2!'++j"9!:;  ;;{ # DKKgk&:; <##D)== D6 "!  }}ll##D$56ll  *q _<||$$%d3n&;&;C&@%A.AQ R [ rPc~|t|d}d}|jdrL|jd|jdi|jddurdnd|jd}|jdd|j|}|j }|D]}t jj|d } t jj| jd | jz} |jjtt| jd ||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 the file. )pathsrnrorpTrqr=rsrtrurRrV)r#rIrvr;r$r(rCrDrErwrArHr%rFr rx) r4r~ryrzr(r{r1rLitemr6s r:create_signed_urlsz)SyncBucketActionsMixin.create_signed_urlss3S_= ;;z " DKKW[[%<= >;;z*d2!'++j"9!:;  == DGG9 %!  }}D,,''[(9:C,,$$SXX.1SYYK@C<<(()$sC.*?*?*D)EnEUV    rPcg}d}|jdr)|jddurdnd|jd}|r|j||jdrdnd}|jdr.tjj |jdnd}|r|j|d j |}d |}|j |}|jj|d ||S) z Parameters ---------- path file path, including the path and file name. For example `folder/image.png`. rorpTrqrrz render/imageobjectN&rRz/public/) rIappendrCrDr`joinrBr%rF) r4rArz _query_stringr{ render_pathtransformation_query query_stringrKs r:get_public_urlz%SyncBucketActionsMixin.get_public_urls  ;;z ";;z*d2!'++j"9!:;     0(/ K(@nh {{;' LL " "7;;{#; <   !5 6xx . <.) $$T*,,''( XeW\NSSrPcf|jdd|j||d}|jS)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 sourceKeydestinationKeyrtr;r$r(r4 from_pathto_pathress r:movezSyncBucketActionsMixin.move=mm   GG&") xxzrPcf|jdd|j||d}|jS)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/copyrrtrrs r:copyzSyncBucketActionsMixin.copy$rrPch|jdd|jd|i}|jS)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/prefixesrtr)r4r~r1s r:removezSyncBucketActionsMixin.remove:s>== twwi e$!  }}rPcf|jdd|jd|}|jS)z Lists info for a particular file. Parameters ---------- path The path to the file. GET /object/info/rVrr4rAr1s r:infozSyncBucketActionsMixin.infoJs7== DGG9AdV , }}rPcd|jdd|jd|}|jdk(S)z Returns True if the file exists, False otherwise. Parameters ---------- path The path to the file. HEADrrV)r;r$ status_coders r:existszSyncBucketActionsMixin.exists\s<== DGG9AdV , ##s**rPc|xsi}ddi}it|d|xsdi}|jdd|j||}|jS)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/jsonprefixror=z /object/list/)r(r')rr;r$r()r4rArz extra_options extra_headersbodyr1s r:listzSyncBucketActionsMixin.listnsy 2 ');<  $   djb  == DGG9 %! !  }}rPc|jdrdnd}tjj|jdxsi}|rd|nd}|j |}|j d|d||}|j S)z Downloads a file. Parameters ---------- path The file path to be downloaded, including the path and file name. For example `folder/image.png`. rrzrender/image/authenticatedrrRrorrV)rIrCrDr`rBr;content)r4rArzrrrrKr1s r:rpzSyncBucketActionsMixin.downloads-4KK ,D ((  &||55gkk+6N6TRTU5I/01r $$T*== m1UGL> 2 rPc|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|| | |}|j }t%||j'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. NrSupsertzx-upsertmetadatar'z x-metadatar=rVr.rWrYrTrUr-rZr[rr]r^r_)rcrvr%r'rr(dumpsbase64 b64encodeencoderar2rrbrrdrBr;rrI)r4r5rAr-rerhrirrfile_opts_headersr' metadata_strrjr)rKr1rLs r:_upload_or_updatez(SyncBucketActionsMixin._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_==rPc*|jd|||S)rr=rr4rAr-res r:uploadzSyncBucketActionsMixin.uploads&%%fdD,GGrPc*|jd|||S)Nr\rrs r:rvzSyncBucketActionsMixin.update s %%eT4FFrPc$|jd|S)NrV)r$)r4rAs r:rBz&SyncBucketActionsMixin._get_final_paths''!D6""rP)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]reOptional[FileOptions]rr)rAr#ryintrzrrdict[str, str])r~z list[str]ryrrzrrlist[dict[str, str]])rAr#rzrrr#)rr#rr#rr)r~rrzlist[dict[str, Any]])rAr#rr)rAr#rbool)NN)rAz Optional[str]rzz Optional[ListBucketFilesOptions]rr)rAr#rzrrrb) r5zLiteral['POST', 'PUT']rAr#r-rrerrr)rAr#r-rrerrr)rAr#rr#)__name__ __module__ __qualname____doc____annotations__r;rOrlr|rrrrrrrrrprrrvrBrPr:r"r" s22 G  -1)-# *  '    0 :/3 @>@>@>> @> , @>  @>FAC))%()3=) )XUW'',/':Q' 'R?A TD,,  $++ +(#482  <>@ 6/3 Q>&Q>Q>> Q> , Q>  Q>n/3 HH>H, H  H2/3 GG>G, G  G#rPr"FreprceZdZdZy)r zRepresents a storage bucket.N)rrrrrrPr:r r s&rPc8eZdZUdZded<edZded<y) SyncBucketProxyzPA bucket proxy, this contains the minimum required fields to query the File API.r#r$Frrr%N)rrrrrrr%rrPr:rrsZ GU+GZ+rPr). __future__rrr( urllib.parserC dataclassesrriorrpathlibrtypingr r r r r httpxrr constantsrr exceptionsrtypesrrrrrrrrrrutilsrr__all__r"r rrrPr:rs" (%66+D(   1 .s#s#l ''' ,,, ,rP