o x[h @sUdZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z m Z mZddlmZddlmZddlmZdd lmZmZdd lmZdd lmZd eegd gdZeed<eeZ dZ!Gdddej"Z#Gddde#Z$Gddde#Z%de&dedede'ddf ddZ(de)fddZ*de)de)fd d!Z+d"e#de)fd#d$Z,de)d e#fd%d&Z-de)d e#fd'd(Z.dS))z9ansible enables running on first boot either ansible-pullN)deepcopy)Optional) lifecyclesignal_handlersubp)Cloud)Config) MetaSchema) ALL_DISTROSDistro) PER_INSTANCE)get_cfg_by_path cc_ansibleansible)id frequencydistrosactivate_by_schema_keysmetaansible_configc@s|eZdZdefddZdeejfddZde fddZ d d Z d e fd d Z ddZejddZejde fddZdS) AnsiblePulldistrocCs<dg|_ddg|_||_i|_d|_tjdd|jd<dS)Nz ansible-pullz --versionHOMEz/root)cmd_pull cmd_versionrenvrun_userosenvironget)selfrr!=/usr/lib/python3/dist-packages/cloudinit/config/cc_ansible.py__init__ s  zAnsiblePull.__init__returncCsH||j\}}|d}td|}|r"|d}tj |SdS)Nrz ([\d\.]+)) do_asr splitlinespopresearchgrouprVersionfrom_str)r stdout_ first_linematchesversionr!r!r" get_version+s   zAnsiblePull.get_versioncGs|g|j|\}}|SN)r%r)r argsr-r.r!r!r"pull4szAnsiblePull.pullcCs|stddS)Nz!command: ansible is not installed) is_installed ValueErrorr r!r!r" check_deps8szAnsiblePull.check_depscommandcKs0|js |j|fi|S|jj||jfi|Sr3)rrrr%r r:kwargsr!r!r"r%<szAnsiblePull.do_ascKsDttj|fd|ji|WdS1swYdS)N update_env)r suspend_crashrrr;r!r!r"rAs $zAnsiblePull.subpcCdSr3r!r8r!r!r"r6EzAnsiblePull.is_installedpkg_namecCr?r3r!r rAr!r!r"installIr@zAnsiblePull.installN)__name__ __module__ __qualname__r r#rrr+r2strr5r9listr%rabcabstractmethodr6rCr!r!r!r"rs   rcsVeZdZdedeeffdd ZddZddZd efd d Z d e fd dZ Z S)AnsiblePullPiprusercst|||_|dSr3)superr#radd_pip_install_site_to_pathr rrL __class__r!r"r#Os  zAnsiblePullPip.__init__cCs`|jr.|tjddg\}}|d}|jd}|r'd||g|jd<dS||jd<dSdS)Nz-cz&import site; print(site.getuserbase())z/bin/PATH:)rr%sys executablerrjoin)r user_baser. ansible_pathold_pathr!r!r"rNTs  z+AnsiblePullPip.add_pip_install_site_to_pathcCs6zddl}WdSty|j|jjgYdSw)Nr)pip ImportErrorrinstall_packagespip_package_name)r rZr!r!r"bootstrap_pip_if_requiredes  z(AnsiblePullPip.bootstrap_pip_if_requiredrAcCs||s@tjdddg}tjtjt ddr"| d|j r*| d| g|dd| g||d Sd S) zishould cloud-init grow an interface for non-distro package managers? this seems reusable -mrZrCstdlibzEXTERNALLY-MANAGEDz--break-system-packages--userz --upgradeN) r^r6rTrUrpathexistsrV sysconfigget_pathappendrr%)r rAcmdr!r!r"rCks$   zAnsiblePullPip.installr$cCs4tjdddg}|jr|d||\}}d|vS)Nr_rZrHrar)rTrUrrfr%)r rgr-r.r!r!r"r6s  zAnsiblePullPip.is_installed) rDrErFr rrGr#rNr^rCboolr6 __classcell__r!r!rPr"rKNs rKcsFeZdZdedeeffdd ZdefddZdefd d Z Z S) AnsiblePullDistrorrLcst|||_dSr3)rMr#rrOrPr!r"r#s  zAnsiblePullDistro.__init__rAcCs|s |j|gdSdSr3)r6rr\rBr!r!r"rCszAnsiblePullDistro.installr$cCsttdS)Nr)rhrwhichr8r!r!r"r6szAnsiblePullDistro.is_installed) rDrErFr rrGr#rCrhr6rir!r!rPr"rjsrjnamecfgcloudr4r$cCs|di}|d}|d}|d}|d}|d} |dd} |rlt||j} |d kr8t| |} nt| |} | | | |d d} | rS| | jt<|rZt || | rct | t | |rnt || dSdSdS) Nrrinstall_methodsetup_controllergalaxyr5 package_namerZr) rvalidate_configrrKrjrCr9r CFG_OVERRIDEansible_galaxyrun_ansible_pullransible_controller)rlrmrnr4 ansible_cfg ansible_userrorp galaxy_cfgpull_cfgrrrrrr!r!r"handles4             r}cCsd}|D]}t||std|d|q|dr/dD]}t||s.td|d|q|d}|rIt|d|dgsItd ||d }|d vrUtd dS) N)rorrzMissing required key 'z' from r5)zpull/urlzpull/playbook_namerp repositories run_ansiblezMissing required key from ro)rZrz Invalid install method {install})r r7rany)rm required_keyskeycontroller_cfgrCr!r!r"rts,    rtcCsdd|DS)zremove boolean false valuescSs&i|]\}}|dur|dd|qS)Fr.-)replace.0rvaluer!r!r" s  zfilter_args..)items)rmr!r!r" filter_argssrr5cCs|d}|}|stdn|tdddkr0|dr0td|jd|j d|j d |j gd d t | D|R}|rNtj|dSdS) N playbook_namezCannot parse ansible versionrdiffzAnsible version .z%doesn't support --diff flag, exiting.cSs2g|]\}}|durd|d|nd|qS)T--=r!rr!r!r" s z$run_ansible_pull..)r'r2LOGwarningrr+rr7majorminorpatchr5rrrTr-write)r5rmrvr-r!r!r"rws&    rwcCs4|dg}|std||D]}||qdS)NactionszInvalid config: %s)rrrr%)rmrrr:r!r!r"rvs   rvcCs|dgD]}|dd|d|dgq|dgD]"}|d}|d}d |gd d t|D}|j||d qdS) Nr~gitclonesourcerbr playbook_dirrzansible-playbookcSs g|] \}}d|d|qS)rrr!rr!r!r"rs z&ansible_controller..)cwd)rr%r'rr)rmr repositoryr4rrr:r!r!r"rxs  rx)/__doc__rIloggingrr(rTrdcopyrtypingr cloudinitrrrcloudinit.cloudrcloudinit.configrcloudinit.config.schemar cloudinit.distrosr r cloudinit.settingsr cloudinit.utilr r__annotations__ getLoggerrDrruABCrrKrjrGrHr}dictrtrrwrvrxr!r!r!r"s@        /A %