Vh-zdZddlmZddlZddlZddlZddlZddlm Z m Z ddl m Z m Z mZmZmZmZmZmZmZmZmZddlmZmZmZmZddlmZmZmZm Z dd l!m"Z"dd l#m$Z$dd l%m&Z&dd l'm(Z(dd Z)ddZ*dddZ+ddZ,e(ddZ-d dZ.e(ddZ/GddZ0Gdde Z1d!dZ2 d" d#dZ3y)$zJMiscellaneous utility functions and classes specific to ansible cli tools.) annotationsN)ANSIBLE_BIN_SYMLINK_MAPSOFT_RLIMIT_NOFILE) common_environmentApplicationErrorANSIBLE_LIB_ROOTANSIBLE_TEST_ROOTANSIBLE_TEST_DATA_ROOT ANSIBLE_ROOTANSIBLE_SOURCE_ROOTANSIBLE_TEST_TOOLS_ROOTMODE_FILE_EXECUTE raw_commandverified_chmod)create_temp_dir ResultTypeintercept_pythonget_injector_path)IntegrationConfigPosixIntegrationConfigEnvironmentConfig CommonConfig) data_context)install_requirements) PythonConfig)mutexc dd|dg}t|}tjt||j||ddd}|S)z3Return a dict parsed from the given inventory file.zansible-inventory-iz--listT)capturealwaysr)ansible_environmentjsonloadsrcontroller_python)argsinventory_pathcmdenv inventorys S/home/dcms/DCMS/lib/python3.12/site-packages/ansible_test/_internal/ansible_util.pyparse_inventoryr,;sL nh ?C d #C +D$2H2H#s\`imnopqrI c|jdijdi|j|i}|jdg}tfd|D}|S)zGReturn a dict of hosts from the specified group in the given inventory._metahostvarshostsc3FK|]}|j|ifywN)get).0namer0s r+ zget_hosts..Hs!GD$ T2./Gs!)r4dict)r* group_namegroup host_namesr1r0s @r+ get_hostsr<CsU}}Wb)--j"=H MM*b )E7B'J GJG GE Lr-ct}|d}t|}|j|tjj zs |tjj z|z}|s|j }|js-tjj|std|ztd"idttd|jr|rddzndddd d d d d dddddd|dd dddt|ddd|dddddt}t!|t"rD|j$r8|j'tjj)t+dt!|t,r|j'd|j'||j.rI|j'dtjj)t0j2j4d t7j8j:r8|j't7j8j:j<!t7j8j>r|j'tA||S)#zRReturn a dictionary of environment variables to use when running Ansible commands.PATHzConfiguration not found: %s#ANSIBLE_PYTHON_MODULE_RLIMIT_NOFILEANSIBLE_FORCE_COLORz%struefalseANSIBLE_FORCE_HANDLERSANSIBLE_HOST_PATTERN_MISMATCHerrorANSIBLE_INVENTORYz /dev/nullANSIBLE_DEPRECATION_WARNINGSANSIBLE_HOST_KEY_CHECKINGANSIBLE_RETRY_FILES_ENABLEDANSIBLE_CONFIGANSIBLE_LIBRARYANSIBLE_DEVEL_WARNING PYTHONPATHPAGERz/bin/cat"ANSIBLE_WORKER_SHUTDOWN_POLL_COUNT100"ANSIBLE_WORKER_SHUTDOWN_POLL_DELAYz0.1ANSIBLE_TEST_ANSIBLE_LIB_ROOTzansible_connection_cli_stub.py)_ANSIBLE_CONNECTION_PATHz,/set/ansible_python_interpreter/in/inventory)ANSIBLE_PYTHON_INTERPRETERz debug.log) ANSIBLE_DEBUGANSIBLE_LOG_PATHANSIBLE_COLLECTIONS_PATH)!rget_ansible_bin_path startswithospathpathsepget_ansible_configexplainexistsrr8strrcolorget_ansible_python_pathr isinstancercoverageupdatejoinrrdebugrLOGSr6rcontent collectionroot is_ansibleconfigure_plugin_paths)r&rcansible_configr)r]ansible_bin_pathansibles r+r"r"LsY  C v;D+D1 ??+bggoo= >"''//1D8 002 <<~ ><~MNN,/0B,C-1ZZED6Mw &'.  &  &- #*%,&$&+40$,1%&,1'*'7+G0$)*t}} %'WW\\2C2EGg%h  $./'U  JJw zz  WW\\*//*>*> L  ~(( %1^%;%;%F%F%K%K  ~(( )$/0 Jr-ct|tsiStjj t dd|j }tjj |d}t|}tjj |dgd}td|D}|jdd d |jtfd |jDtd |jD}|S)zTReturn environment variables with paths to plugins relevant for the current command.testsupport collectionsrWplugins)actionbecomecachecallbackcliconf connectionfilterhttpapir*lookupnetconfstrategyterminalrtvarsc3*K|] }d|z|f yw)z %s_pluginsNrY)r5r6s r+r7z)configure_plugin_paths..sJd|d*D1Js doc_fragmentsmodules module_utils) doc_fragmentlibraryrc3K|]:\}}d|jztjj|f<yw)z ANSIBLE_%sN)upperr\r]rh)r5keyvalue plugin_roots r+r7z)configure_plugin_paths..s7xWaWZ\a\CIIK/k51QRxsAAc3jK|]+\}}tjj|s&||f-ywr3)r\r]isdir)r5rrs r+r7z)configure_plugin_paths..s'R URWW]]5=QU|Rs'3 3) rerr\r]rhr commandr8rgitems)r& support_pathcollection_rootr) plugin_list plugin_maprs @r+roros d- . 77<< 3VY ULggll<?O !0 C '',,|Y7KK(JkJJJ$# JJtxeoeueuewxxy RciikR RC Jr-c  tjS#t$rYnwxYwtr%tj j td}ntdd}tjDcic]/\}}tj j ||t|1ncc}}w}}}|js>|jD]+\}}tj||t|t -|t_|S)z Return a directory usable for PATH, containing only the ansible entry points. If a temporary directory is required, it will be cached for the lifetime of the process and cleaned up at exit. bin ansible-test-z-bin)prefixsuffix)rZbin_pathAttributeErrorr r\r]rhr rrr get_cli_pathr`shutilcopyrr)r&rr6r] bin_linksdstsrcs r+rZrZs  #,,,    77<< e4 #/&IXoXuXuXwx*$PTRWW\\(D1<3EExx x||%OO- 7S C%s$56 7%-! Os 04B%cttd}|jD]D\}}|j|stj j ||t|dcSt|)zReturn the absolute path to the CLI script from the given path which is relative to the `bin` directory of the original source tree layout.)z../lib/ansible/z../test/lib/ansible_test/N) r r rr[r\r]rhlen RuntimeError)r] path_rewriter destinations r+rrsk,%6L ,113A ??6 "77<< T#f+,-?@ @A t r-c:~ tjS#t$rYnwxYwtr$tj j t}nDtd}t jttj j|d|t_|S)z Return a directory usable for PYTHONPATH, containing only the ansible package. If a temporary directory is required, it will be cached for the lifetime of the process and cleaned up at exit. r)rrr) rd python_pathrr r\r]dirnamer rsymlinkrh)r&rs r+rdrds{  &222    ggoo&67 &_=  #RWW\\+y%IJ*5' s  ceZdZdZddZy)CollectionDetailzCollection detail.cd|_yr3)version)selfs r+__init__zCollectionDetail.__init__s (, r-N)returnNone)__name__ __module__ __qualname____doc__rrYr-r+rrs -r-rc$eZdZdZdfd ZxZS)CollectionDetailErrorz/An error occurred retrieving collection detail.c8t|d|z||_y)Nz&Error collecting collection detail: %s)superrreason)rr __class__s r+rzCollectionDetailError.__init__!s AFJK r-)rrbrr)rrrrr __classcell__)rs@r+rrs9r-rctjj}tjj |j |j}t|jtjj td|gdd}tj|}|jd}|r t||jd}t}|t||_|Sd|_|S)zReturn collection detail.zcollection_detail.pyTr rrErN)rrkrlr\r]rhrm directoryrrr#r$r4rrrbr)pythonrlrstdoutresultrErdetails r+get_collection_detailr&s''22J Z__j.B.BCI &++rww||4KMc'dfopz~ ABCF ZZ F JJw E #E**jj#G  F%,%8S\FN M?CFN Mr-c~tjjtd|}dd||g}|r&|j dt j |g|jr!|jdd|jzzt||jdt|}t||j||| y ) zQRun the specified playbook using the given inventory file and playbook variables. playbookszansible-playbookrz-ez-%svT)rrrN) r\r]rhr extendr#dumps verbosityappendrr%r"r)r&r'playbookr variables playbook_pathr(r)s r+ run_playbookr:sGGLL!7hOM t^] CC D$**Y/01 ~~ 5C$..012t55tD d #CT4113WMr-)r&rr'rbrdict[str, t.Any])r*rr9rbrzdict[str, dict[str, t.Any]])TN)r&rrcboolrpzt.Optional[str]rdict[str, str])r&rrr)r&rrrb)r]rbrrb)rrrrr3) r&rr'rbrrbr rrzt.Optional[dict[str, t.Any]]rr)4r __future__rr#r\rtypingt constantsrrutilrrr r r r r rrrr util_commonrrrrconfigrrrrdatarpython_requirementsr host_configsrthreadrr,r<r"rorZrrdrrrrrYr-r+rsP"      FR3l> 8--,2/3 N NNN N , N  Nr-