8dZddlZddlZddlZddlmZdZdZdZdZ dZ d Z d Z d Z d d Zd ZdZdZdZdZdZd!dZdZdZdZdZdZdZdZdZdZdZdZ dZ!y)"z-Helper utilities for fleet packages commands.N)yamlv1alphaatable(info.rolloutStrategyInfo.rollingStrategyInfo.clusters.membership.basename():label=CLUSTER, info.rolloutStrategyInfo.rollingStrategyInfo.clusters.current.version:label=CURRENT_VERSION, info.rolloutStrategyInfo.rollingStrategyInfo.clusters.current.syncState:label=SYNC_STATE, info.rolloutStrategyInfo.rollingStrategyInfo.clusters.desired.version:label=DESIRED_VERSION, info.rolloutStrategyInfo.rollingStrategyInfo.clusters.startTime:label=START_TIME, info.rolloutStrategyInfo.rollingStrategyInfo.clusters.endTime:label=END_TIME, info.rolloutStrategyInfo.rollingStrategyInfo.clusters.state:label=STATE, trim_message():label=MESSAGE)atable(info.rolloutStrategyInfo.allAtOnceStrategyInfo.clusters.membership.basename():label=CLUSTER, info.rolloutStrategyInfo.allAtOnceStrategyInfo.clusters.current.version:label=CURRENT_VERSION, info.rolloutStrategyInfo.allAtOnceStrategyInfo.clusters.current.syncState:label=SYNC_STATE, info.rolloutStrategyInfo.allAtOnceStrategyInfo.clusters.desired.version:label=DESIRED_VERSION, info.rolloutStrategyInfo.allAtOnceStrategyInfo.clusters.startTime:label=START_TIME, info.rolloutStrategyInfo.allAtOnceStrategyInfo.clusters.endTime:label=END_TIME, info.rolloutStrategyInfo.allAtOnceStrategyInfo.clusters.state:label=STATE, trim_message():label=MESSAGE)atable(info.rolloutStrategyInfo.rollingStrategyInfo.clusters.membership.basename():label=CLUSTER, info.rolloutStrategyInfo.rollingStrategyInfo.clusters.current.version:label=CURRENT_VERSION, info.rolloutStrategyInfo.rollingStrategyInfo.clusters.current.syncState:label=SYNC_STATE, info.rolloutStrategyInfo.rollingStrategyInfo.clusters.desired.version:label=DESIRED_VERSION, info.rolloutStrategyInfo.rollingStrategyInfo.clusters.startTime:label=START_TIME, info.rolloutStrategyInfo.rollingStrategyInfo.clusters.endTime:label=END_TIME, info.rolloutStrategyInfo.rollingStrategyInfo.clusters.state:label=STATE, all_messages():label=MESSAGES)atable(info.rolloutStrategyInfo.allAtOnceStrategyInfo.clusters.membership.basename():label=CLUSTER, info.rolloutStrategyInfo.allAtOnceStrategyInfo.clusters.current.version:label=CURRENT_VERSION, info.rolloutStrategyInfo.allAtOnceStrategyInfo.clusters.current.syncState:label=SYNC_STATE, info.rolloutStrategyInfo.allAtOnceStrategyInfo.clusters.desired.version:label=DESIRED_VERSION, info.rolloutStrategyInfo.allAtOnceStrategyInfo.clusters.startTime:label=START_TIME, info.rolloutStrategyInfo.allAtOnceStrategyInfo.clusters.endTime:label=END_TIME, info.rolloutStrategyInfo.allAtOnceStrategyInfo.clusters.state:label=STATE, all_messages():label=MESSAGES)ctSN)_DEFAULT_API_VERSION@lib/googlecloudsdk/command_lib/container/fleet/packages/utils.py ApiVersionr @s r cX|y|jr|jjr|jjjr!|r t|_n t |_dg|_|jjjr"|r t|_n t|_dg|_yyyy)zSets format for `rollouts describe` depending on rollout strategy. Args: rollout: Rollout from `rollouts describe` args: Command line args less: Whether to show truncate rollout messages Returns: None Nz7info.rolloutStrategyInfo.rollingStrategyInfo.clusters[]z9info.rolloutStrategyInfo.allAtOnceStrategyInfo.clusters[]) inforolloutStrategyInforollingStrategyInfo3ROLLOUTS_DESCRIBE_ROLLING_TRUNCATED_MESSAGES_FORMATformat.ROLLOUTS_DESCRIBE_ROLLING_FULL_MESSAGES_FORMATflattenallAtOnceStrategyInfo5ROLLOUTS_DESCRIBE_ALLATONCE_TRUNCATED_MESSAGES_FORMAT0ROLLOUTS_DESCRIBE_ALLATONCE_FULL_MESSAGES_FORMAT)rolloutargslesss r FormatForRolloutsDescriberDs _  \\gll66||'';; I D OPdl||''== K F Edl >7\r c|jds|jdsgStjj|rgSg}t j |}|D]+}|st j |}|j|-|S)Nz.yamlz.yml)endswithospathisdirr load_all_pathdumpappend)r resourcesloaded_resourcesresourcedumped_resources r _LoadResourcesFromFiler)bsz -- DMM&$9 IWW]]4 I)''-"h (+o'# r c|jds|dz }tjj|dz}t j|dS)N/z/**/*T)pathname recursive)rrr dirnameglob)r glob_paths r _AllFilesUnderDirr1psD s CKDggood#g-) D r c@tj|jSr)pathlibPathstemr s r _VariantNameFromPathr7ys d   r cZ|jds|dz }|jdd}|S)Nr+)rsplit)r variant_names r _VariantNameFromDirr<}s/ s CKDC$, r c@tjj|Sr)rr expanduserr6s r ExpandPathForUserr?s   D !!r ctjj|}|}d|vrtjj|}|S)N$)rr r> expandvars)r user_expanded_pathvars_expanded_paths r _ExpandPathForUserAndVarsrEsBww))$/) ++,>? r c|s|St|}t|}tjj|r tjj ||S|S)a2Creates glob pattern by combining source and variants_pattern. Args: source: Directory or source configuration file. variants_pattern: Optional variants_pattern for use with source. Returns: A glob_pattern for use with 'VariantsFromGlobPattern'. If source is a directory, the pattern is applied within the directory. If source is not a directory i.e., a file, the pattern is not applied. Ex: source=/cfg/, variants_pattern='*.yaml'; returns '/cfg/*.yaml'. source=manifest.yaml, variants_pattern=*; returns manifest.yaml. )rErr r!join)sourcevariants_patternexpanded_sourceexpanded_variants_patterns r 'GlobPatternFromSourceAndVariantsPatternrLsN  M-f5/78HIWW]]?# 77<<)B CC r cd|dS)NzSource file or dir not found: .r r6s r _FileNotFoundMessagerOs )$q 11r cd|dS)Nz)Source is not of type directory or file: rNr r6s r _FileWrongTypeMessagerQs 4TF! <>/ * 2?C DD +1P 'r ctjj|}|}d|vrtjj|}t j|}|j i}t |dk(rtjj|drt|d}|r||d<|Stjj|dr\t|d}g}|D]@}tjj|} t| }|s0|j|B|r||d<|St |dkDr|D]} tjj| rt| }|s0t| } ||| <Atjj| sat| }g}|D]@}tjj|} t| }|s0|j|B|st| } ||| <|S)a!Returns a dictionary of input-format variants discovered from a glob. Gets all files from user-supplied glob pattern and creates variant(s). There will be a single variant 'default' if the inputted glob pattern has no wildcards, and multiple variants if there are wildcards. These variant(s) can be used for more advanced deployment setups. Args: glob_pattern: Pattern compatible with python's glob library Returns: A dict of input-formatted variants, for example: {'us-a': [resources...], 'us-b': [resources...]} rArrdefault)rr r>rBr/sortlenrSr)r!r1abspathextendr7r<) glob_patternuser_expanded_glob expanded_globpathsvariantsr% files_list all_resourcesfilefull_file_pathr r;s r VariantsFromGlobPatternrfsww)),7$-MGG&&}5M ))M "%**, (Z1_ ww~~eAh(q2i ': /9 uQx $U1X.jm$.*>:    y )  +& /% 5zA~  *40 -d3,#,(< 77== &t,  D77??40.,^<)    +  ,T2,#0(< !" /r cd}t|}|syt|dk\rt|d|kDr |dd|dzS|dS)zShows the first cluster-level message and truncates it if it's too long. Args: resource: A RolloutInfo resource Returns: Message limited to 40 characters (rrN...$_GetClusterLevelMessagesFromResourcerZ)r'truncated_message_lengthmessagess r !TransformTrimClusterLevelMessagesrosY  1( ;(  ]aC ,/GG A;00 1E 99 !r c|jdi}|r,|jdd}|rt|dkDr|dddzS|Sy)zTrims rollout-level message if it's too long. Args: resource: A Rollout resource Returns: String message limited to 40 characters rmessagerirhNrj)getrZ)r' rollout_inforollout_messages r TransformTrimRolloutLevelMessagerusTfb),"&&y"5O _  "s#e++  r cg}|sgS|jdijdi}|jD]q}|jdg}d|vr!|j|jdg|jdi}d|vsQ|j|jdgs|jdijdg}|r-|D](}|jdd}|s|j|*|S) zGathers cluster-level messages from a Rollout resource. Args: resource: A Rollout resource, from `... rollouts describe ...` Returns: A list of messages from the Rollout resource. rrclustersrncurrenterrors errorMessageri)rrvaluesr\r$) r'rnrollout_strategy_inforsrwrx info_errorserror info_messages r rlrls(  I",,vr266R,224l B/HXoohll:r23ll9b)GWoogkk*b12 5 VR(,,Xr:+YY~r2l  % /r cHt|}|syt|dk(r|dS|S)zReturns all cluster-level messages from a Rollout resource. Args: resource: A Rollout resource, from `... rollouts describe ...` Returns: A single string or string array of cluster-level messages. rirrrk)r'rns r TransformAllClusterLevelMessagesr8s/2( ;(   8} A; /r cg}|y|jdijdg}|D](}|jdd}|s|j|*|syt|dk(r|dS|S)zBGathers errors from 'info.Errors' and returns their errorMessages.rirryrzrr)rrr$rZ)r'rnryr~ error_messages r TransformListFleetPackageErrorsrIsy (  << # ' '" 5&eIInb1Moom$   8} A; /r c,|js||_|S)aFUpserts the correct fleet package name into fleet package resource. Args: fleet_package: A user-inputted FleetPackage which may or may not have a name fully_qualified_name: The fully qualified name of the FleetPackage resource. Returns: A FleetPackage that definitely has the correct fully qualified name. )name) fleet_packagefully_qualified_names r UpsertFleetPackageNamer\s   -M r c>t|r|jjj}|o|j dr^|dk(rd|jj_|S|jjjdd|jj_|S)aRemoves leading slash from fleet package path if it uses Cloud Build. If we don't remove the leading slash, parsing the path will fail for cloud build. See b/352756986#comment13 Args: fleet_package: A user-inputted FleetPackage which may need its path fixed. Returns: A FleetPackage with a fixed path if it uses Cloud Build, unchanged if it doesn't use Cloud Build. Nr+z./r)_FleetPackageUsesCloudBuildresourceBundleSelectorcloudBuildRepositoryr startswith)rr s r FixFleetPackagePathForCloudBuildrks!/  / / D D I ID DOOC0 IM ,,AAF   0 0 E E J J12 N ,,AAF r cV|xr&|jxr|jjduSr)rr)rs r rrs3P  . .P  . . C C4 Or )Fr)"__doc__r/rr3googlecloudsdk.corerr _RESOURCE_BUNDLE_PROJECT_SEGMENT!_RESOURCE_BUNDLE_LOCATION_SEGMENTrrrrr rr)r1r7r<r?rErLrOrQrVrfrorurlrrrrrr r r rs4 $ #$ $%!75395526.460< !"22=E7t$ &B"& 2r