o ;i@sddlZddlZddlmZddlmZddlZddlmZddl m Z ddl m Z ddlm Z mZddlZddlmZdd lmZdd lmZdd lmZdd lmZmZmZdd lmZddlmZddl m!Z!ddl"m#Z#ddl$m%Z%eddddZ&e&j'ddddej(edfdee)dee*fddZ+dd Z,e&j'd!d"ddefd#e)dee)fd$d!Z-e&j'd%d&d'dej(ej d(d)efd*e)d+e)dee)fd,d%Z.e&j'd-d.d'dej(ej d(d)efd*e)d/e)d0e)dee)fd1d-Z/Gd2d3d3e0Z1e&j'd4d'd5ej(e d6defd*e)d0e)d7e)d8e*dee)f d9d4Z2e&j'd:d;d'dej(ej3ddd?efd*e)d0e)d@e*dee)fdAd:Z4e&j'dBdCddej(e dDd?eefdEe)dFe*dee)fdGdBZ5dS)HN)Path)Optional) UsageError)Syntax)Table)ArgumentTyper)display_location) synchronizer)timestamp_to_localized_str)_volume_download) ENV_OPTION YES_OPTION display_table)_Client) ensure_env)_NetworkFileSystem) OutputManager)api_pb2nfsz5Read and edit `modal.NetworkFileSystem` file systems.T)namehelpno_args_is_helplistz+List the names of all network file systems. Management)rrrich_help_panelFenvjsoncst|}tIdH}|jtj|dIdH}|r!d|dnd}gd}g}|jD]}||j t |j t |j |gq,t|||d|ddS)Nenvironment_namez in environment '')NameLocationz Created atzShared Volumes)title)rrfrom_envstubSharedVolumeListrSharedVolumeListRequestitemsappendlabelr cloud_providerr created_atr)rrclientresponseenv_part column_namesrowsitemr4Q/home/ubuntu/.local/lib/python3.10/site-packages/modal/cli/network_file_system.pylist_s  r6cCs d|dS)NzS @app.function(network_file_systems={"/my_vol": modal.NetworkFileSystem.from_name("z0")}) def some_func(): os.listdir("/my_vol") r4)r+r4r4r5gen_usage_code1sr7createz#Create a named network file system.rcCsNt|tjj||dt}|d|dtt|d}||dS)NrzCreated volume 'z'. Code example: python) rmodalNetworkFileSystemcreate_deployedrgetprintrr7)rroutputusager4r4r5r89s lsz4List files and directories in a network file system.zFile operations/)default volume_namepathc st|t|}||IdH}tjrSt}| d|d|dt }| d| d|D]}|j t jjjkrBdnd}||j|q6| |dS|D]}t |jqUdS)NzDirectory listing of 'z' in 'r filenametypedirfile)rr from_namelistdirsysstdoutisattyrr=r>r add_columnrGr FileEntryFileType DIRECTORYadd_rowrE) rDrErvolumeentriesr?tableentryfiletyper4r4r5rAFs"     putaUpload a file or directory to a network file system. 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. local_path remote_pathc s,t|t|}|dr|tj|}t}t | rV| d}|j |||j dIdH|j ddWdn1sCwY|d|d|ddSd |vr^td | d}|j|||j dIdH}|j ddWdn1swY|d |d|d |d dS)NrBupload progress_cbT)completezUploaded directory 'z' to 'r *z(Glob uploads are currently not supportedzUploaded file 'z' (z bytes written))rrrJendswithosrEbasenamerr=ris_dirtransfer_progress add_local_dirprogressstep_completedradd_local_file)rDrZr[rrTr?rg written_bytesr4r4r5rYfs&     "c@seZdZddZdS)CliErrorcCs ||_dS)N)message)selfrlr4r4r5__init__s zCliError.__init__N)__name__ __module__ __qualname__rnr4r4r4r5rks rkr=)rr.local_destinationforcec svt|t|}t|}t}|d}t|||||jdIdHWdn1s/wY| ddS)aDownload a file from a network file system. Specifying a glob pattern (using any `*` or `**` patterns) as the `remote_path` will download all matching files, preserving their directory structure. For example, to download an entire network file system into `dump_volume`: ``` modal nfs get "**" dump_volume ``` Use "-" as LOCAL_DESTINATION to write file contents to standard output. downloadr]Nz$Finished downloading files to local!) rrrrJrr=rer rgrh) rDr[rsrtr destinationrTr?rgr4r4r5r=s  rmz6Delete a file or directory from a network file system.z-rz --recursivezDelete directory recursively)r recursivecs@t|t|}|j||dIdHt|ddS)N)rxz was deleted successfully!)rrrJ remove_filerr=rh)rDr[rxrrTr4r4r5rws  deletez3Delete a named, persistent modal.NetworkFileSystem.zAName of the modal.NetworkFileSystem to be deleted. Case sensitivenfs_nameyescsNtj||dIdH|stjd|ddddtj||dIdHdS)NrzIAre you sure you want to irrevocably delete the modal.NetworkFileSystem 'z'?FT)rCabort)rrJhydratetyperconfirmrz)r{r|rr4r4r5rzs  )6rbrLpathlibrtypingrrclickr rich.syntaxr rich.tablerrrr:modal._locationr modal._utils.async_utilsr modal._utils.time_utilsr modal.cli._downloadr modal.cli.utilsr rr modal.clientrmodal.environmentsrmodal.network_file_systemr modal.outputr modal_protornfs_clicommandcreate_blockingstrboolr6r7r8rArY Exceptionrkr=Optionrwrzr4r4r4r5s              $