C0dZddlmZddlmZddlmZddlZddlZddlZddlZddl Z ddl m Z ddl m Z ddlmZdd lmZdd lmZd Zd Zd ZdZdZdZdZdZddZ ddZddZdZ ddZ ddZ!y)z;Functions required to interact with Docker to build images.)absolute_import)division)unicode_literalsN)errors) local_util)utils)log) shlex_quotez/homez/usr/app ./setup.py./requirements.txtzcloudai-autogeneratedc6d}tj||duS)z41F1FGHH WW  gnn -FAs%(E\.! }Jzz*'778H  ! !( ++rctjj|jxsd}t ||d}dj |S)atReturns the Dockerfile entries required to append at the end before entrypoint. Including: - copy the parent directory of the main executable into a docker container. - inject an entrypoint that executes a script or python module inside that directory. Args: package: (Package) Represents the main application copied to and run in the container. .z4Copy the source directory into the docker container.r+r,)r/r0dirnamerCr'r)rD parent_dir copy_codes r_PreparePackageEntryrPsDwww~~.5#*"DF)  ##rc  t|} tjdj|t |t |} | t | z } | t | |||||z } | t|z } | t|| z } | S)akGenerates a Dockerfile for building an image. It builds on a specified base image to create a container that: - installs any dependency specified in a requirements.txt or a setup.py file, and any specified dependency packages existing locally or found from PyPI - copies all source needed by the main module, and potentially injects an entrypoint that, on run, will run that main module Args: base_image: (str) ID or name of the base image to initialize the build stage. main_package: (Package) Represents the main application to execute. container_workdir: (str) Working directory in the container. container_home: (str) $HOME directory in the container. requirements_path: (str) Rath of a requirements.txt file. setup_path: (str) Path of a setup.py file extra_requirements: (List[str]) Required dependencies to install from PyPI. extra_packages: (List[str]) User custom dependency packages to install. extra_dirs: (List[str]) Directories other than the work_dir required to be in the container. Returns: A string that represents the content of a Dockerfile. ab FROM {base_image} # The directory is created by root. This sets permissions so that any user can # access the folder. RUN mkdir -m 777 -p {workdir} {container_home} WORKDIR {workdir} ENV HOME={container_home} # Keeps Python from generating .pyc files in the container ENV PYTHONDONTWRITEBYTECODE=1 ) base_imageworkdircontainer_home)r2r3r4r5r6) rr-r.rr rr=rPrJ) rR main_packagecontainer_workdirrTr2r3r4r5r6is_training_prebuilt_image_base dockerfiles r_MakeDockerfilerYsB%C:$N!  &/0$^4  6 7* *+JKK* "%)+# * $\22* #L$CEE* rc d|g} | rdgng} ddg| z| zdd|gz} tjjtjj|t}|rtnd}tjjtjj|t }|rt nd}|xst }|xst}tj|jtjtj|jtjtj|}t|f|||||||d | }d j| }tjd j!|t#j$| | }|d k(rtj&|||St)j*dj!||}t-j.|| |)aBuilds a Docker image. Generates a Dockerfile and passes it to `docker build` via stdin. All output from the `docker build` process prints to stdout. Args: base_image: (str) ID or name of the base image to initialize the build stage. host_workdir: (str) A path indicating where all the required sources locates. main_script: (str) A string that identifies the executable script under the working directory. output_image_name: (str) Name of the built image. python_module: (str) Represents the executable main_script in form of a python module, if applicable. requirements: (List[str]) Required dependencies to install from PyPI. extra_packages: (List[str]) User custom dependency packages to install. container_workdir: (str) Working directory in the container. container_home: (str) the $HOME directory in the container. no_cache: (bool) Do not use cache when building the image. **kwargs: Other arguments to pass to underlying method that generates the Dockerfile. Returns: A Image class that contains info of the built image. Raises: DockerError: An error occurred when executing `docker build` z-tz --no-cachedockerbuildz--rmz-f-N)rC package_pathrA)rUrTrVr2r3r4r5 zRunning command: {}) input_strrzN Docker failed with error code {code}. Command: {cmd} )coder%)r/r0isfiler _DEFAULT_SETUP_PATH_DEFAULT_REQUIREMENTS_PATH _DEFAULT_HOME_DEFAULT_WORKDIRrPackagereplacesep posixpathrYr inforrExecuteCommandImager-r.r DockerError)rR host_workdir main_scriptoutput_image_namerA requirementsr5rVrTno_cachekwargs tag_options cache_argscommand has_setup_pyr3has_requirements_txtr2home_dirwork_dirrUrXjoined_command return_code error_msgs r BuildImager~sR()+!) ~r* w  ( ),2E<+H I' \;N OP,&2"*ggll4H0d  ,}(  2"2(   7'' >!#,   )%#   *88G$.(( ' ' 78))'ZH+A ;;((H ==!  F F 8:I   Y ==r)N)FNNNNN)F)NNNNN)NNNNNT)"__doc__ __future__rrrrr/rirr-googlecloudsdk.command_lib.air)googlecloudsdk.command_lib.ai.custom_jobsr$googlecloudsdk.command_lib.ai.dockerrgooglecloudsdk.corer six.movesr rdrerbrc_AUTONAME_PREFIX_AUTOGENERATED_TAG_LENGTHrrr'r=rJrPrYr~rrrrsB&'  0@6#! "1*IF  "*/)-"&*.&*"& ? D,6$4'+#'+#'#@N" "!%"U>r