o ;i+@sddlZddlZddlmZddlmZddlZddlmZddl m Z ddlm Z m Z m Z ddlZddlmZddlmZdd lmZdd lmZdd lmZmZmZdd lmZdd lmZddlm Z m!Z!ddl"m#Z#e ddddZ$de%de&fddZ'e$j(ddddee dddfde&dee&d ee%fd!dZ)e$j(d"d#d$ej*e d%d&efd'e&d(e&d)e&d*e+dee&f d+d"Z,e$j(d,d-ddej*ed&fdee&d.e+fd/d0Z-e$j(d1d2d#dej*e d3d4d&efd'e&d5e&d.e+dee&fd6d1Z.e$j(d7d8d#dej*e e d3d4e d&d9d:d;d<efd'e&d=e&d(e&d*e+dee&f d>d7Z/e$j(d?d@d#dej*e d&dAdBdCd<efd'e&d(e&dDe+dee&fdEd?Z0e$j(dFdGd#dej*e d&dAdBdHd<efd'e&dIe1e&dDe+dee&fdJdFZ2e$j(dKdLddej*e dMd<fe d&dNdOd<eedPde&dQe+dRe+dee&fdSdKZ3e$j(dTdUddej*eefdVe&dWe&dRe+dee&fdXdTZ4e$j(dYdZddej*e d[d<efd'e&dee&fd\dYZ5dS)]N)Path)Optional) UsageError)Syntax)ArgumentOptionTyper) synchronizer)open_url_and_display)timestamp_to_localized_str)_volume_download) ENV_OPTION YES_OPTION display_table) ensure_env) OutputManager)#_AbstractVolumeUploadContextManager_Volume)api_pb2volumeTz Read and edit `modal.Volume` volumes. Note: users of `modal.NetworkFileSystem` should use the `modal nfs` command instead. )nameno_args_is_helphelpvaluereturncCsv|dkrtdd}d}d}t|}||kr|ddSt|D]\}}||d}||kr0nq |||||S) Nrzvalue should be >= 0)z KiBz MiBz GiBz TiBz PiBz EiBz ZiBz%.1fiz0.0fz B) ValueErrorfloat enumerate)rsuffixformatbasebytes_isunitr&D/home/ubuntu/.local/lib/python3.10/site-packages/modal/cli/volume.pyhumanize_filesize"s  r(createz(Create a named, persistent modal.Volume. Management)rrrich_help_panelz VolumeFS version. (Experimental))defaultrrenvversioncCs`t|}tjjj|||dd|d}t}|d|d|dt|d}||dS)N)environment_namer.z; @app.function(volumes={"/my_vol": modal.Volume.from_name("z0")}) def some_func(): os.listdir("/my_vol") zCreated Volume 'z' in environment 'z'. Code example: python) rmodalVolumeobjectsr)rgetprintr)rr-r.env_name usage_codeoutputusager&r&r'r)2s r4zFile operations)rr+.F volume_name remote_pathlocal_destinationforcec szt|t|}tj||d}t}|d}t|||||jdIdHWdn1s1wY| ddS)aDownload files from a modal.Volume object. If a folder is passed for REMOTE_PATH, the contents of the folder will be downloaded recursively, including all subdirectories. **Example** ``` modal volume get logs/april-12-1.txt modal volume get / volume_data_dump ``` Use "-" as LOCAL_DESTINATION to write file contents to standard output. r/download)rr<r= overwrite progress_cbNz$Finished downloading files to local!) rrr from_namerr4transfer_progressr progressstep_completed) r;r<r=r>r- destinationrr8rEr&r&r'r4Fs listz?List the details of all modal.Volume volumes in an Environment.jsoncsnt|}tjj|dIdH}g}|D]}|IdH}||jt|j ||j fqt gd||dS)Nr?)Namez Created atz Created by) rrr3rHinfoappendrr created_at timestamp created_byr)r-rIvolumesrowsobjrKr&r&r'list_ls"rSlsz4List files and directories in a modal.Volume volume./)r,pathc st|tj||d}||IdH}|s'tjs'|D]}t|jqdSg}|D]G}|j t j j j kr8d}n#|j t j j jkrCd}n|j t j j jkrNd}n |j t j j jkrYd}nd}||jdd|t|jd t|jfq+gd } d |d |d } t| ||| ddS)Nr?dirlinkfifosocketfileunicode_escapezutf-8F)FilenameTypezCreated/ModifiedSizezDirectory listing of 'z' in '')title)rrrClistdirsysstdoutisattyr5rVtyper FileEntryFileType DIRECTORYSYMLINKFIFOSOCKETrLencodedecoder mtimer(sizer) r;rVrIr-volentriesentryrQfiletypecolumnsrar&r&r'rT}s:   putzUpload a file or directory to a modal.Volume. Remote parent directories will be created as needed. Ending the REMOTE_PATH with a forward slash (/), it's assumed to be a directory and the file will be uploaded with its current name under that directory. z-fz--forcezOverwrite existing files.)r local_pathc st|tj||dIdH}|dr|tj|}t }t | r| dL}z1t j|jj|j|j|j|d4IdH}|||WdIdHn 1IdHsYwYWntyq} ztt| d} ~ wwWdn1s|wY|d|d|ddSd|vrtd | dL}z1t j|jj|j|j|j|d4IdH}|||WdIdHn 1IdHswYWnty} ztt| d} ~ wwWdn1swY|d |d|ddS) Nr?rUupload)rBr>zUploaded directory 'z' to 'r`*z(Glob uploads are currently not supportedzUploaded file ')rrrChydrateendswithosrVbasenamerr4ris_dirrDrresolve _metadatar. object_id_clientrE put_directoryFileExistsErrorrstrrFput_file) r;rwr<r>r-rqr8rEbatchexcr&r&r'rvsb   (   (  rmz/Delete a file or directory from a modal.Volume.z-rz --recursivezDelete directory recursively recursivecsDt|tj||d}|j||dIdHt|ddS)Nr?)rz was deleted successfully!)rrrC remove_filerr4rF)r;r<rr-rr&r&r'rs  cpzsCopy within a modal.Volume. Copy source file to destination file or multiple source files to destination directory.zCopy directories recursivelypathscs:t|tj||d}|^}}||||IdHdS)Nr?)rrrC copy_files)r;rrr-r src_pathsdst_pathr&r&r'rs  deletez*Delete a named Volume and all of its data.z6Name of the modal.Volume to be deleted. Case sensitivez--allow-missingz(Don't error if the Volume doesn't exist.) allow_missingyesr-rrcsBt|}|stjd|ddddtjj|||dIdHdS)Nz>Are you sure you want to irrevocably delete the modal.Volume 'z'?FTr,abort)r/r)rtyperconfirmrr3r)rrrr-r&r&r'r s  renamezRename a modal.Volume.old_namenew_namecs8|stjd|ddddtj|||dIdHdS)Nz/Are you sure you want rename the modal.Volume 'z.'? This may break any Apps currently using it.FTrr?)rrrr)rrrr-r&r&r'r#s   dashboardz5Open the Volume's dashboard page in your web browser.z,Name of the Volume to view in the dashboard.cs<t|}tj||dIdH}d|j}t|ddS)zOpen a Volume's dashboard page in your web browser. **Example:** ``` modal volume dashboard my-volume ``` r?Nzhttps://modal.com/id/zVolume dashboard)rrrCrzrr )r;r-rurlr&r&r'r9s  )6r|rcpathlibrtypingrrclickr rich.syntaxrrrrr1modal._utils.async_utilsr modal._utils.browser_utilsr modal._utils.time_utilsr modal.cli._downloadr modal.cli.utilsr rrmodal.environmentsr modal.outputr modal.volumerr modal_protor volume_cliintrr(commandr)create_blockingboolr4rSrTrvrrHrrrrr&r&r&r'sd              $  ( /