o s@s dZddlmZddlmZddlmZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddl m Z ddl mZdd l mZdd l mZdd l mZdd l mZdd l mZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm!Z!ddlm"Z"ddl#m$Z$ddl#m%Z&ddl#m'Z'ddl(Z(ddl)m*Z*dZ+iddddd dd!dd"dd#dd$dd%dd&dd'e+d(e+d)e+d*e+d+e+d,dd-dd.ddddd/Z,gd0Z-d1d2gZ.d3Z/d4Z0d5d6Z1d7d8Z2d9d:Z3Gd;d<dd>e4Z5Gd?d@d@e4Z6GdAdBdBe4Z7GdCdDdDe4Z8GdEdFdFe4Z9GdGdHdHe4Z:GdIdJdJe4Z;GdKdLdLe4ZGdQdRdRe4Z?dSdTZ@GdUdVdVeAZBdWdXZCdYdZZDd`d[d\ZEdad^d_ZFdS)bzFHigher level functions to support updater operations at the CLI level.)absolute_import)division)unicode_literalsN) argv_utils)config) exceptions)execution_utils)log)metrics) properties)yaml) console_attr) console_io)progress_tracker)resource_printer)universe_descriptor) installers) local_state)python_manager) release_notes) snapshots) update_check)encoding)files) platforms)mapaThe standalone App Engine SDKs are no longer distributed through the Google Cloud CLI (however, the appcfg and dev_appserver commands remain the official and supported way of using App Engine from the command line). If you want to continue using these tools, they are available for download from the official App Engine download page here: https://cloud.google.com/appengine/downloads appzapp-engine-go-linux-x86zapp-engine-go-linux-x86_64zapp-engine-go-darwin-x86zapp-engine-go-darwin-x86_64zapp-engine-go-windows-x86zapp-engine-go-windows-x86_64computednszgae-javaz gae-pythonzgae-gozgae-python-launcher-maczgae-python-launcher-winzpkg-corezpkg-javaz pkg-python)zpkg-gopreviewsql)zcompletion.bash.inczcompletion.zsh.incz path.bash.incz path.fish.incz path.zsh.inczgcfilesys.bash.inczgcfilesys.zsh.inczbundled-python3zbundled-python3-unixz{This command is running using a bundled installation of Python. If you remove it, you may have no way to run this command. z`Once started, canceling this operation may leave your SDK installation in an inconsistent state.cCs4tj}|tjjfvrtdtddSdS)NzTo remove bundled python component please re-run this command after setting CLOUDSDK_PYTHON env var to preferred system python.)rOperatingSystemCurrentMACOSXr errorsysexit) current_osr)G/tmp/google-cloud-sdk/lib/googlecloudsdk/core/updater/update_manager.py_HaltIfBundledPythonUnixns r+cCsttttt|@dkS)zFReturn true if components list contains 'bundled python' component(s).r!)lenlistsetBUNDLED_PYTHON_COMPONENTS componentsr)r)r*_ContainsBundledPythonxsr2csz|}|jdur |Sz'fdd|jDD]}|}|j|jvr/|jjr/|WSqW|SY|S)aReturn most accurate VersionString for component with id comp_id. Component versions become stale in the case where only the architecture specific component has content. In these cases show the version from the architecture specific component, not the parent component. Args: component: updater.local_state.InstallationManifest of component to get VersionString of. installed_components: map of str to InstallationManifest of installed components. Returns: str, Most accurate VersionString for comp_id. Ncsg|] }|vr|qSr)r)).0dinstalled_componentsr)r* s z%_GetVersionString..)ComponentDefinitiondata VersionString dependenciesidplatform architectures) componentr6comp_defdepdep_defr)r5r*_GetVersionString}s     rCc@eZdZdZdS)Errorz-Base exception for the update_manager module.N__name__ __module__ __qualname____doc__r)r)r)r*rErEc@rD)InvalidCWDErrorzDError for when your current working directory prevents an operation.NrFr)r)r)r*rLrKrLc@rD)UpdaterDisabledErrorz;Error for when an update is attempted but it is disallowed.NrFr)r)r)r*rMrKrMc@rD)InvalidComponentErrorzCError for when a given component id is not valid for the operation.NrFr)r)r)r*rNrKrNc@rD)GcloudNotFoundErrorz2Error for when gcloud cannot be found on the path.NrFr)r)r)r*rOrKrOc@rD) NoBackupErrorzBError for when you try to restore a backup but one does not exist.NrFr)r)r)r*rPrKrPc@rD)ReinstallationFailedErrorz,Error for when performing a reinstall fails.NrFr)r)r)r*rQrKrQc@rD)MissingRequiredComponentsErrorzCError for when components are required, but you don't install them.NrFr)r)r)r*rRrKrRcs eZdZdZfddZZS)MissingUpdateURLErrorz3Error for when the URL for the manifest is not set.cstt|ddS)NzThe update action could not be performed because the component manager is incorrectly configured. Please re-install the Google Cloud CLI and try again.)superrS__init__self __class__r)r*rUs zMissingUpdateURLError.__init__)rGrHrIrJrU __classcell__r)r)rXr*rSsrSc@rD)MismatchedFixedVersionsErrorzLError for when you have pinned a version but you ask for a different one. NrFr)r)r)r*r[r[c@rD)PostProcessingErrorz)Error for when post processing failed. NrFr)r)r)r*r]r\r]c@rD)NoRegisteredRepositoriesErrorz3Error for when there are no repositories to remove.NrFr)r)r)r*r^sr^cCstdd|DdddS)zCFilters out top level components with no installable ComponentData.cSsg|] }|jdur|qSN)r9)r3compr)r)r*r7sz(FilterMetaComponents..cS|jjSr_details display_namecr)r)r*z&FilterMetaComponents..key)sortedr0r)r)r*FilterMetaComponentss rlc@seZdZdZdZdZeejj ej j Z e ddZe dcddZe d d Ze ddd d Z  deddZddZddZdfddZdgddZddZ dhddZdcddZd d!Zdhd"d#Zdid%d&Zdid'd(Zddd)d*Zddd+d,Zdjd-d.Z d/d0Z!d1d2Z"dhd3d4Z#d5d6Z$d7d8Z% djd9d:Z&d;d<Z'd=d>Z(dkd?d@Z)  dldAdBZ*dCdDZ+dEdFZ,dmdGdHZ-dhdIdJZ.dhdKdLZ/dMdNZ0dOdPZ1dQdRZ2dSdTZ3dUdVZ4dWdXZ5dcdYdZZ6d[d\Z7dcd]d^Z8d_d`Z9dhdadbZ:dS)n UpdateManagerz;Main class for performing updates for the Google Cloud CLI.binzcomponents-v{0}.jsoncCs tjjj}|r|dSgS)zGets the currently registered repositories as a list. Returns: [str], The list of registered repos or [] if there are none. ,)r VALUEScomponent_manageradditional_repositoriesGetsplit)reposr)r)r*GetAdditionalRepositoriess z'UpdateManager.GetAdditionalRepositoriesNcCs$tj}t|dd}||||S)@Installs the given components if necessary and then restarts gcloud. Args: components: [str], The components that must be installed. msg: str, A custom message to print. command: str, the command to run, if not `gcloud` Returns: bool, True if the components were already installed. If installation must occur, this method never returns because gcloud is reinvoked after the update is done. Raises: MissingRequiredComponentsError: If the components are not installed and the user chooses not to install them. Fplatform_filterwarn)rPlatformr#rm_EnsureInstalledAndRestart)r1msgcommandr=managerr)r)r*EnsureInstalledAndRestarts  z'UpdateManager.EnsureInstalledAndRestartcCsXtjjs tjjjrt ddSt }| WdS1s%wYdS)zReturns True if updates are available, False otherwise. Returns: bool, True if updates are available, False otherwise. SDK update checks are disabled.FN) rINSTALLATION_CONFIGdisable_updaterr rprqdisable_update_checkGetBoolr debugrUpdateCheckDataUpdatesAvailable)last_update_checkr)r)r*rs   $zUpdateManager.UpdatesAvailableFcCsLtjjs tjjjrt ddSt j }t |dd}|j||ddS)pChecks to see if a new snapshot has been released periodically. This method can be called as often as you'd like. It will only actually check the server for updates if a certain amount of time has elapsed since the last check (or if force is True). If updates are available, to any installed components, it will print a notification message. Args: command_path: str, The '.' separated path of the command that is currently being run (i.e. gcloud.foo.bar). force: bool, True to force a server check for updates, False to check only if the update frequency has expired. rNFrxforce)rrrr rprqrrr rrr{r#rm_PerformUpdateCheck) command_pathrr=rr)r)r*PerformUpdateCheck#s    z UpdateManager.PerformUpdateCheckcCs||_|js tj|_|jstt|j|_|dur&tj j j }|s/tj j j }|r9|r9td|t}|sHtsH||}|sQtj j j}|sVttj|j|_||_||_tjddd|_||_||_ tj j j!}||_"||_#dS)aCreates a new UpdateManager. Args: sdk_root: str, The path to the root directory of the Cloud SDK is installation. If None, the updater will search for the install directory based on the current directory. url: str, The URL to get the latest component snapshot from. If None, the default will be used. platform_filter: platforms.Platform, A platform that components must match in order to be considered for any operations. If None, only components without OS or architecture filters will match. warn: bool, True to warn about overridden configuration like an alternate snapshot file, fixed SDK version, or additional repo. Should be set to False when using this class for background operations like checking for updates so the user only sees the warnings when they are actually dealing directly with the component manager. skip_compile_python: bool, if True the UpdateManager will not pre compile installed python files. Raises: local_state.InvalidSDKRootError: If the Cloud SDK root cannot be found. MissingUpdateURLError: If we don't know what manifest to download. Nz.You are using an overridden snapshot URL: [%s]F)replace_whitespacedrop_whitespace)$_UpdateManager__sdk_rootrPathssdk_rootrInvalidSDKRootErrorrDecoder rprqdisable_warningr snapshot_urlrsr warningGetUniverseDomainIsDefaultUniverseGetUniverseSnapshotURLoriginal_snapshot_urlrSospathrealpath_UpdateManager__base_url_UpdateManager__platform_filtertextwrap TextWrapper_UpdateManager__text_wrapper_UpdateManager__warn_EnableFallback_UpdateManager__enable_fallbackfixed_sdk_version_UpdateManager__fixed_version#_UpdateManager__skip_compile_python)rWrurlryrzskip_compile_pythonuniverse_domain fixed_versionr)r)r*rU=s<      zUpdateManager.__init__cCsd}|d|S)z:Get the snapshot URL based on the current universe domain.zJhttps://storage.googleapis.com/cloud-cli-release/release/components-2.jsonzgoogleapis.com)replace)rWraddressr)r)r*rs z$UpdateManager.GetUniverseSnapshotURLcCs&|jo|jjtjjko|jjtjjkSr_)roperating_systemrr"r$ architecture ArchitecturearmrVr)r)r*rs zUpdateManager._EnableFallbackcCs*|r|j|}||d|dS)aWrites the given message to the out stream with a new line. Args: stream: The output stream to write to. msg: str, The message to write. word_wrap: bool, True to enable nicer word wrapper, False to just print the string as is.  N)rfillwriteflush)rWstreamr} word_wrapr)r)r*__Writes  zUpdateManager.__WriteTcCsjd}z tjt}WntytdYnw|r$t|j |s&dS|r,|s,dSt dj |j d)aEnsures that the CWD is valid for the update being performed. Args: in_place: bool, Whether we're performing an in-place update vs. replacing with another install state. has_components_to_remove: bool, Whether the update operation involves removing any components. Affects whether we can perform an in-place update from inside the root dir. Raises: InvalidCWDError: If the command is run from a directory within the SDK root that would cause problems when updating. NzHCould not determine CWD, assuming detached directory not under SDK root.zYour current working directory is inside the Google Cloud CLI install root: {root}. In order to perform this update, run the command from outside of this directory.)root) rrr file_utilsGetCWDOSErrorr rIsDirAncestorOfrrLformat)rWin_placehas_components_to_removecwdr)r)r* _CheckCWDs  zUpdateManager._CheckCWDcCs6t}tj|j|j|}tj|rt |SdS)zChecks if mapping files are present and loads them for further use. Args: filename: str, The full filename (with .yaml extension) to be loaded. Returns: Loaded YAML if mapping files are present, None otherwise. N) rrrrjoinrCLOUDSDK_STATE_DIRisfiler load_path)rWfilenamepaths mapping_pathr)r)r*_GetMappingFiles   zUpdateManager._GetMappingFilec sd}dd}dd}dd}||} |r'fdd|Dfdd|D||kr9fddtD}nfd d|D}fd d|D}|rW| jd |d } |s[|sc|d j| d7}rs|djdtjjd7}r|djdd7}|S)aReturns error message containing correct command mapping. Checks the user-provided command to see if it maps to one we support for their package manager. If it does, compute error message to let the user know why their command did not work and provide them with an alternate, accurate command to run. If we do not support the given command/component combination for their package manager, provide user with instructions to change their package manager. Args: command: str, Command from user input, to be mapped against commands_mapping.yaml commands_map: dict, Contains mappings from commands_mapping.yaml components_map: dict, Contains mappings from components_mapping.yaml components: str list, Component from user input, to be mapped against component_commands.yaml Returns: str, The compiled error message. r unavailable update-allNcsg|]}|vr|qSr)r)r3r?components_mapr)r*r7 z8UpdateManager._ComputeMappingMessage..csg|] }|kr|qSr))getr)rrr)r*r7s csg|]}|kr|qSr)r)r)rr)r*r7rcs g|] }|vr|vr|qSr)r)r)not_componentsunavailable_componentsr)r*r7s csg|]}|qSr)r)rrr)r*r7 s )packagez You cannot perform this action because the Google Cloud CLI component manager is disabled for this installation. You can run the following command to achieve the same result for this installation: {correct_command} )correct_commandz The {component} component(s) is unavailable through the packaging system you are currently using. Please consider using a separate installation of the Google Cloud CLI created through the default mechanism described at: {doc_url} , )r?doc_urlz/"{component}" are not valid component name(s). )r?)r.valuesrrrrdocumentation_url) rWr~ commands_maprr1 final_message update_allmapped_componentsmapped_packagesrr))rrrrr*_ComputeMappingMessagesf       z$UpdateManager._ComputeMappingMessagecCshdjtjjd}tjjr2|st||jdd}|jdd}|r"|s&t||||||}t|dS)aChecks if updater is disabled. If so, raises UpdaterDisabledError. The updater is disabled for installations that come from other package managers like apt-get or if the current user does not have permission to create or delete files in the SDK root directory. If disabled, raises UpdaterDisabledError either with the default message, or an error message from _ComputeMappingMessage if a command was passed in. Args: components: str list, Component from user input, to be mapped against component_commands.yaml command: str, Command from user input, to be mapped against command_mapping.yaml Raises: UpdaterDisabledError: If the updater is disabled. zYou cannot perform this action because this Google Cloud CLI installation is managed by an external package manager. Please consider using a separate installation of the Google Cloud CLI created through the default mechanism described at: {doc_url} )rzcommand_mapping.yaml)rzcomponent_mapping.yamlN)rrrrrrMrr)rWr1r~default_messagerrmapping_messager)r)r*_CheckIfDisabledAndThrowError0s"  z+UpdateManager._CheckIfDisabledAndThrowErrorcCs t|jSr_)rInstallationStaterrVr)r)r*_GetInstallState]s zUpdateManager._GetInstallStatecCs|j}|r|jr|j|krtd|j|n|jr)|jr&td|j|j}|rG|d}tj |ddt j ||d<d |}tjjj}|rh|jra|dD]}td|qXd ||g}|S)aGet the snapshot URL we shoould download based on any override versions. This starts with the configured URL (or comma separated URL list) and potentially modifies it based on the version. If a version is specified, it is converted to the fixed version specific snapshot. If the SDK is set to use a fixed version, that is then used. If neither, the original URL is used. Args: version: str, The Cloud SDK version to get the snapshot for. Raises: MismatchedFixedVersionsError: If you manually specify a version and you are fixed to a different version. Returns: str, The modified snapshot URL. zYou have configured your Google Cloud CLI installation to be fixed to version [{0}] but are attempting to install components at version [{1}]. To clear your fixed version setting, run: $ gcloud config unset component_manager/fixed_sdk_versionzTYou have configured your Google Cloud CLI installation to be fixed to version [{0}].ror/z3You are using additional component repository: [%s])rrr[rrr rrtrrdirnamermVERSIONED_SNAPSHOT_FORMATrr rprqrrrs)rWversionrurlsrurepor)r)r*_GetEffectiveSnapshotURL`s>   z&UpdateManager._GetEffectiveSnapshotURLunknowncCsh|j|d}z tjj|dd|iWStjy3|r&td||j r2td|j w)N)rrorzThe component listing for Google Cloud CLI version [{0}] could not be found. Make sure this is a valid archived Google Cloud CLI version.zYou have configured your Google Cloud CLI installation to be fixed to version [{0}]. Make sure this is a valid archived Google Cloud CLI version.) rrComponentSnapshotFromURLsrt URLFetchErrorr r%rr)rWrr effective_urlr)r)r*_GetLatestSnapshots*  z UpdateManager._GetLatestSnapshotcCs.|}|j||d}|j||jd}||fS)Nrrry)rrDiffCurrentStater)rWrr install_statelatest_snapshotdiffr)r)r*_GetStateAndDiffszUpdateManager._GetStateAndDiffcCsT|}i}|}t|D]\}}|}|js|jr |s qt||||<q|S)zGet the current version for every installed component. Args: include_hidden: bool, include hidden components. Returns: {str:str}, A mapping from component id to version string. )rInstalledComponentssix iteritemsr8is_configuration is_hiddenrC)rWinclude_hidden current_stateversionsr6 component_idr? component_defr)r)r*GetCurrentVersionsInformations  z+UpdateManager.GetCurrentVersionsInformationc Cst@}|s |r7tdz|jtjd\}}|j|j t | |dWnt j y6|Ynw||WddS1sGwYdS)rzChecking for updates...rrN)rrShouldDoUpdateCheckr rrrUPDATE_MANAGER_COMMAND_PATHSetFromSnapshotlatestboolAvailableUpdatesrIncompatibleSchemaVersionErrorSetFromIncompatibleSchemaNotify)rWrrr_rr)r)r*rs       "z!UpdateManager._PerformUpdateCheckcCsh|r |\}}d}n|\}}}|sdd|D}tjjjtjjjjkr/dd|D}|||fS)aLists all of the components and their current state. This pretty prints the list of components along with whether they are up to date, require an update, etc. Args: show_hidden: bool, include hidden components. only_local_state: bool, only return component information for local state. Returns: The list of snapshots.ComponentDiffs (or snapshots.ComponentInfos if only_local_state is True) for all components that are not hidden. NcSg|]}|js|qSr))rr3rfr)r)r*r7z&UpdateManager.List..cSrr))gdu_onlyrr)r)r*r7r)_GetPrintListOnlyLocal_GetPrintListWithDiffr rpcorerrsdefault)rW show_hiddenonly_local_stateto_printcurrent_versionlatest_versionr)r)r*Lists    zUpdateManager.Listcs|}|j|jd}|r1tdd|D|j|jd}fdd|D}||tj j }| t j d|||fS)zHelper method that gets a list of locally installed components to print. Returns: List of snapshots.ComponentInfos for the List method as well as the current version string. rcss|]}|jVqdSr_r<rr)r)r* sz7UpdateManager._GetPrintListOnlyLocal..c3s|] }|jvr|VqdSr_r#r native_idsr)r*r$s+ Your current Google Cloud CLI version is: )rSnapshotCreateComponentInfosrrr. DARWIN_X86_64extendrrr_UpdateManager__Writer status)rWrrdarwin_x86_64_allto_print_x86_64r r)r%r*rs" z$UpdateManager._GetPrintListOnlyLocalc Csz |jdd\}}Wntjy'}z||gddfWYd}~Sd}~ww|js-dnd}|j||d\}}|||| }|||fS)zHelper method that computes a diff and returns a list of diffs to print. Returns: List of snapshots.ComponentDiffs for the List method as well as the current and latest version strings. zcomponents.listr Nz!The latest available version is:  latest_msg) rrr_ReinstallOnErrorr_PrintVersionsrRemovedAvailableToInstallUpToDate)rWrrer1r r!rr)r)r*rs(   z#UpdateManager._GetPrintListWithDiffcCsPtjj}|jj}|tjd||r|r|tj|||tj||fS)ahPrints the current and latest version. Args: diff: snapshots.ComponentSnapshotDiff, The snapshot diff we are working with. latest_msg: str, The message to print when displaying the latest version. If None, nothing about the latest version is printed. Returns: (str, str), The current and latest version strings. r')rrrrr,r r-)rWrr1r r!r)r)r*r32s  zUpdateManager._PrintVersionsc Csli}|D]/}z tj|j|}tj|sWqtt| }|||<Wqt y3d||<Yqw|S)zCreates the sha256 checksums of files. Args: shell_rc_files: list, A list of files to get the sha256 checksums. Returns: sha256dict, dictionary of sha256 file sums. r) rrrrexistshashlibsha256rReadBinaryFileContents hexdigestr)rWshell_rc_files sha256dictnamefpathr:r)r)r* _HashRcfilesIs"    zUpdateManager._HashRcfilescCsHddj|dg}gd}djd|d|d}tj||tjdd S) zPrints info about components we are going to install or remove. Args: components: list(schemas.Component), The components that are going to be acted on. action: str, The verb to print for this set of components. boxz*title="These components will be {action}.")action)z*details.display_name:label=Name:align=leftz0version.version_string:label=Version:align=rightz:data.size.size(zero="",min=1048576):label=Size:align=rightztable[{attributes}]({columns})ro) attributescolumns)outN)rrrPrintr r-)rWr1rCrDrEfmtr)r)r*_PrintPendingActionas z!UpdateManager._PrintPendingActionc Csi}t|D]>\}}dj||jjd} tj| tj||o"|t|dkd} ||j | j d} | ||j <Wdn1s=wYd}q|S)aZPerforms an update on a component while using a progress bar. Args: components: [schemas.Component], The components that are going to be acted on. action: str, The action that is printed for this update. action_func: func, The function to call to actually do the update. It takes a single argument which is the component id. first: bool, True if this is the first stacked ProgressBar group. last: bool, True if this is the last stacked ProgressBar group. Returns: dict, Map of component ID to result of action_func for each component. z{action}: {name})rCr?r!)labelrfirstlastprogress_callbackNF) enumeraterrcrdr ProgressBarr r-r,r< SetProgress) rWr1rC action_funcrKrL results_mapindexr?rJpbresultr)r)r*_UpdateWithProgressBarvs z$UpdateManager._UpdateWithProgressBarcsfdd}|S)Ncsjj||ddS)Ncomponents.update)rNr)DownloadrrrNrrr)r*Innersz.UpdateManager._DownloadFunction..Innerr))rWrrr\r)r[r*_DownloadFunctionszUpdateManager._DownloadFunctioncsfdd}|S)Ncsjj|||dS)NrM)InstallrrZr downloads_maprr)r*r\s z-UpdateManager._InstallFunction..Innerr))rWrrr`r\r)r_r*_InstallFunctionszUpdateManager._InstallFunctioncCsB|stdtjj}tjjjrt dd}|j ||||dS)aInstalls the given components at the version you are current on. Args: components: [str], A list of component ids to install. throw_if_unattended: bool, True to throw an exception on prompts when not running in interactive mode. restart_args: list of str or None. If given, this gcloud command should be run in event of a restart (ex. if we're using a bundled Python installation to do this update). Raises: InvalidComponentError: If any of the given component ids do not exist. Returns: bool, True if the update succeeded (or there was nothing to do, False if if was cancelled by the user. z&You must specify components to installz_Additional component repositories are currently active. Running `update` instead of `install`.N)throw_if_unattendedr restart_args) rNrrrr rprqrrrsr rUpdate)rWr1rbrcrr)r)r*r^s zUpdateManager.Installc Cs`|t}|r|j|ddn|jddz |j|dd\}}Wntjy8}z ||WYd}~Sd}~ww|} |rE||||}nt|j j }| |} | |} ||| |tj| s| s|tjdt} | j|jt|d d Wdn1swY|d St|j|rt| rttt |j!|d | rd } nd } |j"|| d\}}|j#d t| d|$t%|&| | d|$t%|'| | @d|$t%|'| | d|tjt()|jj*j+tj,j-|jj-t.j/t0|dsdS|'| }|&| }|D] }t12|j3|j-j4qt5j6tjd9|tjd|j7|d|8||d dd}|j7|d|j9| | d|j7|d|:|||dd dWdn 1slwY||;t} |j<|j|j=d}| j|jt|d d Wdn 1swY|j>|jdz t?@}|AWntBy}z td|WYd}~nd}~wwtCD|t}||kr|tjt.Ed|tjd| s|jtjd jF|d!dd"|jGr.tHItJjKd#s.|L}|rtd$Fd%M||N}|r.td&Fd'M|d S)(a4Performs an update of the given components. If no components are provided, it will attempt to update everything you have installed. Args: update_seed: list of str, A list of component ids to update. throw_if_unattended: bool, True to throw an exception on prompts when not running in interactive mode. version: str, The SDK version to update to instead of latest. restart_args: list of str or None. If given, this gcloud command should be run in event of a restart (ex. if we're using a bundled Python installation to do this update). Returns: bool, True if the update succeeded (or there was nothing to do, False if if was cancelled by the user. Raises: InvalidComponentError: If any of the given component ids do not exist. updater1r~r)r~rXrNzAll components are up to date.Tr)argsz$Installing components from version: z!You will be upgraded to version: r0rrremovedupdated installed)messagerbFrPerforming in place update... DownloadingrKrL Uninstalling Installingr)snapshotz)Failed to update universe descriptors: %sz1Start a new shell for the changes to take effect.z Update done! zvTo revert your CLI to the previously installed version, you may run: $ gcloud components update --version {current} )currentrCLOUDSDK_REINSTALL_COMPONENTSz There are other instances of Google Cloud tools on your system PATH. Please remove the following to avoid confusion or accidental invocation: {0} rz There are alternate versions of the following Google Cloud tools on your system PATH. Please double check your PATH: {0} z )OrA_SHELL_RCFILESrrrrr2_HandleInvalidUpdateSeedsr-rtr1keysToRemove ToInstall_HandleBadComponentInstallStater,r r-rrr rrrClearDeprecatedDirsrEnsureSDKWriteAccessrIsPythonBundledr2r+rBUNDLED_PYTHON_REMOVAL_WARNING_RestartIfUsingBundledPythonr3rrIrlDetailsForCurrentDetailsForLatestrPrintReleaseNotesDiffsdk_definitionrelease_notes_urlrrrPromptContinue_DONT_CANCEL_MESSAGEr Installsr<version_stringrUninterruptibleSectionrWr] Uninstallra ClearBackuprr _PostProcessrUniverseDescriptorUpdateAllDescriptors ExceptionrPromptAndInstallPythonOnMacFormatRequiredUserActionrrrGetEncodedValuerenvironFindAllOtherToolsOnPathrFindAllDuplicateToolsOnPath)rW update_seedrbrrc sha256dict1rrr7original_update_seed to_remove to_installrr1r rcomponents_to_installcomponents_to_removerfr`new_diffuniverse_descriptor_data sha256dict2 bad_commandsduplicate_commandsr)r)r*rds&                          zUpdateManager.UpdatecCsD|rdt||Bvrtj|jds |ddSdSdSdS)zDeal with bad install state of kubectl component on darwin-arm machines. Args: update_seed: list of str, A list of component ids to update. to_install: set of str, A set of component ids to install. kubectlz /bin/kubectlzkubectl-darwin-armN)rr.rrrradd)rWrrr)r)r*r|s z-UpdateManager._HandleBadComponentInstallStatecCs||}|s |Sttjdrt||Stt}||@}|D]}td|t |}|r4t|q ||8}|rv|} t} |rS|j dd\} } | |} || } g} | rb| d d | | ro| d d | td | t||S) aChecks that the update seeds are valid components. Args: diff: The ComponentSnapshotDiff. version: str, The SDK version if in install mode or None if in update mode. update_seed: [str], A list of component ids to update. Raises: InvalidComponentError: If any of the given component ids do not exist. Returns: [str], The update seeds that should be used for the install/update. rvz Component [%s] no longer exists.rXr z*The following components are unknown [{}].rzThe following components are not available for your current CLI version [{}]. Please run `gcloud components update` to update your Google Cloud CLI.r)InvalidUpdateSeedsrrrrr._IGNORED_MISSING_COMPONENTSr rrrappendrrrN)rWrrr invalid_seedsignored deprecateditemadditional_msgcompletely_invalid_seedsupdate_required_seedsr latest_diffmsgsr)r)r*rxsD         z'UpdateManager._HandleInvalidUpdateSeedsc stjtjjtjtjstSfddt D}t}t}|D]<}t j ||d}|rftj|} |rQ| tfdd|Dt| g|rf| tfdd|Dt| gq*| |S)zBHelper function to find commands matching SDK bin dir on the path.cs0g|]}tjtj|r|ds|qS).)rrrr startswithr3f)bin_dirr)r*r7s  z2UpdateManager._FindToolsOnPath..rc3s.|]}jtj|vrtj|VqdSr_rrrrrrVr)r*r$  z1UpdateManager._FindToolsOnPath..c3s.|]}jtj|vrtj|VqdSr_rrrVr)r*r$r)rrrrrrm BIN_DIR_NAMEr8r.listdirrSearchForExecutableOnPathreunion) rWr duplicatesothercommandsduplicates_in_sdk_rootrr~existing_paths this_toolr))rrWr*_FindToolsOnPaths4  zUpdateManager._FindToolsOnPathcC|j|dddS)zSearches the PATH for any other instances of Cloud SDK tools. Args: path: str, A path to use instead of the PATH environment variable. Returns: {str}, The other executable paths that are not in the SDK root directory. FTrrrrrWrr)r)r*rs z%UpdateManager.FindAllOtherToolsOnPathcCr)aSearches PATH for alternate versions of Cloud SDK tools in installation. Some gcloud components include duplicate versions of commands, so if those component locations are on a user's PATH then multiple versions of the commands may be found. Args: path: str, A path to use instead of the PATH environment variable. Returns: {str}, Alternate executable paths that are in the SDK root directory. TFrrrr)r)r*rs z)UpdateManager.FindAllDuplicateToolsOnPathc s|j|dd|s dS|}|}t|}|t|j}|r-tdjd|d|j ||j d| rC|j ||j dOtfdd t |jD}|r_td jd|dsj|tjd dS|jd d d t|ddd}t|}||d|tjt|j|rtrttt|t !t"sdSt#j$tjd|tjd|j%|d|j&d d dWdn1swY|'|tjddS)zUninstalls the given components. Args: ids: list of str, The component ids to uninstall. Raises: InvalidComponentError: If any of the given component ids are not installed or cannot be removed. removerfNzCThe following components are not currently installed [{components}]rr0rc3s&|]\}}|vr|jr|VqdSr_) is_required)r3c_idr?rr)r*r$sz'UpdateManager.Remove..zxThe following required components are included in or depend on the given components and cannot be removed [{components}]zNo components to remove. TrhcSrar_rbrer)r)r*rg-rhz&UpdateManager.Remove..ririrmrnrqrpz Uninstall done! )(rrr(r.r1ryrNrrConsumerClosureForComponentsrrr*rrr,r r-rrkComponentsFromIdsrlrIrr~rrr2r+rrrrrrrrrWrr) rWidsrrsid_set not_installedrequired_components_removedrcomponents_to_displayr)rr*Removesr           zUpdateManager.RemovecCs||}|std|jdddt|jt| }| r0|s0t t|tjdds rmcCsFt}tj|jd}ttjtj |tj|tj tj S)zCCopy the current Python to temporary directory and return its path.python) rTemporaryDirectoryrrrshutilcopytreerr&rbasename)temp_dirtemp_python_install_dirr)r)r* CopyPython]s  rcCst|tjSr_)rrr&r)rr)r)r*rgsrcCstj}|tjjur@t|rBts0tj tj t j pdd}td|tdt||tddtddSdSdS)Nrz gcloud.cmdaCannot use bundled Python installation to update Google Cloud CLI in non-interactive mode. Please run again in interactive mode. If you really want to run in non-interactive mode, please run the following command before re-running this one: FOR /F "delims=" %i in ( '""{0}"" components copy-bundled-python' ) DO ( SET CLOUDSDK_PYTHON=%i ) (Substitute `%%i` for `%i` if in a .bat script.)r!F)rgr~rblockr)rr"r#rrr CanPromptrrrrrrrr r%rr&r'rr)rrgr~r(gcloud_cmd_pathr)r)r*rks"    rTc Cs |pt}|ptdd}tj|g|Rd|i}dd|D}tj|}|dkr/d}d d d|D}t j d j ||d t d ||t jt j|r^t|dStj}i}tr|}|jtjjurd d} d d t| |}tj|fddi|dS)aSCalls command again with the same arguments as this invocation and exit. Args: command: str, the command to run (full path to Python file). If not specified, defaults to current `gcloud` installation. args: list of str or None. If given, use these arguments to the command instead of the args for this process. python: str or None, the path to the Python interpreter to use for the new command invocation (if None, uses the current Python interpreter) block: bool, whether to wait for the restarted command invocation to terminate before continuing. r!NrcSg|]}t|qSr))rEncoder3ar)r)r*r7rz"RestartCommand..z gcloud.pyrrcSrr))r SafeTextr!r)r)r*r7s z)Restarting command: $ {command} {args} )r~rgzRestarting command: %s %scSsdt|dS)N")rr)sr)r)r*QuoterzRestartCommand..Quotezcmd.exe /c "{0} & pause"shellT)r GcloudPathrrrArgsForPythonToolrrrrr r-rGrrrFrerrr rr{r#rrAsyncPopenArgsrr"rrrr) r~rgrr command_args short_commandlog_argscurrent_platform popen_argsr&r)r)r*rs<       rr)NNNT)GrJ __future__rrrr9rrrr&rgooglecloudsdk.corerrrrr r r r googlecloudsdk.core.consoler rrgooglecloudsdk.core.resourcer'googlecloudsdk.core.universe_descriptorrgooglecloudsdk.core.updaterrrrrrrgooglecloudsdk.core.utilrrrrr six.movesr_GAE_REDIRECT_MSGrrwr/rrr+r2rCrErLrMrNrOrPrQrRrSr[r]r^rlobjectrmrrrrr)r)r)r*s                                 "  %