B `"-@sdZddlmZmZmZeZddlZddlZddl Z ddl m Z m Z m Z mZmZddlmZmZmZddlmZddlmZdd lmZdd lmZGd d d ZGd ddZGdddZdS)z'Access Ansible Core CI remote services.)absolute_importdivisionprint_functionN)SubprocessErrorApplicationError cmd_quotedisplayANSIBLE_TEST_DATA_ROOT)intercept_commandget_network_settings run_command) AnsibleCoreCI)ansible_environment) ShellConfig)create_payloadc@sJeZdZdZddZddZddZdd Zd d ZdddZ ddZ d S)ManageWindowsCIz@Manage access to a Windows instance provided by Ansible Core CI.cCs\||_d|jjjg|_tdddddd}x.t|D]"}|jdd |||fg7_q2Wd S) z. :type core_ci: AnsibleCoreCI z-iyesnoz /dev/null) BatchModeStrictHostKeyCheckingUserKnownHostsFileServerAliveIntervalServerAliveCountMaxz-oz%s=%sN)core_cissh_keykeyssh_argsdictsorted)selfr ssh_options ssh_optionr%O/home/dcms/DCMS/lib/python3.7/site-packages/ansible_test/_internal/manage_ci.py__init__*szManageWindowsCI.__init__cCsdS)zxUsed in delegate_remote to setup the host, no action is required for Windows. :type python_version: str Nr%)r"python_versionr%r%r&setup<szManageWindowsCI.setupc Csdd|jjjd|jjjd|jjjd|jjjdg}d|jj}t|jj}dd d d d ||d d |g}xLt ddD]>}yt |jj|d|dddSt k rt dYqzXqzWtd|jj|jj|jjfdS)z-Wait for instance to respond to ansible ping.zansible_connection=winrmzansible_host=%szansible_user=%szansible_password=%szansible_port=%sz+ansible_winrm_server_cert_validation=ignorez windows_%sansiblez-mzansible.windows.win_pingz-iz%s,z-e rxpingT)envdisable_coverageN z&Timeout waiting for %s/%s instance %s.)r connectionhostnameusernamepasswordportversionrargsjoinranger rtimesleeprplatform instance_id)r" extra_varsnamer.cmddummyr%r%r&waitAs"      zManageWindowsCI.waitcCs&|d|jjj|jjj|f|dS)z< :type remote: str :type local: str z%s@%s:%sN)scprr1r3r2)r"remotelocalr%r%r&download[szManageWindowsCI.downloadcCs&||d|jjj|jjj|fdS)z< :type local: str :type remote: str z%s@%s:%sN)rCrr1r3r2)r"rErDr%r%r&uploadbszManageWindowsCI.uploadNTcCsx|sg}|r|dt|tr4ddd|D}t|jjddg|j|ddd |jjj |jjj fg|gd S) zv :type command: str | list[str] :type options: list[str] | None :type force_pty: bool z-ttr+css|]}t|VqdS)N)r).0cr%r%r& usz&ManageWindowsCI.ssh..sshz-qz-p22z%s@%sN) append isinstancelistr8r rr7rr1r3r2)r"commandoptionsZ force_ptyr%r%r&rKis  .zManageWindowsCI.sshc Csrx\tddD]N}y*t|jjdg|jdddd||gdStk rXtdYq Xq Wtd ||fdS) z7 :type src: str :type dst: str rr0rCz-PrLz-qz-rNzFailed transfer: %s -> %s) r9r rr7rrr:r;r)r"srcdstrAr%r%r&rC~s zManageWindowsCI.scp)NT) __name__ __module__ __qualname____doc__r'r)rBrFrGrKrCr%r%r%r&r(s rc@s eZdZdZddZddZdS)ManageNetworkCIz@Manage access to a network instance provided by Ansible Core CI.cCs ||_dS)z. :type core_ci: AnsibleCoreCI N)r)r"rr%r%r&r'szManageNetworkCI.__init__c Cs6t|jj|jj|jj}d|jjjd|jjjd|jjj gdd|j D}d|jj|jj ddf}t |jj}d d d |jr|jdnd |jjfd dd|jjjdd|dd||g }xNtddD]@}yt|jj|d|dddStk rtdYqXqWtd|jj|jj|jjfdS)z-Wait for instance to respond to ansible ping.zansible_host=%szansible_port=%szansible_ssh_private_key_file=%scSsg|]\}}d||fqS)z%s=%sr%)rHrvaluer%r%r& sz(ManageNetworkCI.wait..z%s-%s.-r*z-mz %s%s_commandz-az commands=?z-uz-iz%s,z-er+rZr-T)r.r/Nr0z&Timeout waiting for %s/%s instance %s.)r rr7r<r6r1r2r5rrZinventory_varsitemsreplacerZ collectionr3r8r9r rr:r;rr=)r"settingsr>r?r.r@rAr%r%r&rBs,      zManageNetworkCI.waitN)rTrUrVrWr'rBr%r%r%r&rXsrXc@sZeZdZdZddZddZddZdd Zd d Zd d Z ddZ dddZ ddZ dS) ManagePosixCIz>Manage access to a POSIX instance provided by Ansible Core CI.cCs*||_d|jjjg|_tdddddd}x.t|D]"}|jdd |||fg7_q2W|jjd kr|jjd krd d ddg|_n,|jjdkrddddg|_nt d|jjnx|jjdkrdddddg|_nZ|jjdkrdddg|_n@|jjdks|jjdkrddddg|_n|jjdkr&g|_dS)z. :type core_ci: AnsibleCoreCI z-irrz /dev/nullrr)rrrrrz-oz%s=%sZfreebsdZawsZsuz-lrootz-cZazureZsudoz-inshz$provider %s has not been implementedZmacoszPATH=/usr/local/bin:$PATHZosxrhelcentosZbash)aixZibmiN) rrrrr r!r<ZproviderbecomeNotImplementedError)r"rr#r$r%r%r&r's0"     zManagePosixCI.__init__cCsL|}tjd|ddt|jjtr6|jjjr6|S||| |S)zStart instance and wait for it to become ready and respond to an ansible ping. :type python_version: str :rtype: str zRemote working directory: %sr) verbosity) rBr inforNrr7rraw configure upload_source)r"r(pwdr%r%r&r)s  zManagePosixCI.setupc CsxtddD]r}yN|jdddd}|jjjr2dS|d}|d s\td ||f|St k r|t d Yq Xq Wt d |jj |jj|jjfd S)z$Wait for instance to respond to SSH.rr^roT)capturerz/pwd/zGUnexpected current working directory "%s" from "pwd" command output: %sr0z&Timeout waiting for %s/%s instance %s.N)r9rKrr7explainstrip splitlines startswith Exceptionrr:r;rr<r6r=)r"rAstdoutror%r%r&rBs  zManagePosixCI.waitcCs2|tjtddd|d|jj|fdS)zMConfigure remote host for testing. :type python_version: str r)z remote.shz/tmpz/chmod +x /tmp/remote.sh && /tmp/remote.sh %s %sN)rGospathr8r rKrr<)r"r(r%r%r&rmszManagePosixCI.configurec CsjtjdddR}d}tj|tj|j}t|jj |j| |j|| d||fWdQRXdS)zUpload and extract source.zansible-source-z.tgz)prefixsuffixz/tmpzZrm -rf ~/ansible ~/ansible_collections && cd ~/ && gunzip --stdout %s | tar oxf - && rm %sN) tempfileNamedTemporaryFileryrzr8basenamer?rrr7rGrK)r"Zlocal_source_fdZremote_source_dirZremote_source_pathr%r%r&rn szManagePosixCI.upload_sourcecCs&|d|jjj|jjj|f|dS)z< :type remote: str :type local: str z%s@%s:%sN)rCrr1r3r2)r"rDrEr%r%r&rFszManagePosixCI.downloadcCs&||d|jjj|jjj|fdS)z< :type local: str :type remote: str z%s@%s:%sN)rCrr1r3r2)r"rErDr%r%r&rG%szManagePosixCI.uploadNFcCs|sg}t|tr&ddd|D}|jr4t|n|}t|jjdddg|j|dt |jj j d|jj j |jj j fg|j|g|d S) z :type command: str | list[str] :type options: list[str] | None :type capture: bool :rtype: str | None, str | None r+css|]}t|VqdS)N)r)rHrIr%r%r&rJ7sz$ManagePosixCI.ssh..rKz-ttz-qz-pz%s@%s)rp)rNrOr8rhrr rr7rstrr1r5r3r2)r"rPrQrpr%r%r&rK,s zManagePosixCI.sshc Cs|xftddD]X}y4t|jjdg|jdt|jjjdd||gdStk rbt dYq Xq Wt d||fdS) z7 :type src: str :type dst: str rr0rCz-Pz-qz-rNzFailed transfer: %s -> %s) r9r rr7rrr1r5rr:r;r)r"rRrSrAr%r%r&rCAs zManagePosixCI.scp)NF) rTrUrVrWr'r)rBrmrnrFrGrKrCr%r%r%r&rbs" rb)rW __future__rrrtype __metaclass__ryr}r:utilrrrr r Z util_commonr r r rrZ ansible_utilrconfigrpayloadrrrXrbr%r%r%r&s    g,