o i) @sUdZddlZddlZddlZddlZddlZddlZddlmZddl m Z m Z ddl 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)ListOptional) 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)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_ascKstj|fd|ji|S)N update_env)rrr;r!r!r"rAszAnsiblePull.subpcCdSr3r!r8r!r!r"r6DzAnsiblePull.is_installedpkg_namecCr>r3r!r r@r!r!r"installHr?zAnsiblePull.installN)__name__ __module__ __qualname__r r#rrr+r2strr5r9listr%rabcabstractmethodr6rBr!r!r!r"rs   rcsdeZdZdedeeffdd ZddZddZd efd d Z d e fd dZ de fddZ ZS)AnsiblePullPiprusercst|||_|dSr3)superr#radd_pip_install_site_to_pathr rrK __class__r!r"r#Ns  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"rMSs  z+AnsiblePullPip.add_pip_install_site_to_pathcCs6zddl}WdSty|j|jjgYdSw)Nr)pip ImportErrorrinstall_packagespip_package_name)r rYr!r!r"bootstrap_pip_if_requiredds  z(AnsiblePullPip.bootstrap_pip_if_requiredr@cCs||sFtjdddg}tjtjt ddr"| d|j r*| d| |t d||g||t d |d Sd S) zishould cloud-init grow an interface for non-distro package managers? this seems reusable -mrYrBstdlibzEXTERNALLY-MANAGEDz--break-system-packages--userzInstalling the %s packagezInstalled the %s packageN)r]r6rSrTrpathexistsrU sysconfigget_pathappendr_AnsiblePullPip__upgrade_pipLOGinfor%)r r@cmdr!r!r"rBjs(     zAnsiblePullPip.installric Cshtdz |g|ddWntjy,}z td|WYd}~dSd}~wwtddS)zOTry to upgrade pip then not raise an exception if the pip upgrade failsz Upgrading pipz --upgraderYzZFailed at upgrading pip. This is usually not criticalso the script will skip this step. %sNz Upgraded pip)rgrhr%rProcessExecutionErrorwarning)r rier!r!r" __upgrade_pips  zAnsiblePullPip.__upgrade_pipr$cCs4tjdddg}|jr|d||\}}d|vS)Nr^rYrGr`r)rSrTrrer%)r rir-r.r!r!r"r6s  zAnsiblePullPip.is_installed)rCrDrEr rrFr#rMr]rBrGrfboolr6 __classcell__r!r!rOr"rJMsrJcsFeZdZdedeeffdd ZdefddZdefd d Z Z S) AnsiblePullDistrorrKcst|||_dSr3)rLr#rrNrOr!r"r#s  zAnsiblePullDistro.__init__r@cCs|s |j|gdSdSr3)r6rr[rAr!r!r"rBszAnsiblePullDistro.installr$cCsttdS)Nr)rnrwhichr8r!r!r"r6szAnsiblePullDistro.is_installed) rCrDrEr rrFr#rBrnr6ror!r!rOr"rpsrpnamecfgcloudr4r$cCs|di}|d}|d}|d}|d}|d} |dd} |ryt||j} |d kr8t| |} nt| |} | | | |d d} | rS| | jt<|rZt || | rpt | t rd| g} | D] }t | t |qf|r{t|| dSdSdS) Nrrinstall_methodsetup_controllergalaxyr5 package_namerYr)rvalidate_configrrJrprBr9r CFG_OVERRIDEansible_galaxy isinstancedictrun_ansible_pullransible_controller)rrrsrtr4 ansible_cfg ansible_userrurv galaxy_cfgpull_cfgrxrrrr!r!r"handles:              rc Cs6d}|D]}t||std|d|q|d}|rst|tr&|g}n t|ts2td||D]>}t|tsBtd|t|dsNtd|t|d }t|d }t||gsetd |t||grrtd |q4|d }|rt|d|dgstd||d}|dvrtddS)N)rurxzMissing required key 'z' from r5zLInvalid value ansible.pull. Expected either dict of list of dicts but found z7Invalid value of ansible.pull. Expected dict but found urlz Missing required key 'url' from playbook_nameplaybook_namesz+Missing required key 'playbook_names' from zzKey 'ansible.pull.playbook_name' and 'ansible.pull.playbook_names' are mutually exclusive. Please use 'playbook_names' in rv repositories run_ansiblezMissing required key from ru)rYrz Invalid install method {install})r r7rr}r~rGanyall) rs required_keyskeyrp_cfg has_playbook has_playbookscontroller_cfgrBr!r!r"rzsb           rzcCsdd|DS)zremove boolean false valuescSs&i|]\}}|dur|dd|qS)Fr.-)replace.0rvaluer!r!r" s  zfilter_args..)items)rsr!r!r" filter_args srr5cCs|dd}|dd}|r|g}|}|stdn|tdddkr<|dr#s z$run_ansible_pull.. )r'r2rgrkrr+rr7majorminorpatchrrr5rSr-write)r5rsrrv pull_argsr-playbookr!r!r"rs4     rcCs4|dg}|std||D]}||qdS)NactionszInvalid config: %s)rrgrkr%)rsrrr:r!r!r"r|5s   r|cCs|dgD]}|dd|d|dgq|dgD]"}|d}|d}d |gd d t|D}|j||d qdS) Nrgitclonesourcerar playbook_dirrzansible-playbookcSs g|] \}}d|d|qS)rrr!rr!r!r"rIs z&ansible_controller..)cwd)rr%r'rr)rsr repositoryr4rrr:r!r!r"r>s  r)/__doc__rHloggingrr(rSrccopyrtypingrr cloudinitrrcloudinit.cloudrcloudinit.configrcloudinit.config.schemar cloudinit.distrosr r cloudinit.settingsr cloudinit.utilr r__annotations__ getLoggerrCrgr{ABCrrJrprFrGrr~rzrrr|rr!r!r!r"s@       .T (5 !