o ;i @sddlmZmZddlZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZdd lmZmZmZmZdd lmZdd lmZdd lmZdd lmZddlmZddlm Z ddl!m"Z"ej#ddddZ$e$%de j&edfdee'de(fddZ)e$%dej*ddej+ddddfd e'd!e(fd"dZ,e$%d#e j&ej+dd$d%ej*ddej*d&dfd'ee(d e'd(e-e'fd)d#Z.e$%d*e j&ej*ddfd e'fd+d*Z/dS),)OptionalUnionN)Column)Text)_get_environment_name) get_pty_info) synchronizer)timestamp_to_localized_str) ENV_OPTION display_tableis_ttystream_app_logs)_Client)config)_ContainerProcess) ensure_env) InvalidError) StreamType)api_pb2 containerz)Manage and connect to running containers.T)namehelpno_args_is_helplistFenvjsonc st|}tIdH}t|}|jtj|dIdH}tdddtddddd g}g}|j j d d d d |j D]}| |j |j |j|jrMt|j|ndgq:t|||d|ddS)z/List all containers that are currently running.N)environment_name Container ID) min_widthzApp IDzApp Namez Start TimecSs|jS)N) started_at)taskr#G/home/ubuntu/.local/lib/python3.10/site-packages/modal/cli/container.py*szlist_..T)keyreversePendingz"Active Containers in environment: )rtitle)rrfrom_envrstubTaskListrTaskListRequestrtaskssortappendtask_idapp_idapp_descriptionr!r r )rrclientrres column_namesrows task_statsr#r#r$list_s.     r9logsr)rz --timestampsz!Show timestamps for each log line container_id timestampscCsHd\}}|dr |}n|dr|}ntd|t|||ddS)z;Show logs for a specific container, streaming while active.)NNzsb-zta-zInvalid container ID: )r1 sandbox_idshow_timestampsN) startswithrr )r;r<r1r=r#r#r$r:8s  execzRun the command using a PTY.)defaultrzA command to run inside the container. To pass command-line flags or options, add `--` before the start of your commands. For example: `modal container exec -- /bin/bash -c 'echo hi'`ptycommandcs|durt}tIdH}tj|||rtddndtdd}|j |IdH}|r;t |j || IdHdSt |j ||t jt jdIdHdS)z!Execute a command in a container.NT)shellfunction_runtime_debug)r1rCpty_info runtime_debug)stdoutstderr)r rr*rContainerExecRequestrrgetr+ ContainerExecrexec_idattachrSTDOUTwait)rBr;rCr4reqr5r#r#r$r@Hs$  stopcs2tIdH}tj|d}|j|IdHdS)zStop a currently-running container and reassign its in-progress inputs. This will send the container a SIGINT signal that Modal will handle. N)r1)rr*rContainerStopRequestr+ ContainerStop)r;r4requestr#r#r$rRks )0typingrrtyper rich.tabler rich.textr modal._objectrmodal._output.ptyrmodal._utils.async_utilsrmodal._utils.time_utilsr modal.cli.utilsr r r r modal.clientr modal.configrmodal.container_processrmodal.environmentsrmodal.exceptionrmodal.stream_typer modal_protorTyper container_clirCcreate_blockingstrboolr9ArgumentOptionr:rr@rRr#r#r#r$sZ                  !"