ɪWc @sddkZddkZddklZddklZddklZddklZddklZddklZdefd YZ d efd YZ e d jo e Z e i e indS( iN(tInstallComponent(tPrepackInstalledDecider(tInstallComponentBuilder(t InstallHelper(tComponentSelectionConfig(tCommontInstallComponentCollectorcBs+eZdZdZdZdZdZdZdZd e gdZ dZ d Z d Zd Zee d Zd ZdZdZdZdZdZdZdZdZdZdZdZddZdZdZ dZ!dZ"dZ#RS(!scomponent-selections.jsonsupgrade-components.jsoncCsit|_d|_tii|io:ti ti |i|_ |i i i |_ndS(N(RthelpertNonetdecidertostpathtexiststCOMPONENT_SELECTION_CONFIGRtfromJsont loadFromJsontfullComponentSelectionConfigtconstructInstallComponentst resolveLeavestfullInstallComponents(tself((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pyt__init__s   cCsti}|iioti|ii}nR|i}|i}|o|i|i }n|ii |i ||S(N( RtcreateEmptyInstallComponentRthasInstalledComponentstfromJsonObjecttloadInstalledComponentstgetPrepackInstalledDecidertisAnyPrepackInstalledt#getInstalledComponentsByPrepackNametgetInstalledPrepackNamet*saveNeccessaryInfoOnceLegacyPrepackDecidedt toJsonObject(RtinstalledComponentsR tlegacyPrepackInstalled((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pytgetInstalledComponentss   cCs!|ipt|_n|iS(N(R R(R((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pyR#s cCsti}|io|iii|i}|iii|}|i|ti ||i }t i |i dt}xl|p`|i|i|dtd}ti|}t i dGH|i||iid}qWn|S(NsCYou can remove the installed components if they are no longer used.sSelect %s to removeis;Please confirm your selection of components to be removed: sRType 'yes' to continue, type 'no' to reselect components to remove. Your input==> (RRthasSubComponentExceptBasetcopyRt substractRtprintInstalledComponentsRtconstructFromInstallComponentsRRtprintSeperateLinetprintSelectionInstructionstFalsetpromptForSelectiont rootSelectiont!createRootInstallComponentForListtprintInstallComponentsRt promptYesOrNo(RR tcomponentsToDeletetinstalledDownsteamstinstalledcomponentSelectiont toContinuet componentList((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pytgetComponentsToDelete+s$       cCsti}|iii|}|ioti||i}t i |i dt }x|p]|i |i|dd}ti|}t i dGH|i||iid}qfWnt i |i d|S(Ns.Please select available components to install.sSelect %s to installis=Please confirm your selection of components to be installed: sSType 'yes' to continue, type 'no' to reselect components to install. Your input==> s)No other components available to install.(RRRR$R%R#RR'RRR(R)R*R+R,R-R.RR/(RR tcomponentsToInstalltavailableInstallComponentstavailableComponentSelectionR3tselectedComponents((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pytgetComponentsToInstallBs$        cCs|i}ti|}ti}ti}ti}|ido|i|idnt} |ii oNt i o@|ii } | di d} | o$td| } | | ddhst TO_INSTALLcSs|iddjS(R>R;(R?(R@((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pyRAmssrThe last installation failed with unexpected errors, however your selection have been saved, your last selections:sType 'yes' to continue with last selections, type 'no' to re-select components. Please noted: To re-select components, system will roll back the partially installed components of your last selections before installing your new selections. Your input ==> cSs|id S(R;(t startswith(R@((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pyRAscSs|id S(R;(RC(R@((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pyRAsts^Below downsteams are not licensed, System will remove the unlicensed downstremas automaticallysJType 'yes' to continue, type 'no' to skip the installation. Your input==> s9You have selected below components to remove or install: ( R"RtbuildInstallComponentsRRtgetInstallComponentByNametaddSubComponentR*RthasTempSelectedComponentsRt isPatchModetloadTempSelectedComponentsR?tfilterRRR(R$RR%tmergetprintAllUserSelectionsR/Rt!loadSelectionFromPreConfigurationR5R:thasSubComponentR.tappendBaseComponentR(RtpreConfiguredFilet allowSelecttlicensed_downstreams_pathsR tlicensedComponentsR0R6R1treselectttempSelectedComponentst todeletest toinstallstdownsteamsToDeletetdownsteamToInstalltpreConfiguredSelectiont to_installst downstreamstproceed((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pytprocessToCollectAllSelectionsYs               cCs:h}|i|d<|i|d<|ii|dS(s Save current user selection as temp, in case there is failure during the deployment, user would be allowed to rerun with the same selections again. RBR<N(RRtsaveUserSelectionAsTemp(RR0R6tsaveSelections((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pyR`scCs8|ii|i|}|ii|idS(sV Save the current installed components after the deployment done. N(R$R%RLRtsaveCurrentInstalledComponentsR(RR R0R6tcurrentInstalledComponents((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pyRbscCs#t}ti}ti}|d}|djo|ido,|i|di|ii|}n|ido2|i|di|ii|i }qn`|i|d}|i |i i|ii|}|i i|ii|}||fS(s support two modes, DELTA mode should have install and delete items defined, FULL mode can have only install items defined, the items is all the user expect the system to have. tmodetDELTARBR<( RRRthas_keyRERRR%t intersectt'removeNonLeafComponentIfNoSubcomponentsRPR$(RR[R tbuilderR0R6RdtfullCurrentComponents((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pyRNs     ,6 !!cCsti|gdS(N(RtprintHighlightMessage(RtprimaryMessage((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pyR)scCsg}d}xo|D]g}|i|} | oK|oC|i||g\} } |pti}n|i| nt|} | iod} x| djo|i| |||}|i|t t | idjo d|fS|i || }|i |||t |\} }qWx| D]}| i|q0W| io|i| qvqz|i| qqW||fS(s: prompt for selection recursively via the selection config. @param selectionItems the current selectionItems to handle, this is an array containing item names. @param componentSelectionConfig the configuration that would be used to prompt selection iN(RtgetSelectionConfigt-getAvailableComponentsAndAutoSelectComponentsRRRLt selectionst!constructPromptMessageAndGetInputttrimtstrtlentcollectSelectItemsR+tTrueRGROtappend(RtselectionItemstcomponentSelectionConfigt selectMessaget includeBacktfilterSelectionByDependencytinstallComponentstallAvailalbeComponentst selectionItemtselectionConfigtavailableComponentstautoSelectComponentstinstallComponentRot userInputt selectedItemstchildComponent((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pyR+s8   ) & c Cs8||id}g}d}xu|iD]j}|i|} | o(| io|i|n| i}n|t|d|d}|d}q'W|o6|t|d|iid}|i|nti |GHd} d} xD| djp |i | t |i| o|i i| } qW| S(s construct the prompt message with selections and get the value from user userInput. @param selectionConfig: the current item to prompt for selection. @param componentSelectionConfig: this full configuration that would be used to prompt selection. @param selectMessage the customized message to prompt for selection. s: is. s; sEnter the number of the item you want to select, and then press ENTER. For multiple items, separate them by spaces. For example: 1 3. If none of the components are needed, press ENTER directly. Your input ===> N(t displayNameRoRmtsingleSelectionRvRrRtgetBackDisplayNameRtprintShortSeperateLineRt validateInputRsRt getRawInput( RRRxRyRztmessagetsingleSelectioinIndexestit selectiontsubSelectionConfigt promptMessageR((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pyRps.   % -c Csg}t}x|idD]}|i|}|od}yt|}Wntj od|GHtSX||jo|i|q||djp |djod|GHtSt}qqW|oC|o<ddig}|D]} |t| q~dGHtStS(Nt is[ERROR] Invalid index number: is1[ERROR] You can not select other items when item s or s is selected. ( R*tsplitRqtintt ValueErrorRvRutjoinRr( RRtmaxIndexRtsingleSelectioinst otherSelectedt selectIndextindext_[1]ta((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pyRs,   7cCsg}g}xp|idD]_}|i|}|pqn||jo/|i|it|d|i|qqW|S(NRi(RRqRvRoR(RRRRtselectedIndexsR((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pyRt-s cCsF|i|ii||}|i|ii||}||fS(N(t!filterByCurerntAvailableSelectionRt+collectAutoSelectComponentsForSelectedItemst5collectAvailableComponentsExcludeAutoSelectComponents(RRxRtallAutoSelectComponentst-availableInstallComponentsExcludeAutoSelected((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pyRn:scCs/|i}|i|ii|i|S(N(RRRRgRh(Rt componentstcurrentSelectionConfigtcurrentAvailableComponents((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pyR?s cCs$|i|||i|||S(N(tresolveSingleSelectiontresolveAvaliableComponents(RR R7((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pyt*resolveAvailalbeComponentsWithDependenciesDscCsx|iD]}|ii|i}|oh|io|o g|_qin|o|i|n|i||o|i|indq q W|o|i ndS(N( tgetSubComponentsRRmR>RR=tremoveSingleSelectionComponentsRRFRRh(RR R7tinstalledComponentR((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pyRIs  2cCsvxottdt|iD]L}|i|}|ii|i}|io|ii|q"q"WdS(Ni( treversedtrangeRsRRRmR>Rtremove(RR7Rt subComponentR((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pyRWs " cCsd}x|iD]}|ii|i}|o^|ip |io4|pti}n|i |ii |n|i ||}qqW|o |i |i i|iS|SdS(N(RRRRmR>RtunAvailableComponentsRRRLt_collectAvailableComponentsRRRRgRh(RR R7t)availableComponentsForInstalledComponentsRR((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pyR^s  cCs |idS(Ns (tstrip(Rtstring((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pyRqmscCs tidGH|i|dS(s{ print the installed components @param installedComponents: installedComponents json file. s7Below Components has been installed on current system: N(RR(R.(RR ((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pyR&ps cCstid|i|dGHtid|i|dGHtid|i|ii|ii|dGHtid|i|ii|ii|i|dGHtidS(NsComponents to delete: RDsComponents to install: sComponents to keeps:After the installation, system will have below components:( RRkR.R$RRR%RLR((RR0R6R ((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pyRMys     ( 1RDcCs|io|i o dGHn|i o7|i o)|d}||ii|iiGHnx$|iD]}|i||qvWdS(Ns No Component is selecteds--|-( tisRootROtisBaseRRmR>RRR.(RRtintentR((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pyR.s   cCsNti|||i}tii|o|i|S|i||SdS(s decide all upgrade components in current upgrade folder. either by looping the folder structure or have an config file for it. N(tComponentPathBuildertgetComponentsPathPrefixtUPGRADE_COMPONENTS_FILE_NAMER R R t(collectAllUpgradeComponentsViaConfigFilet)collectAllUpgradeComponentsViaDirectories(Rt pathPrefixt upgradePathtupgradeComponentsFilePath((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pytcollectAllUpgradeComponentsscCsti|}ti|S(N(RRRRE(RRtupgradeComponents((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pyRscCs6|i|i||}|pti}n|S(N(t2collectAllUpgradeComponentsViaDirectoryRecursivelyR"RR(RRRtavailableUpgradeComponents((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pyRscCst|i}|ioHxE|iD]3}|i|||}|o|i|q)q)Wn|i o|i||| odS|SdS(N(RR>RORRRGtcheckComponentExistsR(RRRRtavailableUpgradeComponenttsubInstallComponenttavailableSubUpgradeComponent((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pyRs  %cCs*tiiti|||idS(Ns config.json(R R R RRtgetConfigLocation(RRRt component((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pyRsN($t__name__t __module__R RRR"RR5R:RRuR_R`RbRNR)R*R+RpRRtRnRRRRRRqR&RMR.RRRRR(((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pyR s>     P   "             RcBseZdZedZRS(RcCsL|o|ti}nd}|o||ti}n|titiS(NRD(R tsepRtCOMPONENTS_FOLDER(RR((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pyRs (RRRt staticmethodR(((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pyRst__main__(R R$RRRRRRtobjectRRRt collectorR.R"(((ss/home/jenkins/workspace/prepack-aio-release-4.2/prepack-aio/src/main/resources/scripts/InstallComponentCollector.pyts