BSrSSKJr SSKJr SSKJr SSKrSSKrSSKJr SSKJ r SSK J r SS KJr SS KJr SS KJr S rS r\R,"SSS/5rSr"SS\R25r"SS\5r\ R:R<4SjrSr S,Sjr!Sr"Sr#Sr$Sr%Sr&Sr'Sr(S r)S!r*S"r+S#r,S-S$jr-\ R:R<4S%jr.S&r/S'r0S(r1S)r2S*r3S+r4g).z6Common utility functions for Image Version validation.)absolute_import)division)unicode_literalsN)environments_util)image_versions_util)base)flags)util)semverz1.0.0z2.1.12UpgradeValidator upgrade_validerrorc\rSrSrSrSrg)InvalidImageVersionError,zEClass for errors raised when an invalid image version is encountered.N)__name__ __module__ __qualname____firstlineno____doc____static_attributes__r>lib/googlecloudsdk/command_lib/composer/image_versions_util.pyrr,sMrrc(\rSrSrSrSSjrSrSrg)_ImageVersionItem0zGClass used to dissect and analyze image version components and strings.Nc,SnSnSnUb-[R"XA5SnUSUlUSUlUbX lUbX0l[R"UUR5Ul[R"UUR5Ulg)Nz]^composer-(\d+(?:(?:\.\d+\.\d+(?:-[a-z]+\.\d+)?)?)?|latest)-airflow-(\d+(?:\.\d+(?:\.\d+)?)?)z^(\d+|latest)$z^(\d+|\d+\.\d+)$r)refindall composer_ver airflow_vermatchcomposer_contains_aliasairflow_contains_alias)self image_verr#r$image_version_regexcomposer_version_alias_regexairflow_version_alias_regexiv_partss r__init___ImageVersionItem.__init__3sz#4 "5/;A>h"1+d!!d&$$&88,H,0,=,=$?D "$((+F+/+;+;#=DrcNSRURUR5$)Nzcomposer-{}-airflow-{})formatr#r$)r(s rGetImageVersionString'_ImageVersionItem.GetImageVersionStringIs! # * *4+<+N>N OOr)r'r$r&r#)NNN)rrrrrr.r2rrrrrr0sO=,Prrc[R"X5nUR5nURRR nURRR n[X4XQ5$)z>List of available image version upgrades for provided env_ref.)environments_api_utilGetParentconfigsoftwareConfig imageVersion pythonVersion_BuildUpgradeCandidateList)env_ref release_track env_detailsproj_location_refcur_image_version_idcur_python_versions rListImageVersionUpgradesrCMsd%))'A+nn&$++::GG"))88FF #$5$6 GGrc[US9n[U5nU(d)[[UR5S::d [ S5e[ UU5$)zEChecks if image version candidate is a valid upgrade for environment.r)r+This environment does not support upgrades.)rIsVersionComposer3CompatibleCompareVersionsMIN_UPGRADEABLE_COMPOSER_VERr#r _ValidateCandidateImageVersionId)cur_image_version_strimage_version_id cur_image_ver is_composer3s rIsValidImageVersionUpgraderOXsc $%'-., 2M4N4NO   #5 77 ) rcU=(a [U5nU(a[U5ROSn[UUS9R5$)z?rcjUSL=(a) URS5=(d URS5$)zEChecks if string composer-X.Y.Z-airflow-A.B.C is Composer v1 version.Nz composer-1.z composer-1- startswith image_versions rIsImageVersionStringComposerV1rZ5 d " }-1  ! !- 0rcjUSL=(a) URS5=(d URS5$)zEChecks if string composer-X.Y.Z-airflow-A.B.C is Composer v2 version.Nz composer-2.z composer-2-rVrXs rIsImageVersionStringComposerV2r]r[rcjUSL=(a) URS5=(d URS5$)zEChecks if string composer-X.Y.Z-airflow-A.B.C is Composer v3 version.Nz composer-3.z composer-3-rVrXs rIsImageVersionStringComposerV3r_r[rc:USL=(d URS5$)Nzcomposer-latestrVrXs rIsDefaultImageVersionras $  M-":":;L"MMrcSnU(aURS5$U(aURS5$URS5$)z>= >> >>A  6 77rcX:XagX:agg)zCompares versions.rr rv1v2s r_CompareVersionsrksX  w rcB[U5[U5p[X5$)zCompares loose version strings. Args: v1: first loose version string v2: second loose version string Returns: Value == 1 when v1 is greater; Value == -1 when v2 is greater; otherwise 0. )_VersionStrToLooseVersionrkrhs rCompareLooseVersionsrns! %R (*CB*Gb " !!rcB[U5[U5p[X5$)zCompares semantic version strings. Args: v1: first semantic version string v2: second semantic version string Returns: Value == 1 when v1 is greater; Value == -1 when v2 is greater; otherwise 0. )_VersionStrToSemanticVersionrkrhs rrHrHs! ( +-I"-Mb " !!rcU(a^[U5nU(aLUR(a;URnUS:XagUS:Xa[n[U[R SS5$g)zChecks if given `image_version` is compatible with Composer 3. Args: image_version: image version str that includes Composer version. Returns: True if Composer version is greater than or equal to 3.0.0 or its prerelease variant, otherwise False. 3TrQNF)rr##COMPOSER_LATEST_VERSION_PLACEHOLDERIsVersionInRanger MIN_COMPOSER3_VERSIONrY version_itemcomposer_versions rrGrGsc$]3L 11%22 S  X %>  E77  rcU(aK[U5nU(a9UR(a(URn[U[RSS5$g)zChecks if given `version` is compatible with Composer Airflow Commands API. Args: image_version: image version str that includes Composer version. Returns: True if Composer version is compatible with Aiflow Commands API, otherwise False. NTF)rr#rtr $MIN_COMPOSER_RUN_AIRFLOW_CLI_VERSIONrvs r%IsVersionAirflowCommandsApiCompatibler{sL$]3L 11%22    4 4    rcbU(a[U5n[U5(agU(aUR(atUR(acURnURnUS:Xa[n[ U[ RSS5=(a [ U[ RSS5$g)aChecks if given `version` is compatible with triggerer . Args: image_version: image version str that includes airflow version. Returns: True if given airflow version is compatible with Triggerer(>=2.2.x) and Composer version is >=2.0.31 otherwise False TrQNF) rrGr$r#rsrtr MIN_TRIGGERER_COMPOSER_VERSIONMIN_TRIGGERER_AIRFLOW_VERSION)rYrwrcrxs rIsVersionTriggererCompatiblers$]3L#M22  00\5N5N$00o%22 X %>  E@@$  5>>d  rcU(a[O[nUSL=(d U"X5S:*=(a USL=(d U"X5S:$)azChecks if given `version` is in range of (`range_from`, `range_to`). Args: version: version to check range_from: left boundary of range (inclusive), if None - no boundary range_to: right boundary of range (exclusive), if None - no boundary loose: if true use LooseVersion to compare, use SemVer otherwise Returns: True if given version is in range, otherwise False. Nr)rnrH)version range_fromrange_toloose compare_fns rrtrt"sH(-#/*   EJ!@A!E B t  @z'image_version_serviceimage_version_itemavailable_upgradesrs rr<r<3s 1DD()9:##344  &(:(G(G   )--l; ) 22    >> >!!'*<  #5 77rc0US:Xa[$[U5$)NrQ)COMPOSER_LATEST_MAJOR_VERSIONint)composer_ver_aliass r_GetComposerMajorVersionrUs8# ((   rcUR(a[URS5nO*[R"UR5R nUR(a[URS5nO*[R"UR5R n[ X#:H=(d US:H=(a US:HS5$)zDValidates whether Composer major only version upgrade is compatible.rrN)r&rr SemVerr#majorr ) parsed_curr parsed_candmajor_version_currmajor_version_cands r,_IsComposerMajorOnlyVersionUpgradeCompatibler[s((1++A. {'?'?@FF((1++A. {'?'?@FF .?  ! =&8A&=  rc[SS5nX:XaSRU5n[SU5n[US9n[US9nUR=(d URnU(a [ XE5nO2UR (a![ URURS5nUR (a2UR(d![ URURS5nU$)aDetermines if candidate version is a valid upgrade from current version. Args: current_image_version_id: current image version candidate_image_version_id: image version requested for upgrade Returns: UpgradeValidator namedtuple containing True and None error message if given version upgrade between given versions is valid, otherwise False and error message with problems description. TNzqExisting and requested image versions are equal ({}). Select image version newer than current to perform upgrade.FrEComposerAirflow) r r1rr&rr _IsVersionUpgradeCompatibler#r'r$)current_image_version_idcandidate_image_version_idupgrade_validator error_messagerr$has_alias_or_major_only_composer_vers rrJrJss'tT2; !'(@!A) >!,DE+!,FG+))-  , ,' *D%%3  +":":J $$[-O-O3K4K4K4?4K4K4=? rc.[R"U5$)z)Parses version_str into semantic version.)r r version_strs rrprps { ##rc.[R"U5$)z&Parses version_str into loose version.)r LooseVersionrs rrmrms   [ ))rcxUS;de[U5n[U5nX4:aSRUUUUUS9n[SU5$URS:wdURS:waMURUR:wa3SRUURUR5n[SU5$[SS 5$) a6Validates whether version candidate is greater than or equal to current. Applicable both for Airflow and Composer version upgrades. Composer supports both Airflow and self MINOR and PATCH-level upgrades. Args: cur_version: current 'a.b.c' version candidate_version: candidate 'x.y.z' version image_version_part: part of image to be validated. Must be either 'Airflow' or 'Composer' Returns: UpgradeValidator namedtuple containing boolean value whether selected image version component is valid for upgrade and eventual error message if it is not. )rrzUpgrade cannot decrease {composer_or_airflow1}'s version. Current {composer_or_airflow2} version: {cur_version}, requested {composer_or_airflow3} version: {req_version}.)composer_or_airflow1composer_or_airflow2 cur_versioncomposer_or_airflow3 req_versionFrrzwUpgrades between different {}'s major versions are not supported. Current major version {}, requested major version {}.TN)rpr1r r)rcandidate_versionimage_version_partcurr_semantic_versioncand_semantic_versionrs rrrs$ 6 66 66{C67HI2&(.v.@.@%0.@%6 (.(8 E= 11""a'+@+F+F!+K  % %)>)D)D D  f####  E= 11 $ %%r)N)F)5r __future__rrr collectionsr!googlecloudsdk.api_lib.composerrr5rrgooglecloudsdk.callioper#googlecloudsdk.command_lib.composerr r command_utilgooglecloudsdk.core.utilr rIrs namedtupler rErrorrobjectr ReleaseTrackGArCrOrTrZr]r_rarerkrnrHrGr{rrtr<rrrJrprmrrrrrs=&' VY(5D+ ''/#))*<+:G*DF!"N|11NPP:594E4E4H4HG.?"N8$ " "20:C(.2->->-A-AD! 00f$ * 2&r