llSrSSKJr SSKJr SSKJr SSKrSSKrSSKrSSKrSSK r SSK r SSK r SSK r SSK Jr SSK Jr SSKJr SS KJr SS KJr SS KJr SS KJr SSKr"S S\R45r"SS\5r"SS\5r"SS\5rSr"SS\5r "SS\5r!Sr"g)zManages the state of what is installed in the cloud SDK. This tracks the installed modules along with the files they created. It also provides functionality like extracting tar files into the installation and tracking when we check for updates. )absolute_import)division)unicode_literalsN)config) exceptions) console_io) installers) snapshots)encoding)filesc\rSrSrSrSrg)Error/z*Base exception for the local_state module.N)__name__ __module__ __qualname____firstlineno____doc____static_attributes__r.lib/googlecloudsdk/core/updater/local_state.pyrr/s2rrc,^\rSrSrSrU4SjrSrU=r$)InvalidSDKRootError4zGError for when the root of the Cloud SDK is invalid or cannot be found.c,>[[U] S5 g)NzThe components management action could not be performed because the installation root of the Cloud SDK could not be located. If you previously used the Cloud SDK installer, you could re-install the SDK and retry again.)superr__init__self __class__s rrInvalidSDKRootError.__init__7s t- 89rrrrrrrrr __classcell__r!s@rrr4sO99rrc,^\rSrSrSrU4SjrSrU=r$)InvalidDownloadError?z9Exception for when the SDK that was download was invalid.c,>[[U] S5 g)Nz#The Cloud SDK download was invalid.)rr'rrs rrInvalidDownloadError.__init__Bs .-/rrr#r%s@rr'r'?sA//rr'c,^\rSrSrSrU4SjrSrU=r$)PermissionsErrorGzCError for when a file operation cannot complete due to permissions.cF>[[U] SRXS95 g)zInitialize a PermissionsError. Args: message: str, The message from the underlying error. path: str, The absolute path to a file or directory that needs to be operated on, but can't because of insufficient permissions. zh{message}: [{path}] Ensure you have the permissions to access the file and that the file is not in use.messagepathN)rr,rformat)r r0r1r!s rrPermissionsError.__init__Js& D* 0  +-rrr#r%s@rr,r,GsK - -rr,c^U4SjnU$)a'Use this decorator for functions that deal with files. If an exception indicating file permissions is raised, this decorator will raise a PermissionsError instead, so that the caller only has to watch for one type of exception. Args: func: The function to decorate. Returns: A decorator. c z>T"U0UD6$![RarnURSSnUSRS5(aA[R "[ US[RRUS5S95 eSnAf[[4anUR[R:Xaw[R "[ [R"UR 5[R"[RRUR"55S95 eSnAff=f)Nrz [Errno 13]r/)shutilrargs startswithrreraiser,osr1abspathOSErrorIOErrorerrnoEACCESr Decodestrerrorfilename)r8kwargsefuncs r_TryFunc)_RaisesPermissionsError.._TryFuncfs 4 "6 "" <<  VVAYq\d a  L ) ) T!W"$''//$q'": < = W  ELL    3__RWW__QZZ%@A C D  s" D:A-B  D:BD55D:r)rFrGs` r_RaisesPermissionsErrorrIXs ( /rc\rSrSrSr\R RrSr Sr Sr Sr Sr \S5rS r\S 5r\S 5r\S 5r\S 5rSr\S5r\S5rS"Sjr\S"Sj5r\S"Sj5r\S5rSrSr\S"Sj5r \S"Sj5r!\S"Sj5r"\S#Sj5r#\S"Sj5r$\S"Sj5r%\S5r&Sr'S$S jr(S!r)g)%InstallationState}z@The main class for checking / updating local installation state.z.backupz.trashz.stagingz.snapshot.json)zlib/third_party/grpcc[R"5RnU(d [5e[ [ R RU55$)zGets the installation state for the SDK that this code is running in. Returns: InstallationState, The state for this area. Raises: InvalidSDKRootError: If this code is not running under a valid SDK. )rPathssdk_rootrrKr;r1realpath)rOs r ForCurrentInstallationState.ForCurrents;||~&&H   !! RWW--h7 88rcUR5(dg[[RR UR 55$)zGets the installation state for the backup of this state, if it exists. Returns: InstallationState, The state for this area or None if the backup does not exist. N) HasBackuprKr;r1rP$_InstallationState__backup_directoryr s rBackupInstallationState)InstallationState.BackupInstallationStates4 >>    RWW--d.E.EF GGrc[R5n[URU5R 5$! [ R "SU[R"55 g=f)aGets the version string for the given installed component. This function is to be used to get component versions for metrics reporting. If it fails in any way or if the component_id is unknown, it will return None. This prevents errors from surfacing when the version is needed strictly for reporting purposes. Args: component_id: str, The component id of the component you want the version for. Returns: str, The installed version of the component, or None if it is not installed or if an error occurs. z8Failed to get installed version for component [%s]: [%s]N) rKrQInstallationManifest_state_directory VersionStringloggingdebugsysexc_info) component_idstates rVersionForInstalledComponent.InstallationState.VersionForInstalledComponentsY"2**,e !  ,00= @2 mmN #,,.2 s 7:-A)c[RRU5(d[SR U55e[ R "U5Ul[RRUR[R5Ul [RRUR[R5Ul [RRUR[R5Ul[RR!UR5[R"-Ul[&R("URUR5Ulg)zInitializes the installation state for the given sdk install. Args: sdk_root: str, The file path of the root of the SDK installation. Raises: ValueError: If the given SDK root does not exist. z.The given Cloud SDK root does not exist: [{0}]N)r;r1isdir ValueErrorr2r rA_InstallationState__sdk_rootjoinrKSTATE_DIR_NAMEr[BACKUP_DIR_NAMErUTRASH_DIR_NAME#_InstallationState__trash_directorynormpathSTAGING_ROOT_SUFFIX$_InstallationState__sdk_staging_rootr ComponentInstaller_component_installer)r rOs rrInstallationState.__init__s 77== " " Gx( **ooh/DOGGLL):)I)IKD ggll4+@+@+<+L+LNDWW\\$*?*?*;*J*JLD "ww//@0DD ED * = = ..!0Drc[RRUR5(d![R "UR5 gg)z1Creates the state directory if it does not exist.N)r;r1rfr[ file_utilsMakeDirrVs r_CreateStateDir!InstallationState._CreateStateDirs6 77==.. / /../ 0rcUR$)zjGets the root of the SDK that this state corresponds to. Returns: str, the path to the root directory. )rhrVs rrOInstallationState.sdk_roots ??rc [RRUR5(d/$[R"UR5nUVs/sHkn[RR [RR URU55(dMQURU5(dMiUPMm nnU$s snf)zReturns the files in the state directory that have the given suffix. Args: suffix: str, The file suffix to match on. Returns: list of str, The file names that match. )r;r1rfr[listdirisfileriendswith)r suffixr fmatchings r_FilesForSuffix!InstallationState._FilesForSuffixs 77==.. / / i JJt,, -E (5a77>>"'',,t/D/Da"HIJJv&5H( O(sAC )C C cUR[R5n0nUH8nUS[[R5*n[ UR U5X$'M: U$)zGets all the components that are currently installed. Returns: A dictionary of component id string to InstallationManifest. N)rrKCOMPONENT_SNAPSHOT_FILE_SUFFIXlenrZr[)r snapshot_files manifestsrras rInstalledComponents%InstallationState.InstalledComponentssi))88:NI N.MMNNOl 4T5J5J5A!Ci rc@[RRU5$)zFGenerates a ComponentSnapshot from the currently installed components.)r ComponentSnapshotFromInstallStaterVs rSnapshotInstallationState.Snapshot s  & & 7 7 ==rNc>UR5RUUS9$)aZGenerates a ComponentSnapshotDiff from current state and the given state. Args: latest_snapshot: snapshots.ComponentSnapshot, The current state of the world to diff against. platform_filter: platforms.Platform, A platform that components must match in order to be considered for any operations. Returns: A ComponentSnapshotDiff. )platform_filter)r CreateDiff)r latest_snapshotrs rDiffCurrentState"InstallationState.DiffCurrentStates) ==? % %o6E & GGrcUR5 [R"5n[RR US5n[RR US5n[ R"USS/5upg[R"XUSS9n[R"XUS9 [R"U5n [U 5S:wa [5e[RR XYS5n [R"XR5 SSS5 [!UR5n U R#5 UR%U 5 U $!,(df  NF=f) aCreates a new staging area from a fresh download of the Cloud SDK. Args: url: str, The url to download the new SDK from. progress_callback: f(float), A function to call with the fraction of completeness. Returns: An InstallationState object for the new install. Raises: installers.URLFetchError: If the new SDK could not be downloaded. InvalidDownloadError: If the new SDK was malformed. z .downloadz.extractzcomponents.reinstallprogress_callback command_pathrrN) _ClearStagingruTemporaryDirectoryr;r1rirSplitProgressBarr DownloadTar ExtractTarr|rr'MoveDirrprKrwCopyMachinePropertiesTo) r urlrt download_dir extract_dirdownload_callbackextract_callbackdownloaded_tarr rO staging_sdks rCreateStagingFromDownload+InstallationState.CreateStagingFromDownloads    & & (AWW\\![1lGGLLJ/k  % %&7!Q @,!-- /@-/n 9IKjj%e Uq"$$k84h#:#:; ) $D$;$;>   t..0G0GH%d&=&=>M!!#t (G(GH}//A rcT[RRUR5$)zDetermines if this install has a valid backup that can be restored. Returns: bool, True if there is a backup, False otherwise. )r;r1rfrUrVs rrTInstallationState.HasBackups 77==00 11rcFUR5(a UR$g)zGets the backup directory of this installation if it exists. Returns: str, The path to the backup directory or None if it does not exist. N)rTrUrVs rBackupDirectory!InstallationState.BackupDirectorys  ~~  $ $$ rc[RRUR5(a [R "UR5 U(a U"S5 gg)zDeletes the current staging directory if it exists. Args: progress_callback: f(float), A function to call with the fraction of completeness. rN)r;r1existsrpruRmTreer rs rrInstallationState._ClearStagingsA ww~~d--..//0rc[RRUR5(a [R "UR5 U(a U"S5 gg)zDeletes the current backup if it exists. Args: progress_callback: f(float), A function to call with the fraction of completeness. rN)r;r1rfrUrurrs rrInstallationState.ClearBackupsA ww}}T,,--//0rc[RRUR5(a [R "UR5 U(a U"S5 gg)zDeletes the current trash directory if it exists. Args: progress_callback: f(float), A function to call with the fraction of completeness. rN)r;r1rfrmrurrs rrInstallationState.ClearTrashsA ww}}T++,,../rc|UR5 URU5nURRXSUS9nU$)aDownloads the given component based on the given snapshot. Args: snapshot: snapshots.ComponentSnapshot, The snapshot that describes the component to install. component_id: str, The component to install from the given snapshot. progress_callback: f(float), A function to call with the fraction of completeness. command_path: the command path to include in the User-Agent header if the URL is HTTP Returns: Optional[str], The path of the downloaded archive, or None if the component has no actual sources. Raises: installers.URLFetchError: If the component associated with the provided component ID has a URL that is not fetched correctly. r)rwComponentFromIdrrDownload)r snapshotrarr componentdownloaded_archives rrInstallationState.DownloadsJ, ((6I22;;!<# rcURRX4S9n[URU5nUR X5 g)aInstalls the archive previously downloaded from self.Download(). Args: snapshot: snapshots.ComponentSnapshot, The snapshot that describes the component to install. component_id: str, The component to install from the given snapshot. downloaded_archive: Optional[str], The path to the archive downloaded previously. progress_callback: f(float), A function to call with the fraction of completeness. rN)rrExtractrZr[ MarkInstalled)r rrarrr manifests rInstallInstallationState.InstallsC  % % - - . AE#D$9$9<HH 8+rc[URU5nUR5n[U5nURn[ 5n[ 5n[ USS9GHup[RRXj5n [RRU 5(d%[RRU 5(Ga4[R"U 5 [RR[RRU 55n U RS5(anU S-n [RRU 5(a[R"U 5 UR![RRXlS55 U (aVUR![RRXl55 [RRU 5n U (aMVOR[RR#U 5(a.UR![RRU 55 U(dGMU"X- 5 GM UHen[RR#U5(dM)[RRU5(aMO[$R&"U5 Mg [)U[SS9Hn[RR#U5(dM)[RRU5(aMO[R*"U5(aMl[R,"U5 M UR/5 g) aUninstalls the given component. Deletes all the files for this component and marks it as no longer being installed. Args: component_id: str, The id of the component to uninstall. progress_callback: f(float), A function to call with the fraction of completeness. r)startz.pyc __pycache__T)keyreverseN)rZr[InstalledPathsrrhset enumerater;r1rir}islinkremovedirnamernr~addrfrursortedr|rmdirMarkUninstalled)r rarrpaths total_pathsrootdirs_to_remove pycache_dirsnumpr1dir_pathpyc_pathds r UninstallInstallationState.Uninstalls+$D$9$9<HH  # # %Ee*K ??DUN5LE+ WW\\$ "d   !5!5 $77??277#3#3A#67 ::e  CZ( WW^^H % % IIh    277<< F G   RWW\\$9 :WW__X.(h 77==  277++D12  #+,1,4 q  "''.."3"3! NT : q  "''.."3"3BJJqMM  ; rcURHin[RRURU5n[RR U5(dMS[ R"U5 Mk g)z=Clear deprecated directories that were not removed correctly.N)DEPRECATED_DIRSr;r1rirOrfrur)r rr1s rClearDeprecatedDirs%InstallationState.ClearDeprecatedDirs"sM ! ! WW\\$-- +d t  $"rc[RRUR[R R 5n[RRUR[R R 5n[RRU5(dg[R"X#5 g)a6Copy this state's properties file to another state. This is primarily intended to be used to maintain the machine properties file during a schema-change-induced reinstall. Args: other_state: InstallationState, The installation state of the fresh Cloud SDK that needs the properties file mirrored in. N) r;r1rirOrrNCLOUDSDK_PROPERTIES_NAMErr7copyfile)r other_state my_propertiesother_propertiess rr)InstallationState.CopyMachinePropertiesTo*stGGLL v||<<>Mww||fllCCE 77>>- ( (  OOM4rc [R(a[R"S5nO^[R SS:Xa[R"S5nO0[R SS:a[R"S5nOSn[ R"UR5 [RRSS 5[RRS S 5S S /nUc[[R"5S5OUnUHn[R"XcSXS9 M SSS5 g!,(df  g=f)aRAttempts to compile all the python files into .pyc files. Args: force: boolean, passed to force option of compileall.compiledir, workers: int, can be used to explicitly set number of worker processes; otherwise we determine it automatically. Only set for testing. This does not raise exceptions if compiling a given file fails. a((httplib2/python3|typing/python3|platform/bq/third_party/yaml/lib3|third_party/google/api_core|third_party/google/auth|third_party/google/oauth2|third_party/overrides|third_party/proto|dulwich|gapic|pubsublite|pubsub/lite_subscriptions.py|logging_v2|platform/bundledpythonunix|pubsub_v1/services)rz.*a (kubernetes/utils/create_from_yaml.py|platform/google_appengine|gslib/vendored/boto/boto/iam/connection.py|gslib/vendored/boto/tests/|third_party/.*/python2/|third_party/yaml/[a-z]*.py|third_party/yaml/lib2/|third_party/antlr3/|appengine/|google/cloud/appengine_|google/cloud/bigquery_logging_v1|third_party/fancy_urllib/|platform/bq/third_party/gflags|platform/ext-runtime/nodejs/test/|platform/gsutil/third_party/apitools/ez_setup|platform/gsutil/third_party/pyparsing|platform/gsutil/third_party/crcmod_osx/crcmod/test)Nbin bootstrappingdataclilibplatformr6)rxquietforceworkers)sixPY2recompiler_ version_inforuChDirrOr;r1rimin cpu_count compileall compile_dir)r rrregex_exclusion to_compile num_workersrs rCompilePythonFiles$InstallationState.CompilePythonFiles<s ww $: ;o  !  !**T*   A ! #** CD&   $-- ( '',,uo . '',,vu %   j/6oC *7k!  % N  ) ( (s +BD:: E)__backup_directory __sdk_root__sdk_staging_root__trash_directoryrrr[)N)Nunknown)FN)*rrrrrrrNCLOUDSDK_STATE_DIRrjrkrlrorr staticmethodrQrWrcrIrrwpropertyrOrrrrrrrrTrrrrrrrrrrrrrrrKrK}sH<<22./."#3 -/ 9 9 H600400   $  >> G%%N:82      ?C%8 $,,$66p  5$KNrrKcF\rSrSrSrSrSrSrSrSr Sr S r S r S r g ) rZizDClass to encapsulate the data stored in installation manifest files.z .manifestcXlX l[RR URU[ R -5Ul[RR URU[R-5Ul g)zCreates a new InstallationManifest. Args: state_dir: str, The directory path where install state is stored. component_id: str, The component id that you want to get the manifest for. N) state_diridr;r1rirKr snapshot_filerZMANIFEST_SUFFIX manifest_file)r r ras rrInstallationManifest.__init__sgNG (GGGID +;;;=Drc [R"UR5n[U5HnUR US-5 M SSS5 UR UR URS9 g!,(df  N3=f)a^Marks this component as installed with the given snapshot and files. This saves the ComponentSnapshot and writes the installed files to a manifest so they can be removed later. Args: snapshot: snapshots.ComponentSnapshot, The snapshot that was the source of the install. files: list of str, The files that were created by the installation.  N)ra)ru FileWriterr$_NormalizeFileListwrite WriteToFiler"r!)r rr fprs rr"InstallationManifest.MarkInstalledsf   t11 2b!%(! T) 3 ++$''B 3 2s 'A55 BcURUR4H?n[RR U5(dM)[R "U5 MA g)zMarks this component as no longer being installed. This does not actually uninstall the component, but rather just removes the snapshot and manifest. N)r$r"r;r1r}r)r rs rr$InstallationManifest.MarkUninstalleds=   $"4"4 5    ! 6rcT[RRUR5$)zvLoads the local ComponentSnapshot for this component. Returns: The snapshots.ComponentSnapshot for this component. )r rFromFiler"rVs rr&InstallationManifest.ComponentSnapshots!  & & / /0B0B CCrcTUR5RUR5$)zLoads the ComponentSnapshot and get the schemas.Component this component. Returns: The schemas.Component for this component. )rrr!rVs rComponentDefinition(InstallationManifest.ComponentDefinitions"  ! ! # 3 3DGG <rLs ''   &*221-8 J   9%9/5/-u-""JJNJNZO6Od