o xi@sdZddlmZddlZddlZddlmZddlZddlm Z ddl m Z ddl m Z mZddlmZddlmZd d lmZmZd d lmZmZd d lmZd d lmZd dlmZmZm Z ddl!m"Z"m#Z#ddl$m%Z%e&e'Z(Gddde Z)dS)z%Implementation of the docker builder.) annotationsN)Any)JobAndRunStatusTracker)AbstractBuilderregistry_from_uri)AbstractEnvironment)AbstractRegistry) EntryPoint LaunchProject)LaunchDockerError LaunchError)AnonynmousRegistry) LocalRegistry) LOG_PREFIXevent_loop_thread_exec$warn_failed_packages_from_build_logs)_WANDB_DOCKERFILE_NAMEvalidate_docker_installation)BuildContextManagerc@sReZdZdZdZdZdd d Zedd dZd ddZ d ddZ d!d"ddZ dS)# DockerBuilderzxBuilds a docker image for a project. Attributes: builder_config (Dict[str, Any]): The builder config. dockerz linux/amd64 environmentrregistryrconfigdict[str, Any]cCs||_||_||_dS)aInitialize a DockerBuilder. Arguments: environment (AbstractEnvironment): The environment to use. registry (AbstractRegistry): The registry to use. Raises: LaunchError: If docker is not installed N)rrr)selfrrrr[/home/ubuntu/.local/lib/python3.10/site-packages/wandb/sdk/launch/builder/docker_builder.py__init__*s zDockerBuilder.__init__returncCs@|d}|r|durttd|dt|}||||S)aCreate a DockerBuilder from a config. Arguments: config (Dict[str, Any]): The config. registry (AbstractRegistry): The registry to use. verify (bool, optional): Whether to verify the functionality of the builder. login (bool, optional): Whether to login to the registry. Returns: DockerBuilder: The DockerBuilder. destinationNz.Overriding registry from registry config with z from builder config.)getwandbtermwarnrr)clsrrr image_urirrr from_config=s  zDockerBuilder.from_configNonecstIdHdS)zVerify the builder.N)r)rrrrverify\szDockerBuilder.verifycszt|jtrttddSt|jtr!ttddS|jIdH\}}tt j }||||jj IdHdS)zLogin to the registry.z'No registry configured, skipping login.z#Anonymous registry, skipping login.N) isinstancerr_loggerinforrget_username_passwordrrloginuri)rusernamepasswordr/rrrr/`s   zDockerBuilder.loginNlaunch_projectr entrypointr job_trackerJobAndRunStatusTracker | Nonestrc s|IdH|IdHt|d}|d\}}tj|t}|js'dn|j IdH}|r9|d|} n|j d|} | sP|j | IdHrP| St d| dzttj| g|||jddIdH} t| | |j|Wntjy} z|r|d td | d} ~ wwzt|Wntytd |d } t | Ynw|r| d\} }ttd | ttj| |IdH}|durt d|j!dkrd|d|vrt d|| S)zBuild the image for the given project. Arguments: launch_project (LaunchProject): The project to build. entrypoint (EntryPoint): The entrypoint to use. N)r3r:zimage z0 does not already exist in repository, building.platform)tagsfile context_pathr9buildz(Error communicating with docker client: zTemporary docker context file z was not deleted.zPushing image z"Failed to push image to repository sagemakerzThe push refers to repository []z'Unable to push image to ECR, response: )"r*r/rcreate_build_contextospathjoinrr get_repo_uri image_namebuild_requiredcheck_image_existsr,r-rrr=rr#rapi DockerError set_err_stager remove Exceptionrsplitr$termlogpushr resource)rr3r4r5build_context_managerbuild_ctx_path image_tag dockerfile repositoryr'outpute_msgregtag push_resprrr build_imageksd         zDockerBuilder.build_image)rrrrrr)rrrrrrr!r)r!r))N)r3r r4r r5r6r!r7) __name__ __module__ __qualname____doc__ builder_typetarget_platformr classmethodr(r*r/r\rrrrrs    r)*r` __future__rloggingrAtypingrr$ wandb.dockerr)wandb.sdk.launch.agent.job_status_trackerr!wandb.sdk.launch.builder.abstractrr%wandb.sdk.launch.environment.abstractr"wandb.sdk.launch.registry.abstractr _project_specr r errorsr r registry.anonrregistry.local_registryrutilsrrrr=rrcontext_managerr getLoggerr]r,rrrrrs(