& hdZddlmZddlmZddlmZddlmZddlmZdZ dd e d fd Z d d Z y ) z>Functions required to interact with Docker to run a container.)absolute_import)division)unicode_literals)utils)configz/tmp/keys/cred_key.jsonFNc|g}|rddgng}|rddj||g}n>tjxstj}ddj||g}ddj|g}dg|z|z|zdd gz|zS) a;Returns a list of 'docker run' options. Args: enable_gpu: (bool) using GPU or not. service_account_key: (bool) path of the service account key to use in host. cred_mount_path: (str) path in the container to mount the credential key. extra_run_opts: (List[str]) other custom docker run options. z --runtimenvidiaz-vz{}:{}z-ez!GOOGLE_APPLICATION_CREDENTIALS={}z--rmz--ipchost)formatrADCEnvVariable ADCFilePath) enable_gpuservice_account_keycred_mount_pathextra_run_optsruntimemount adc_file_pathenv_vars /lib/googlecloudsdk/command_lib/ai/docker/run.py_DockerRunOptionsrsN'1[( #r' 7>>"5G HE))+Cv/A/A/CM 7>>-A BE 6==oN O' G e #g -&12 24B CCcv|g}|g}t|||}ddg|z|gz|z}tj|y)aCalls `docker run` on a given image with specified arguments. Args: image_name: (str) Name or ID of Docker image to run. enable_gpu: (bool) Whether to use GPU service_account_key: (str) Json file of a service account key auth. run_args: (List[str]) Extra custom options to apply to `docker run` after our defaults. user_args: (List[str]) Extra user defined arguments to supply to the entrypoint. N)rrrdockerrun)rrExecuteDockerCommand) image_namerrrun_args user_argsrun_optscommands r RunContainerr"8sX$HI -( u  (J< 7) C'W%r)FNNN) __doc__ __future__rrr$googlecloudsdk.command_lib.ai.dockerrgooglecloudsdk.corer _DEFAULT_CONTAINER_CRED_KEY_PATHrr"rrr)sGE&'6&#< "'*.&F%)C<"%) &r