o ;iE1@sddlZddlZddlZddlmZmZddlmZmZm Z ddl Z ddl m Z ddl mZddlmZddlmZdd lmZmZdd lmZdd lmZdd lmZdd lmZddlmZddl m!Z!ddl"m#Z#ddl$m%Z%ddl&m'Z'm(Z(m)Z)ddl*m+Z+ddl,m-Z-m.Z.dedefde/e0e j1j2ffddZ3de/e0e j1j2fde/e0efdee0ge4fde5e0fddZ6d e0d!e0de4fd"d#Z7d e e0de4fd$d%Z8d e0d&e0d'e4ddfd(d)Z9d e0d*e4defd+d,Z:d-ed.e5e0d/e0d0e;d1ed'e4ddfd2d3ZZ=e j>dd?d@e j?dAdBdCdDdEe-e j?ddFdGdEe j?ddHdIdEe j?ddJdKdEe j?ddLdMdEe j?ddNdOdEe j?ddPdQdEe j?ddRdSdEe j?ddTdUdEe j?ddVdWdEe j?ddXdYdEe j?ddZd[dEe j?d\d]d^dEfd e e0d&e0d/e e0d_e e0d`e e0d5e e5e0d7e e5e0d6e e5e0d8e e;d9e e;d:e e0d;e e0d$sz*_params_from_signature..css|] }t|tjjVqdS)N) isinstancetypermodels ParameterInfo)r!r#rrr$ %sz)_params_from_signature..zAll params to z, must be of type typer.models.ParameterInfo.)inspect signature parametersitemsallvalues__name__)rsigparamsrrr$_params_from_signature s  r4 param_objs passed_argsallowedcCsXg}|D]#\}}||rq|jdusJd|||jkr)|d|jq|S)zDCheck which forbidden arguments were passed with non-default values.NzDAll params must be typer.models.ParameterInfo, and have param_decls./)r. param_declsgetr appendjoin)r5r6r7passed_forbiddenr" param_objrrr$_passed_forbidden_args+sr?refprefixcCsD|dsJ||o!t|t|ddko!|t|dS)N-r)endswith startswithlenisalnum)r@rArrr$_is_valid_modal_id>s6rGcCs |durdSt|dpt|dS)NFsb-ta-)rG)r@rrr$_is_running_container_refCsrJcmdptyc Cst|dr modal shell z ::my_functionzThe selected module 'z' has the following choices: z$' has no Modal functions or classes.z z0Referenced entity is not a Modal Function or Cls) rrfile_or_modulerrr&rcls_get_class_service_functionspecr ValueError)r@rX import_refrunnableall_usable_commands help_header help_footerclass_service_functionrrr$_function_spec_from_ref\s(     rhappcmdsenvtimeout function_speccCsRt|f||||j|j|j|j|j|j|j|j|j |j r|j j nd||j ddS)N)rjenvironment_namerlimagemountssecretsnetwork_file_systemsgpucloudcpumemoryvolumesregionrLproxy) rrorprqrrgpusrtrurvrwscheduler_placementregionsry)rirjrkrlrmrLrrr$_start_shell_from_function_specys& r} modal_imagevolumesecret add_localrurvrsrtrxcCsdd|D}dd|D}g}|D]*}t|}td|j}|r0tj||d}ntj||d}| |qt |||||||| | | ||| rQ| dng| ddS) NcSsi|] }d|t|qS)/mnt/)r from_name)r!volrrr$r%sz+_start_shell_from_image..cSsg|]}t|qSr)rr)r!srrr$ sz+_start_shell_from_image..r) remote_path,) rjrnrlrorprurvrsrtrwrqrxrL) r expanduserresolvernameis_dirr_from_local_dir_from_local_filer;rrT)rirjrkrlr~rrrrurvrsrtrxrLrwrqrplocal_path_str local_pathrmrrr$_start_shell_from_images4  rzID of running container or Sandbox, or path to a Python file containing an App. Can also include a Function specifier, like `module.py::func`, if the file defines multiple Functions.)r helpz /bin/bashz-cz--cmdz&Command to run inside the Modal image.)rz--imagez env.txt ``` Connect to a running Sandbox by ID: ``` modal shell sb-abc123xyz ``` NWindowsz3`modal shell` is currently not supported on WindowscS|dvS)N>rKrLr@rprrr$<zshell..)r7z2Cannot specify container configuration arguments (z, z>) when attaching to an already running container or Sandbox ('z').rYz/bin/bash -c ""izim-cSr)N>rKrkrLr@rXrrrrr$rQrz<) when starting a new container from a function reference ('cSs|dvS)N>rorrrrrr$r^rzCannot specify zC argument(s) when starting a new container from a Modal Image ID (')r)rplatformsystemr r4shellrJr?localsrr<rWr r rSrTrGrhr}rrO from_registryr)r@rKrkrorrrrrurvrsrtrxrLrXr5r=rirjrlrmr~rrr$rsvl          r)Ar+rrSpathlibrrtypingrrrr'clickr _functionsr rir environmentsr exceptionr r functionsrrormountrrunnerrrUrrrrr containerr import_refsrrrrunrutilsrrdictrRr(r)r4boollistr?rGrJrWrhintr}rArgumentOptionrrrrr$sN                          1      #&).5=>