o o@sBdZddlmZddlmZddlmZddlZddlZddlZddlZddl m Z ddl m Z ddl m Z dd l mZdd l mZdd lmZdd lmZddlZddlZGd dde jZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZdS)a3Classes for working with component snapshots. A snapshot is basically a state of the world at a given point in time. It describes the components that exist and how they depend on each other. This module lets you do operations on snapshots like getting dependency closures, as well as diff'ing snapshots. )absolute_import)division)unicode_literalsN)config) exceptions)log) installers)schemas)files) platformsc@seZdZdZdS)Errorz(Base exception for the snapshots module.N)__name__ __module__ __qualname____doc__rrB/tmp/google-cloud-sdk/lib/googlecloudsdk/core/updater/snapshots.pyr -sr cs"eZdZdZdfdd ZZS) URLFetchErrorz%Error for problems fetching via HTTP.NFcsTd}|r |d|7}n |r|d7}n|d7}|r |d|7}tt||dS)Nz.Failed to fetch component listing from server.z Received response code [{0}].z" The repository URL was malformed.z+ Check your network settings and try again.z Please ensure that the additional component repository [{0}] is correct and still valid. To remove it, run: $ gcloud components repositories remove {0})formatsuperr__init__)selfcode malformed extra_repomsg __class__rrr5s zURLFetchError.__init__)NFNr rrrr __classcell__rrrrr2src eZdZdZfddZZS)MalformedSnapshotErrorz(Error with the contents of the snapshot.cstt|ddS)Nz0Failed to process component listing from server.)rr!rrrrrrIs zMalformedSnapshotError.__init__rrrrrr!Fr!cr )IncompatibleSchemaVersionErrorzFError for when we are unable to parse the new version of the snapshot.cstt|d||_dS)NzCThe latest version snapshot is incompatible with this installation.)rr$rschema_version)rr%rrrrQs  z'IncompatibleSchemaVersionError.__init__rrrrrr$Nr#r$c@seZdZdZedZeddZeddZ eddZ ed-d d Z ed d Z eddZ ddZddZddZddZddZd.ddZd.ddZd.dd Zd!d"Zd.d#d$Zd.d%d&Zd.d'd(Zd.d)d*Zd.d+d,ZdS)/ComponentSnapshota[Contains a state-of-the-world for existing components. A snapshot can be loaded from different sources. It can be the latest that exists on the server or it can be constructed from local install state. Either way, it describes the components that are available, how they depend on each other, and other information about them like descriptions and version information. Attributes: revision: int, The global revision number for this snapshot. If it was created from an InstallState, this will be -1 to indicate that it is potentially a composition of more than one snapshot. sdk_definition: schemas.SDKDefinition, The full definition for this component snapshot. url: str, The full URL of the file from which this snapshot was loaded. This could be a web address like http://internet.com/components.json or a local file path as a URL like file:///some/local/path/components.json. It may also be None if the data did not come from a file. components = dict from component id string to schemas.Component, All the Components in this snapshot. z^\w+://cCs$tj|r|Stj|d|S)aConvert the potentially relative value into an absolute URL. Args: url: str, The URL of the component snapshot this value was found in. value: str, The value of the field to make absolute. If it is already an absolute URL, it is returned as-is. If it is relative, it's path is assumed to be relative to the component snapshot URL. Returns: str, The absolute URL. /)r& ABSOLUTE_REsearchospathdirname)urlvaluerrr_GetAbsoluteURLos z!ComponentSnapshot._GetAbsoluteURLcCsdt| }t|}Wdn1swYd|ds"dnd|dd}t||fS)zLoads a snapshot from a local file. Args: snapshot_file: str, The path of the file to load. Returns: A ComponentSnapshot object Nzfile://r'\)r FileReaderjsonload startswithreplacer&_FromDictionary) snapshot_filefdatar-rrrFromFiles  zComponentSnapshot.FromFilecsdtjj}t|tdg}|rtd|||dd|dfdd|D}tj|S)aLoads a snapshot from a series of URLs. Args: *urls: str, The URLs to the files to load. **kwargs: command_path: the command path to include in the User-Agent header if the URL is HTTP Returns: A ComponentSnapshot object. Raises: URLFetchError: If the URL cannot be fetched. TypeError: If an unexpected keyword argument is given. command_pathz,{0} got an unexpected keyword argument '{1}'unknownrcs$g|]}tj||kd|fqS)) is_extra_repo)r& _DictFromURL).0r-r<firstrr s z.ComponentSnapshot.FromURLs..) r&FromURLsr set TypeErrorrpopgetr7)urlskwargscurrent_function_nameunexpected_argsr:rrArrDs   zComponentSnapshot.FromURLsFc Cs|r|nd}zt||}Wntjjy&tjdj|dddd}Ynw|dur0t|d|j }|tj j kr?t||dz t |jd}|WStyc}z td ||td}~ww) a{Loads a json dictionary from a URL. Args: url: str, The URL to the file to load. command_path: the command path to include in the User-Agent header if the URL is HTTP is_extra_repo: bool, True if this is not the primary repository. Returns: A ComponentSnapshot object. Raises: URLFetchError: If the URL cannot be fetched. NzCould not fetch [{url}])r-T)exc_info)r)rrzutf-8z!Failed to parse snapshot [{}]: {})r MakeRequestrequestsr HTTPErrorrdebugrr status_codecodesokr3loadscontentdecode ValueErrorr!)r-r<r>rresponserr:errrr?s(    zComponentSnapshot._DictFromURLc Cs<|}dd|D}tjdddddd|id}t|S)axLoads a snapshot from the local installation state. This creates a snapshot that may not have actually existed at any point in time. It does, however, exactly reflect the current state of your local SDK. Args: install_state: install_state.InstallState, The InstallState object to load from. Returns: A ComponentSnapshot object. cSsg|]}|qSr)ComponentDefinition)r@manifestrrrrCs z6ComponentSnapshot.FromInstallState..N)revisionr%release_notes_urlversiongcloud_rel_pathpost_processing_command components notifications)InstalledComponentsvaluesr SDKDefinitionr&) install_state installedrcsdk_definitionrrrFromInstallStatesz"ComponentSnapshot.FromInstallStatecGsd}|D]e\}}tj|}|r|jrt||j|_|jtjj kr&t |tj |}|r_|j jr&z-ComponentSnapshot.__init__..css |] }|jt|jfVqdSrv)rxrE dependenciesryrrrrz'sc3s|] }|vr|VqdSrvr)r@dep_iddepsrrrz+s css|]}|tfVqdSrvrE)r@rxrrrrz.r{) rjr^r`dictrc _ComponentSnapshot__dependenciessix iteritemsrE_ComponentSnapshot__consumersadd)rrjcompdep_ids component_idr}rr~rr"szComponentSnapshot.__init__cCs`t}t|}|r.|}||jvs||vrq||j|s q||||||s |S)a~Calculates a connected closure for the components with the given ids. Performs a breadth first search starting with the given component ids, and returns the set of components reachable via the given adjacency map. Args: ids: [str], The component ids to get the closure for. adjacencies: {str: set}, Map of component ids to the set of their adjacent component ids. component_filter: schemas.Component -> bool, A function applied to components that determines whether or not to include them in the closure. Returns: set of str, The set of component ids in the closure. )rE collectionsdequepopleftrcrextend)rids adjacenciescomponent_filterclosure to_processcurrentrrr _ClosureFor3s  zComponentSnapshot._ClosureForcCs |j|S)zGets the schemas.Component from this snapshot with the given id. Args: component_id: str, The id component to get. Returns: The corresponding schemas.Component object. rcrH)rrrrrComponentFromIdPs z!ComponentSnapshot.ComponentFromIdcstfdd|DS)zGets the schemas.Component objects for each of the given ids. Args: component_ids: iterable of str, The ids of the components to get Returns: The corresponding schemas.Component objects. c3s|] }j|VqdSrvrr@rr"rrrzdsz6ComponentSnapshot.ComponentsFromIds..rr component_idsrr"rComponentsFromIds[s z#ComponentSnapshot.ComponentsFromIdscstfddt|jDS)zGets all components in the snapshot that match the given platform. Args: platform_filter: platforms.Platform, A platform the components must match. Returns: set(str), The matching component ids. c3s$|] \}}|jr|VqdSrvplatformMatches)r@c_id componentplatform_filterrrrzps z.)rErrrc)rrrrrAllComponentIdsMatchinggs z)ComponentSnapshot.AllComponentIdsMatchingNcfdd}|||j|S)aGets all the components that are depended on by any of the given ids. Args: component_ids: list of str, The ids of the components to get the dependencies of. platform_filter: platforms.Platform, A platform that components must match to be pulled into the dependency closure. Returns: set of str, All component ids that are in the dependency closure, including the given components. c |jSrvrrurrr zBComponentSnapshot.DependencyClosureForComponents..)rrrrrrrrrDependencyClosureForComponentsss z0ComponentSnapshot.DependencyClosureForComponentscr)aGets all the components that depend on any of the given ids. Args: component_ids: list of str, The ids of the components to get the consumers of. platform_filter: platforms.Platform, A platform that components must match to be pulled into the consumer closure. Returns: set of str, All component ids that are in the consumer closure, including the given components. crrvrrrrrrrz@ComponentSnapshot.ConsumerClosureForComponents..)rrrrrrConsumerClosureForComponentss z.ComponentSnapshot.ConsumerClosureForComponentscs.fddjD}fdd}|||S)aKGets all the components that are connected to any of the given ids. Connected means in the connected graph of dependencies. This is basically the union of the dependency and consumer closure for the given ids. Args: component_ids: list of str, The ids of the components to get the connected graph of. platform_filter: platforms.Platform, A platform that components must match to be pulled into the connected graph. Returns: set of str, All component ids that are connected to the given ids, including the given components. cs"i|] }|j|j|BqSr)rrr@rr"rr sz9ComponentSnapshot.ConnectedComponents..crrvrrrrrrrz7ComponentSnapshot.ConnectedComponents..)rcr)rrrrrrrrrConnectedComponentss  z%ComponentSnapshot.ConnectedComponentscCs4dd}||g|j|}||g|j|}||@S)aGets the components strongly connected to the given component id. In other words, this functions returns the strongly connected "component" of the dependency graph for the given component id. In this context the strongly connected "component" is the set of ids whose components are both dependencies and consumers of the given component. In practice this can be used to determine the platform-specific subcomponent ID's for a given component ID, since they will always be mutually dependent. Args: component_id: str, The id of the component for which to get the strongly connected components. Returns: set of str, The ids of the components that are strongly connected to the component with the given id. cSsdS)NTrrrrrrsz?ComponentSnapshot.StronglyConnectedComponents..)rrr)rrrdependency_closureconsumer_closurerrrStronglyConnectedComponentss  z-ComponentSnapshot.StronglyConnectedComponentscstd}|}|r8|jr8|jr|jjSfddj|D}fdd|D}|D] }||jjp5d7}q-|S)aComputes the effective size of the given component. If the component does not exist or does not exist on this platform, the size is 0. If it has data, just use the reported size of its data. If there is no data, report the total size of all its direct hidden dependencies (that are valid on this platform). We don't include visible dependencies because they will show up in the list with their own size. This is a best effort estimation. It is not easily possible to accurately report size in all situations because complex dependency graphs (between hidden and visible components, as well as circular dependencies) makes it infeasible to correctly show size when only displaying visible components. The goal is mainly to not show some components as having no size at all when they are wrappers around platform specific components. Args: component_id: str, The component to get the size for. platform_filter: platforms.Platform, A platform that components must match in order to be considered for any operations. Returns: int, The effective size of the component. rcsg|]}|qSr)rr@dr"rrrCsz?ComponentSnapshot.GetEffectiveComponentSize..cs(g|]}|jr|jr|jr|qSr)rr is_hiddenr:rrrrrCs  )rrrr:sizer)rrrrrrrrrrGetEffectiveComponentSizes  z+ComponentSnapshot.GetEffectiveComponentSizecCst|||dS)aCreates a ComponentSnapshotDiff based on this snapshot and the given one. Args: latest_snapshot: ComponentSnapshot, The latest state of the world that we want to compare to. platform_filter: platforms.Platform, A platform that components must match in order to be considered for any operations. Returns: A ComponentSnapshotDiff object. r)ComponentSnapshotDiff)rlatest_snapshotrrrr CreateDiffs zComponentSnapshot.CreateDiffcs"}fdd|D}|S)Ncsg|] }t|dqSr) ComponentInforrrrrC sz:ComponentSnapshot.CreateComponentInfos..)r)rrall_componentsinfosrrrCreateComponentInfoss  z&ComponentSnapshot.CreateComponentInfosc s|j}r`fdd|dD}|s&tdddd|dDd|dvrLt|ddD] }|d vrC|dd|=q6d |ddd <||d<t|D] }|d vr_||=qVt|t j |d ddddS)aWrites this snapshot back out to a JSON file. Args: path: str, The path of the file to write to. component_id: Limit snapshot to this component. If not specified all components are written out. Raises: ValueError: for non existent component_id. csg|] }|dkr|qSrwrryrrrrCs z1ComponentSnapshot.WriteToFile..rcz'Component {} is not in this snapshot {},cSsg|]}|dqSrwrryrrrrC!sr:r)contents_checksumtyperpr0rp)rcr%r^r`T)rz: )indent sort_keys separatorsN) rj ToDictionaryrXrjoinlistkeysr WriteFileContentsr3dumps)rr+r sdk_def_dictcomponent_dictr9keyrrr WriteToFiles4   zComponentSnapshot.WriteToFilecsvt}|D]3|jvr8|jjjs8|jjjs8|jjr8|jg|d}fdd|D}|s8|q|S)aGets all the components that miss required platform-specific executables. Args: component_ids: list of str, The ids of the components to check for. platform_filter: platforms.Platform, A platform that components must match to be pulled into the dependency closure. Returns: set of str, All component ids that miss required platform-specific executables. rcs$g|]}t|dr|qS)z{}-)strr5rrrrrrCHs$zDComponentSnapshot.CheckMissingPlatformExecutable..)rErcr architecturesoperating_systemsplatform_requiredrr)rrr invalid_seedsr qualifiedrrrCheckMissingPlatformExecutable3s"      z0ComponentSnapshot.CheckMissingPlatformExecutable)Frv)r rrrrecompiler( staticmethodr/r;rDr?rkr7rrrrrrrrrrrrrrrrrrr&Ws:    ! $  .     3  %r&c@seZdZdZeejjejj Z gdZ d"ddZ ddZ dd Zd d Zd d ZddZddZddZddZddZddZddZddZddZd d!ZdS)#raProvides the ability to compare two ComponentSnapshots. This class is used to see how the current state-of-the-word compares to what we have installed. It can be for informational purposes (to list available updates) but also to determine specifically what components need to be uninstalled / installed for a specific update command. Attributes: current: ComponentSnapshot, The current snapshot state. latest: CompnentSnapshot, The new snapshot that is being compared. )z@/Library/Apple/System/Library/LaunchDaemons/com.apple.oahd.plistz(/Library/Apple/usr/share/rosetta/rosettazJ/Library/Apple/System/Library/Receipts/com.apple.pkg.RosettaUpdateAuto.bomzL/Library/Apple/System/Library/Receipts/com.apple.pkg.RosettaUpdateAuto.plistNcs____B_jr\jjB_jj_ t j_ jj O_fddj D_ j fddj Dn fddjD_ t ddj D_t ddj D_t ddj D_d S) aBCreates a new diff between two ComponentSnapshots. Args: current: The current ComponentSnapshot latest: The ComponentSnapshot representing a new state we can move to platform_filter: platforms.Platform, A platform that components must match in order to be considered for any operations. cg|] }t|dqSr ComponentDiffrrlatestrrrrC z2ComponentSnapshotDiff.__init__..csg|] }t|jdqSr)r DARWIN_X86_64r)rrrrrrCs crrrrrrrrCrcs"|] }|jtjur|jVqdSrv)stateComponentStateREMOVEDrxr@diffrrrrz  z1ComponentSnapshotDiff.__init__..csrrv)rrNEWrxrrrrrzrcsrrv)rrUPDATE_AVAILABLErxrrrrrzsN)rr'_ComponentSnapshotDiff__platform_filter_EnableFallback'_ComponentSnapshotDiff__enable_fallbackr&_ComponentSnapshotDiff__all_componentsr4_ComponentSnapshotDiff__all_darwin_x86_64_components0_ComponentSnapshotDiff__darwin_x86_64_componentsrE-_ComponentSnapshotDiff__native_all_components_ComponentSnapshotDiff__diffsr*_ComponentSnapshotDiff__removed_components&_ComponentSnapshotDiff__new_components*_ComponentSnapshotDiff__updated_components)rrrrr)rrrrrres:      zComponentSnapshotDiff.__init__cCs&|jo|jjtjjko|jjtjjkSrv)roperating_systemr OperatingSystemMACOSX architecture Architecturearmr"rrrrs z%ComponentSnapshotDiff._EnableFallbackcCst||j}|j||j}|rB|j||j}||M}t||j}||j@}|rB| }|sBt d d |||O}|rOt d d |||BS)zSees if any of the given components don't exist locally or remotely. Args: component_ids: list of str, The components that the user wants to update. Returns: set of str, The component ids that do not exist anywhere. zThe ARM versions of the components [{}] are not available yet. To download and execute the x86_64 version of the components, please install Rosetta 2 first by running the command: softwareupdate --install-rosetta., @The platform specific binary does not exist for components [{}].)rErrrrrrrr_CheckRosetta2Existsrwarningrr)rrrmissing_platformmissing_platform_x86_64native_invalid_ids arm_x86_idsrosetta2_installedrrrInvalidUpdateSeedss0    z(ComponentSnapshotDiff.InvalidUpdateSeedscCs"|jD] }tj|rdSqdS)NTF)ROSETTA2_FILESr*r+isfile)rr+rrrrs  z*ComponentSnapshotDiff._CheckRosetta2ExistscCs |dS)zGets all ComponentDiffs for this snapshot comparison. Returns: The list of all ComponentDiffs between the snapshots. N) _FilterDiffsr"rrrAllDiffss zComponentSnapshotDiff.AllDiffscC |tjS)z{Gets ComponentDiffs for components where there is an update available. Returns: The list of ComponentDiffs. )r rrr"rrrAvailableUpdates z&ComponentSnapshotDiff.AvailableUpdatescCr )zrGets ComponentDiffs for new components that can be installed. Returns: The list of ComponentDiffs. )r rrr"rrrAvailableToInstallr z(ComponentSnapshotDiff.AvailableToInstallcCr )zmGets ComponentDiffs for components that no longer exist. Returns: The list of ComponentDiffs. )r rrr"rrrRemovedr zComponentSnapshotDiff.RemovedcCr )zvGets ComponentDiffs for installed components that are up to date. Returns: The list of ComponentDiffs. )r r UP_TO_DATEr"rrrUpToDater zComponentSnapshotDiff.UpToDatecs0s|j}n fdd|jD}t|dddS)Ncsg|] }|jur|qSrrrrrrrCsz6ComponentSnapshotDiff._FilterDiffs..cS|jSrvname)rrrrrsz4ComponentSnapshotDiff._FilterDiffs..r)rsorted)rrfilteredrrrr sz"ComponentSnapshotDiff._FilterDiffscstfddDS)z@Filter out x86_64 components that are available in arm versions.c3s,|]}d|vr|ddvs|VqdS)z darwin-x86_64x86_64rN)r6)r@irrrrzsz.rrrrrFilterDuplicatesArmsz)ComponentSnapshotDiff.FilterDuplicatesArmcCs,|rl|jj||jd}||jj|t|B|jdO}|jj||jd}||jj|t|B|jdO}||O}|||}t|jj }||@}|rXt d d ||t|jj @}|j|jB|B|@S|jj||jd}||jj|t|B|jdO}|t|jj @}|j|jB|@S)aCalculate the components that need to be uninstalled. Based on this given set of components, determine what we need to remove. When an update is done, we update all components connected to the initial set. Based on this, we need to remove things that have been updated, or that no longer exist. This method works with ToInstall(). For a given update set the update process should remove anything from ToRemove() followed by installing everything in ToInstall(). It is possible (and likely) that a component will be in both of these sets (when a new version is available). Args: update_seed: list of str, The component ids that we want to update. Returns: set of str, The component ids that should be removed. rzfThe ARM versions of the following components are available, replacing installed x86_64 versions: [{}].r)rrrrrrErrrcrrrrrrr)r update_seed connectedconnected_darwin_x86_64x86_removal_candidatesinstalled_componentsx86_removal_seedremoval_candidatesrrrToRemovesT  zComponentSnapshotDiff.ToRemovecst|jj|j||j}|r|j||j}|j |}t ||@}t ||}||8}||B}dd|D}|rIt d d||jj||jd} |jj| t |B|jd} |jj||jd} | |jj| |B|jdO} |jj| |B|jd} | |jj| |B|jdO} | | t @O} || } |j| |j} | rt d d| | | 8} nD|jj||jd} |jj| t |B|jd} |jj| t |B|jd} | | t @O} |j| |j} | rt d d| | | 8} |j|jBt fdd| DS) aCalculate the components that need to be installed. Based on this given set of components, determine what we need to install. When an update is done, we update all components connected to the initial set. Based on this, we need to install things that have been updated or that are new. This method works with ToRemove(). For a given update set the update process should remove anything from ToRemove() followed by installing everything in ToInstall(). It is possible (and likely) that a component will be in both of these sets (when a new version is available). Args: update_seed: list of str, The component ids that we want to update. Returns: set of str, The component ids that should be removed. cSsg|]}d|vr|qS)darwinrrrrrrCOsz3ComponentSnapshotDiff.ToInstall..zhThe ARM versions of the following components are not available yet, using x86_64 versions instead: [{}].rrrc3s$|] }|vs |vr|VqdSrvrry differentr!rrrzs z2ComponentSnapshotDiff.ToInstall..)rrrcrrrrrrrrErrrrrrrrr)rrrmissing_platform_darwin_x86_64native_valid_seed native_seed darwin_x86_64 valid_seedplatform_seedslocal_connected all_requiredlocal_connected_darwin_x86_64remote_connecteddep_missing_platformrr&r ToInstall1s           zComponentSnapshotDiff.ToInstallcCt|j|dddS)zGets the schema.Component objects for all ids from the current snapshot. Args: component_ids: list of str, The component ids to get. Returns: A list of schema.Component objects sorted by component display name. cS|jjSrvdetails display_namerrrrrz9ComponentSnapshotDiff.DetailsForCurrent..r)rrrrrrrDetailsForCurrent z'ComponentSnapshotDiff.DetailsForCurrentcCr4)zGets the schema.Component objects for all ids from the latest snapshot. Args: component_ids: list of str, The component ids to get. Returns: A list of schema.Component objects sorted by component display name. cSr5rvr6rrrrrr9z8ComponentSnapshotDiff.DetailsForLatest..r)rrrrrrrDetailsForLatestr;z&ComponentSnapshotDiff.DetailsForLatestrv)r rrrr Platformrrrrrrrrrrr r rrrr rr$r3r:r<rrrrrNs*  1"8` rc@seZdZdZdddZeddZeddZed d Zed d Z ed dZ eddZ eddZ eddZ eddZddZdS)raaEncapsulates information to be displayed for a component. Attributes: id: str, The component id. platform: str, The operating system and architecture of the platform. name: str, The display name of the component. current_version_string: str, The version of the component. is_hidden: bool, If the component is hidden. gdu_only: bool, If the component is only available in GDU. is_configuration: bool, True if this should be displayed in the packages section of the component manager. platform_required: bool, True if a platform-specific executable is required. NcCs"||_||_|||_||_dS)a@Create a new component info container. Args: component_id: str, The id of this component. snapshot: ComponentSnapshot, The snapshot from which to create info from. platform_filter: platforms.Platform, A platform that components must match in order to be considered for any operations. N)_id _snapshotr _component_platform_filter)rrsnapshotrrrrrs   zComponentInfo.__init__cCrrv)r>r"rrrrxzComponentInfo.idcCrrv)rAr"rrrrrCzComponentInfo.platformcC |jjjSrv)r@r`version_stringr"rrrcurrent_version_string z$ComponentInfo.current_version_stringcCrDrv)r@r7r8r"rrrrrGzComponentInfo.namecCr5rv)r@rr"rrrrzComponentInfo.is_hiddencCr5rv)r@gdu_onlyr"rrrrIrHzComponentInfo.gdu_onlycCr5rv)r@is_configurationr"rrrrJrHzComponentInfo.is_configurationcCr5rv)r@rr"rrrrrHzComponentInfo.platform_requiredcCs|jj|j|jdS)Nr)r?rr>rAr"rrrrszComponentInfo.sizecCsdj|j|j|jdS)Nz!{name} ({id}) [{current_version}])rrxcurrent_version)rrrxrFr"rrr__str__s zComponentInfo.__str__rv)r rrrrpropertyrxrrFrrrIrJrrrLrrrrrs,           rc@s,eZdZdZ d ddZddZddZdS) raEncapsulates the difference for a single component between snapshots. Attributes: id: str, The component id. name: str, The display name of the component. current: schemas.Component, The current component definition. latest: schemas.Component, The latest component definition that we can move to. state: ComponentState constant, The type of difference that exists for this component between the given snapshots. NcCs||_|||_|||_|jr|jjjnd|_|jr"|jjjnd|_|jr+|jn|j}|jj |_ |j |_ |j |_ |j |_ |j|_||_||_|jrP|n|}|j||d|_dS)aiCreate a new diff. Args: component_id: str, The id of this component. current_snapshot: ComponentSnapshot, The base snapshot to compare against. latest_snapshot: ComponentSnapshot, The new snapshot. platform_filter: platforms.Platform, A platform that components must match in order to be considered for any operations. Nr)rxr_ComponentDiff__current_ComponentDiff__latestr`rErFlatest_version_stringr7r8rrrIrJr _ComputeStaterrrr)rrcurrent_snapshotrr data_provideractive_snapshotrrrrs.       zComponentDiff.__init__cCs|jdurtjS|jdurtjS|jjj|jjjkrtjS|jjj|jjjkrR|jjdur6|jjdur6tj St |jjt |jjArEtjS|jjj |jjj krRtjStj S)zReturns the component state.N) rNrrrOrr` build_numberrr:rboolrr"rrrrQ"s(  zComponentDiff._ComputeStatecCs dj|jj|j|j|j|jdS)NzA[ {status} ] {name} ({id}) [{current_version}] [{latest_version}])statusrrxrKlatest_version)rrrrxrFrPr"rrrrL=s zComponentDiff.__str__rv)r rrrrrQrLrrrrrs   rc@sLeZdZdZGdddeZedZedZedZedZ e dd Z d S) rz(An enum for the available update states.c@seZdZddZddZdS)zComponentState._ComponentStatecCs ||_dSrvr)rrrrrrJs z'ComponentState._ComponentState.__init__cCrrvrr"rrrrLMsz&ComponentState._ComponentState.__str__N)r rrrrLrrrr_ComponentStateHs rY InstalledzUpdate Available Deprecatedz Not InstalledcCstjtjtjtjgS)zbGets all the different states. Returns: list(ComponentStateTuple), All the states. )rrrrrrrrrAllUszComponentState.AllN) r rrrobjectrYrrrrrr\rrrrrEsr)r __future__rrrrr3r*rgooglecloudsdk.corerrrgooglecloudsdk.core.updaterrr googlecloudsdk.core.utilr r rOrr rr!r$r]r&rrrrrrrrs>           z^LO