VhMdZddlmZddlZddlZddlZddlZddlm Z m Z m Z m Z ddl mZddlmZmZmZmZmZddlmZmZdd lmZdd lmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$dd l%m&Z&dd l'm(Z(d dl)m*Z*m+Z+d dl,m-Z-d dl.m/Z/ddl0m1Z1GddejdZ3 d*dZ4 d+dZ5 d,dZ6 d-dZ7 d.dZ8 d/dZ9 d0dZ: d0dZ; d1dZ< d2dZ= d3dZ> d4dZ? d3d Z@ d4d!ZAd5d"ZBd6d#ZCd6d$ZDd6d%ZEd6d&ZFd7d8d'ZGd9d(ZHd9d)ZIy):z+Command line parsing for test environments.) annotationsN)CONTROLLER_PYTHON_VERSIONSREMOTE_PROVIDERSSECCOMP_CHOICESSUPPORTED_PYTHON_VERSIONS)REMOTE_ARCHITECTURES)docker_completionnetwork_completionremote_completionwindows_completionfilter_completion)CompositeActionCompositeActionCompletionFinder) EnumAction) DelegatedControllerActionNetworkSshTargetActionNetworkTargetActionOriginControllerActionPosixSshTargetActionPosixTargetActionSanityPythonTargetActionUnitsPythonTargetActionWindowsSshTargetActionWindowsTargetAction) TargetMode) TerminateMode)complete_choicesregister_completer)key_value_type) get_epilog)get_ci_providercveZdZdZej Zej Zej Zy)ControllerModez/Type of provisioning to use for the controller.N) __name__ __module__ __qualname____doc__enumauto NO_DELEGATIONORIGIN DELEGATEDW/home/dcms/DCMS/lib/python3.12/site-packages/ansible_test/_internal/cli/environments.pyr%r%Ls,9DIIKM TYY[F Ir0r%c |tjk(xr|tjk(}|j ||ry|j |t ||t |||t||||}tjjD cgc]2\}}||vr)|jjD] \}} |d| 4} }}}} tjs| jd| jt|t j"|_dj'| |_ycc} }}}w)zTAdd arguments for the environments used to run ansible-test and commands it invokes.)no_environmentN) target_mode zXRemote provisioning options have been hidden since no Ansible Core CI API key was found.z )r%r,r NO_TARGETS set_defaultsadd_global_optionsadd_legacy_environment_options!add_composite_environment_optionsrdocumentation_stateitemssectionsr#supports_core_ci_authappendr"argparseRawDescriptionHelpFormatterformatter_classjoinepilog) parser completercontroller_moder4r3 action_types action_typer;headingcontentr=s r1add_environmentsrLTsH%(D(DDmXbXmXmImN ~6 K0v/"6?KH4VYYdeL9H8[8[8a8a8cNN4K!4grwChC(;(D(D(J(J(LN$GW)2gY'N'NHN   2 2 4rs OOJy)*%AAFKK)FMNs7E ctjtj|j d}|j ddtj |j ddtj |j d d tj |j d dd d t||t||y)zmAdd global options for controlling the test environment that work with both the legacy and composite options.zglobal environment argumentstitlez --containersJSONmetavarhelpz --pypi-proxy store_trueactionrSz--pypi-endpointURIz--requirementsFzinstall command requirements)rVdefaultrSN) tcastr@ArgumentParseradd_argument_group add_argumentSUPPRESSadd_global_remoteadd_global_docker)rErG global_parsers r1r8r8us FF822F4M4MTr4M4stM             + m_5m_5r0c  tjtj|j d}|j dtj g d fd }|tjk(r|jdnLt|j dd ||tjk(rtntd |j|tj k(r|jg  S|tj"k(r|j%}t|j d d |t&d |jt)j+rdntj }t|j dd |rt,n |t,|xsd |jt|j dd |rt.n |t.|xsd |j S|j0rd}d} n|tj2k(rd}d} nd}d} tj2t4tj6t8tj:t<tj>t@tjBtDi} | |} t|j |d || | |j S)z;Add composite options for controlling the test environment.zWcomposite environment arguments (mutually exclusive with "environment arguments" above)rNz --host-path)rSc*j||S)z:Register the provided composite action type and return it.)r?)rIrHs r1register_action_typez?add_composite_environment_options..register_action_typesK(r0N controllerz --controllerOPTz configuration for the controllerrRrVrS)targetsz--target-posixzconfiguration for the targetz--target-windowsz--target-networkz--target-pythonz2configuration for the target python interpreter(s)z--targetzconfiguration for the target(s))rIt.Type[CompositeAction]returnrj)#rYrZr@r[r\r]r^r%r,r7r r.rrrFrr6SHELLadd_mutually_exclusive_grouprr#r>rrmultiple_pythonsPOSIX_INTEGRATIONrWINDOWS_INTEGRATIONrNETWORK_INTEGRATIONrSANITYrUNITSr) rErFrGr4composite_parserrdgroupsuppress target_option target_helptarget_actions target_actionrHs @r1r:r:svvh55v7P7Pg8Q8ij!!   " 35L .666%%%6+88 '_XfXpXpEp(AwMN3 9     j+++%%b%1l k  (( ( ==?5-- '(<=/ .     +,BBD4(J[J[5-- -5);OPf;g;; .      5-- -5);OPf;g;; .     D 7  ' '-MNK J88 8&M8K&M;K  ( (*;  * *,?  * *,?   7   5  '{3 +88 ' 6 9      r0cZ|jd}t||t|||y)z8Add legacy options for controlling the test environment.zWenvironment arguments (mutually exclusive with "composite environment arguments" below)rNN)r\add_environments_pythonadd_environments_host)rErGr4 environments r1r9r9s5 ,2+D+Dg,E,KK5+ Dr0c |jrt}nt}|jdd|dzddj |z|jddd y ) z@Add environment arguments to control the Python version(s) used.z--pythonzX.Y)rXzpython version: %sz, rRchoicesrSz--python-interpreterPATHzpath to the python interpreterrQN) has_pythonrrr]rC)environments_parserr4python_versionss r1r|r|sg34$$,. !DIIo$> > %$$ -%r0c&|j}t|t|||tjk(rt |||t ||||tjk(r t||tjk(r t|yy)z@Add environment arguments for the given host and argument modes.N) rmadd_environment_localadd_environment_venvr%r.add_environment_remoteadd_environment_dockerrrpadd_environment_windowsrqadd_environment_network)rrGr4environments_exclusive_groups r1r}r}s =P !! -" $$% *%r0c|tjk7r|jdddddy|jddd|jdd tj tt d |jd d dtjrdntj}|jdd |xsd|jdddd|xsdy)zAdd global options for Docker.NF)docker_networkdocker_terminateprime_containersdev_systemd_debugdev_probe_cgroupsz--docker-networkNETzrun using the specified networkrQz--docker-terminateTz;terminate the container: %(choices)s (default: %(default)s)rRrXrrVrSz--prime-containersrTz)download containers without running testsrUz--dev-systemd-debugz&enable systemd debugging in containersz--dev-probe-cgroupsDIR?z.probe container cgroups, with optional log dirrRnargsconstrS) r%r.r7r]rALWAYSrr#r>r@r^rErGrvs r1r`r`s  .222!"#"     .  $$  J   8'(>>@thFWFWH   AA    II r0c |tjtjfvrtt t }ntt t d}t |jdddddtjt||jd d d |jd dtd|jddtdy)z;Add environment arguments for running in docker containers.Tcontroller_onlyz--dockerIMAGErrXzrun from a docker containerrz--docker-privilegedrTz'run docker container in privileged moderUz--docker-seccompSCz;set seccomp confinement for the test container: %(choices)srz--docker-memoryINTz memory limit for docker in bytes)rRrrSN) rrorlsortedrr r r] functoolspartialrrint)rrr4 docker_imagess r1rrs z33Z5E5EFF01B1DEF 01B1DVZ[\ '44 * 5 )=9 ;$$ 6% $$ J %$$  / %r0c |tjk7r|jdddytj rdnt j }t|jddd|xsdt|jdd |xsd |jd d tjtt|xsdy)z(Add global options for remote instances.N) remote_stageremote_endpointremote_terminatez--remote-stageSTAGEprodzremote stage to use: prod, dev)rRrXrSz--remote-endpointEPz#remote provisioning endpoint to userQz--remote-terminaterzAterminate the remote instance: %(choices)s (default: %(default)s)r) r%r.r7r#r>r@r^r r]complete_remote_stagerNEVERrrs r1r_r_s .222 !  &(>>@thFWFWHv**  99 +     >>  ##   \\ r0c|tjk(r t}nO|tjk(r1t t tt t z}n td}tjrdntj}t|jdd|xsdtjt||jddt |xsd |jd d t"|xsd y)z]Add environment arguments for running in ansible-core-ci provisioned remote virtual machines.TNz--remoteNAMEzrun from a remote instancerQz--remote-providerPRz#remote provider to use: %(choices)srz --remote-archARCHzremote arch to use: %(choices)s)rroget_remote_platform_choicesrlrsetget_windows_platform_choicesr#r>r@r^r r]rrrrr )rrr4remote_platformsrvs r1rr s j22268  (( (!#&A&C"DsKgKiGj"jk6t<&(>>@thFWFWH'44  555)+;< > $$  >> %$$$  :: %r0c PdDcgc]}|j|s|c}Scc}w)zz7complete_network_platform_collection..Ae5e)splitrrrr valuesrplatform_collectionrrrleftrrxs r1rr> << Q D Ce4EFXFZ4[4b4b4dee fF# aq||D'9;CbCbfgJ^^p_yzpqrsptp_g_AG aap_ a$B'+B' B"B'B'"B'c Z|jdd}ttdtt j D}|Dcgc]E}|j |s|jr ||jDcgc]}|d c}vsA|dzGc}}Scc}wcc}}w)zReturn a list of supported network platforms matching the given prefix, excluding connection platforms already parsed from the command line.rrc34K|]}|jywrrrs r1rz7complete_network_platform_connection..Irr)rrrrr rrplatform_connectionrs r1rrFrrc>ttt|S)zFReturn a list of supported remote platforms matching the given prefix.r)rrr res r1rrNs #$5$7T UUr0cdtdttjDS)zFReturn a list of supported Windows versions matching the given prefix.c3:K|]}d|jyw)zwindows/Nversionrrs r1rz/get_windows_platform_choices..Uso7HW__-.osrrr rr/r0r1rrSs& o>OPbPd>e>l>l>no oor0cdtdttjDS)z,Return a list of supported Windows versions.c34K|]}|jywrrrs r1rz.get_windows_version_choices..Zsbg'//brrr/r0r1rrXs& b1BCUCW1X1_1_1ab bbr0) rEargparse.ArgumentParserrFrrGr%r4rrkNone)rErrGr%) rErrFrrGr%r4rrkzlist[t.Type[CompositeAction]])rErrGr%r4r)rrr4rrkr)rrrGr%r4rrkr)rrrkr)rrrkr)rrrrrkr)rErrGr%rkr)rrrrr4rrkr)rstrrk list[str])rrrzargparse.Namespacerkr)F)rfboolrkr)rkr)Jr) __future__rr@r*rtypingrY constantsrrrrutilr completionr r r r rcli.argparsingrrcli.argparsing.actionsr cli.actionsrrrrrrrrrr cli.compatrconfigr completersrr convertersr!rDr"cir#Enumr%rLr8r:r9r|r}rrrrr`rr_rrrrrrrrrr/r0r1r s1"       TYY* #*.*$* *  *B!6 #!6#!6HW #W.W$W W # Wt E # E# E E0 4505#55 5,0 D0 $- -0 $5 #5#5 5p%-%0%% %P$ #$#$ $N!-!0!! !HL L waaV p cr0