o f @s|UdZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z m Z ddlmZddlmZddlmZdd lmZmZdd lmZdd lmZd eegd gdZeed<eeZdZ 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) lifecyclesubp)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)rstdout_ first_linematchesversionr r r! get_version+s   zAnsiblePull.get_versioncGs|g|j|\}}|SN)r$r)rargsr,r-r r r!pull4szAnsiblePull.pullcCs|stddS)Nz!command: ansible is not installed) is_installed ValueErrorrr r r! check_deps8szAnsiblePull.check_depscommandcKs0|js |j|fi|S|jj||jfi|Sr2)rrrr$rr9kwargsr r r!r$<szAnsiblePull.do_ascKstj|fd|ji|S)N update_env)rrr:r r r!rAszAnsiblePull.subpcCdSr2r r7r r r!r5DzAnsiblePull.is_installedpkg_namecCr=r2r rr?r r r!installHr>zAnsiblePull.installN)__name__ __module__ __qualname__r r"rrr*r1strr4r8listr$rabcabstractmethodr5rAr r r r!rs   rcsFeZdZdedeeffdd ZdefddZdefd d Z Z S) AnsiblePullPiprusercsht|||_|tjddg\}}|d}|jd}|r-d||g|jd<dS||jd<dS)Nz-cz('import site; print(site.getuserbase())'z/bin/PATH:) superr"rr$sys executablerrjoin)rrrJ user_baser- ansible_pathold_path __class__r r!r"Ns    zAnsiblePullPip.__init__r?cCs|sUzddl}Wnty|j|jjgYnwtjdddg}tj tj t ddr7|d|jr?|d |g|d d|g||dSdS) zishould cloud-init grow an interface for non-distro package managers? this seems reusable rN-mpiprAstdlibzEXTERNALLY-MANAGEDz--break-system-packagesz--userz --upgrade)r5rW ImportErrorrinstall_packagespip_package_namerNrOrpathexistsrP sysconfigget_pathappendrr$)rr?rWcmdr r r!rA]s,     zAnsiblePullPip.installr#cCs |tjdddg\}}d|vS)NrVrWrFr)r$rNrO)rr,r-r r r!r5xszAnsiblePullPip.is_installed) rBrCrDr rrEr"rAboolr5 __classcell__r r rTr!rIMsrIc@s(eZdZdefddZdefddZdS)AnsiblePullDistror?cCs|s |j|gdSdSr2)r5rrZr@r r r!rA~szAnsiblePullDistro.installr#cCsttdS)Nr)rbrwhichr7r r r!r5szAnsiblePullDistro.is_installedN)rBrCrDrErArbr5r r r r!rd}srdnamecfgcloudr3r#cCs|di}|d}|d}|d}|d}|d} |dd} |rkt||j} |d kr8t| |} nt| } | | | |d d} | rR| | jt<|rYt || | rbt | t | |rmt || dSdSdS) Nrrinstall_methodsetup_controllergalaxyr4 package_namerWr) rvalidate_configrrIrdrAr8r CFG_OVERRIDEansible_galaxyrun_ansible_pullransible_controller)rfrgrhr3 ansible_cfg ansible_userrirj galaxy_cfgpull_cfgrlrrrr r r!handles4            rwcCsd}|D]}t||std|d|q|dr/dD]}t||s.td|d|q|d}|rIt|d|dgsItd ||d }|d vrUtd dS) N)rirlzMissing required key 'z' from r4)zpull/urlzpull/playbook_namerj repositories run_ansiblezMissing required key from ri)rWrz Invalid install method {install})r r6rany)rg required_keyskeycontroller_cfgrAr r r!rns,    rncCsdd|DS)zremove boolean false valuescSs&i|]\}}|dur|dd|qS)Fr--)replace.0r|valuer r r! s  zfilter_args..)items)rgr r r! filter_argssrr4cCs|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&r1LOGwarningrr*rr6majorminorpatchr4rrrNr,write)r4rgrvr,r r r!rqs&    rqcCs4|dg}|std||D]}||qdS)NactionszInvalid config: %s)rrrr$)rgrrr9r r r!rps   rpcCs|dgD]}|dd|d|dgq|dgD]"}|d}|d}d |gd d t|D}|j||d qdS) Nrxgitclonesourcer\ry playbook_dirrzansible-playbookcSs g|] \}}d|d|qS)rrr rr r r!rs z&ansible_controller..)cwd)rr$r&rr)rgr repositoryr3rrr9r r r!rrs  rr).__doc__rGloggingrr'rNr^copyrtypingr cloudinitrrcloudinit.cloudrcloudinit.configrcloudinit.config.schemarcloudinit.distrosr r cloudinit.settingsr cloudinit.utilr r__annotations__ getLoggerrBrroABCrrIrdrErFrwdictrnrrqrprrr r r r!s@        .0 %