1VpdZddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l m Z dd lmZdd lm Zdd lmZdd lmZdd lmZddlmZddlZGddeZidej0dej2dej2dej4dej4dej0dej0dej2dej2dej4dej4dej4dej2dej2d ej4d!ej0d"ej2ej2ej4ej4ej0ej0ej2ej2ej4ej4ej4ej2ej2ej4d# Zd$Zd%Zd&Zd'Zd(Z d)Z!d*Z"d+Z#d,Z$d-Z%d.Z&d/Z'd0Z(d1Z)d2Z*d3Z+d4Z,d5Z-d6Z.d7Z/d8Z0d9Z1d:Z2d;Z3d<Z4d=Z5 dDd>Z6d?Z7d@Z8dAZ9dBZ:dCZ;y)Ez-Utility functions for GCE OS Config commands.)absolute_import)division)unicode_literals)Enum)encoding)apis)base) exceptions) common_args)log) properties) resources)yamlNceZdZdZdZdZdZy)InstanceDetailsStatesz9Indicates instance progress during a patch job execution.N)__name__ __module__ __qualname____doc__NOTIFIEDPATCHINGFINISHED9lib/googlecloudsdk/command_lib/compute/os_config/utils.pyrr$sA ( ( (rrinstancesAckedinstancesApplyingPatchesinstancesDownloadingPatchesinstancesFailedinstancesInactiveinstancesNotifiedinstancesPendinginstancesRebootinginstancesStartedinstancesSucceeded instancesSucceededRebootRequiredinstancesTimedOutinstancesRunningPrePatchStepinstancesRunningPostPatchStepinstancesNoAgentDetectedackedInstanceCountapplyingPatchesInstanceCount) downloadingPatchesInstanceCountfailedInstanceCountinactiveInstanceCountnotifiedInstanceCountpendingInstanceCountrebootingInstanceCountstartedInstanceCountsucceededInstanceCount$succeededRebootRequiredInstanceCounttimedOutInstanceCountprePatchStepInstanceCountpostPatchStepInstanceCountnoAgentDetectedInstanceCount)zgs://z&https://www.googleapis.com/storage/v1/zhttps://storage.googleapis.com/dc|jr |jS|jrt|jtSy)z-Returns the batch size for listing resources.N) page_sizelimitmin_MAX_LIST_BATCH_SIZE)argss rGetListBatchSizerDUs/ ^^ >> zz tzz/ 00 rc(dj||gS)z.Returns the URI path of a GCP parent resource./join) parent_name parent_ids rGetParentUriPathrK_s ; * ++rctd|S)z&Returns the URI path of a GCP project.projectsrK)projects rGetProjectUriPathrPds *g ..rcBttd|td|S)z/Returns the URI path of projects/*/locations/*.rM locationsrN)rOlocations rGetProjectLocationUriPathrTis# z7+{H- //rctd|S)z%Returns the URI path of a GCP folder.foldersrN)folders rGetFolderUriPathrXps )V ,,rctd|S)z+Returns the URI path of a GCP organization. organizationsrN) organizations rGetOrganizationUriPathr\us /< 88rc,djd|d|gS)z.Returns the URI path of an osconfig patch job.rFrM patchJobsrG)rO patch_jobs rGetPatchJobUriPathr`zs :w Y? @@rc*|jddS)z0Returns the name of a GCP resource from its URI.rFr)split)uris rGetResourceNamerds 3 rc*dj|d|gS)z6Returns the relative path of an osconfig guest policy.rF guestPoliciesrG)parent guest_policys rGetGuestPolicyRelativePathris 6?L9 ::rc*dj|d|gS)z>Returns the relative path of an osconfig os policy assignment.rFosPolicyAssignmentsrG)rgos_policy_assignments r!GetOsPolicyAssignmentRelativePathrms 602FG HHrc.tjd|S)z7Returns the messages module with the given api_version.osconfig)rGetMessagesModule) api_versions r GetApiMessagerrs   K 88rc|jj}|tjjk(ry|tjjk(ry|tjj k(ryt j|)z7Return api version for the corresponding release track.v1alphav1betav1calliope_command ReleaseTrackr ALPHABETAGAcore_exceptionsUnsupportedReleaseTrackErrorrC release_tracks r GetApiVersionrl''446-d''--- ))... )),,,   6 6} EErc|jj}|tjjk(ry|tjjk(ry|tjj k(ryt j|)z:Return v2 api version for the corresponding release track.v2alphav2betav2rwrs rGetApiVersionV2rrrc|jj}|tjjk(ry|tjjk(r yt j |)z8Returns api version for the corresponding release track.zkTo check operation status, run: gcloud alpha compute os-config os-policy-assignments operations describe {}zeTo check operation status, run: gcloud compute os-config os-policy-assignments operations describe {})rxryr rzr|r}r~rs r!GetOperationDescribeCommandFormatrs`''446-d''--- ;)),,, 78  6 6} EErc |jdj|d}tjdj||djtj j jtj jjj||jdd td j|| |jd d tdj|| y)z;Adds project, folder, and organization flags to the parser.zi The scope of the {}. If a scope is not specified, the current project is used as the default.T)helpmutexzThe project of the {} {}.aj The project name to use. If a project name is not specified, then the current project is used. The current project can be listed using gcloud config list --format='text(core.project)' and can be set using gcloud config set project PROJECTID. `--project` and its fallback `{core_project}` property play two roles. It specifies the project of the resource to operate on, and also specifies the project for API enablement check, quota, and billing. To specify a different project for quota and billing, use `--billing-project` or `{billing_project}` property. ) core_projectbilling_project)help_text_to_prependhelp_text_to_overwritez--folder FOLDER_IDzThe folder of the {} {}.)metavartyperz--organizationORGANIZATION_IDzThe organization of the {} {}.N) add_groupformatr ProjectArgumentr VALUEScorerObilling quota_project AddToParser add_argumentstr)parsernounverbparent_resource_groups rAddResourceParentArgsrs ** $fTl  + 6==dDI   &!((--55$++33AA  CDEPK#E%$$  % , ,T4 8 % $$  + 2 24 > %rc,djd|d|gS)z5Returns the URI path of an osconfig patch deployment.rFrMpatchDeploymentsrG)rOpatch_deployments rGetPatchDeploymentUriPathrs :w(:):|jxs0|xr |jxs|jxs|xr |j }|s|dfSd}d}d} |r|jrg}|r|jrg}|r|j rg} |jr6g}|jj dD]} |jd| z|jr6g}|jj dD]} |jd| z|jr3g} |jj dD]} | j| |jst|||_d} d}|jjr9|jjD] }|jr|} |js|}"| s t||} |s t||}| |g|j_ | jst||| _ |jst|||_ ||| j_||| j_| | |j_|dfS)aSets selectors inside policy orchestrator. Args: args: args to the command req: request messages: messages for selected v2 API version api_version: api version orchestrator: orchestrator to set selectors in use_clear: if true, clear_projects flag is used to clear selectors (optional) Returns: modified request, boolean indicating if selectors were set FNrz projects/zfolders/T)include_projectsclear_projectsinclude_folders clear_foldersinclude_locationsclear_locationsrbappendorchestrationScoper selectorsresourceHierarchySelectorlocationSelectorrrrincludedProjectsincludedFoldersincludedLocations)rCrrrq orchestrator use_clear selectors_setincluded_projectsincluded_foldersincluded_locations project_id folder_idrShierarchy_selectorlocation_selectorselectors r$ModifyOrchestratorPolicySetSelectorsrzs$ .  +++.  .  ***.    .  ,,,   :4&&4%%4'' ++11#6 {Z787 ))//4 j9455 **005)6  ( ($X{;# $$.. 33==  + +%  $ $$ > =+ <+/,!!+  5 5+HkB0  + +"8[9& "00A!00@#;M&&8 drct|}t|}|j}|jdk(r#t |j |j\}}d}|dk(r-|j }|j|_||_ nc|dk(r-|j}|j|_||_ n1|dk(r,|j}|j|_||_||j_|j"r|j"|j_|jj'|_|j(j'|_t+|||||\}}|j-|_dj1|j2j5ddd|_|S)FReturns modified request with parsed orchestartor's policy assignment.upsertNrrrrF)rrrractionr policy_file,GoogleCloudOsconfigV2alphaPolicyOrchestrator.GoogleCloudOsconfigV2alphaOrchestratedResourceorchestratedResource,googleCloudOsconfigV2alphaPolicyOrchestrator+GoogleCloudOsconfigV2betaPolicyOrchestrator-GoogleCloudOsconfigV2betaOrchestratedResource+googleCloudOsconfigV2betaPolicyOrchestrator'GoogleCloudOsconfigV2PolicyOrchestrator)GoogleCloudOsconfigV2OrchestratedResource'googleCloudOsconfigV2PolicyOrchestratorosPolicyAssignmentV1PayloadridupperstaterNamepolicyOrchestratorIdrHrgrb)rrCrrqrr_req_orchestrators r#ModifyOrchestrorPolicyCreateRequestr*s %+ ; '(&88: [[H$F (55%!qILLN??A)8HC4hKKM>>@)7GC3dGGI::<)3CC/''C ^^/3~~)), KK--/::++- / C;(8 &#q !XXZ#xx ((-cr23#* *rc t|}t|}d}|dk(r |j}n#|dk(r |j}n|dk(r |j}g}|j r0|j j |_|jd|jrEt|j|j\}}||j_ |jd|jr,|j|j_|jd|jr0|jj |_|jdt!|||||d \}} | r|jd d j#||_|S) rNrrrrz5orchestrated_resource.os_policy_assignment_v1_payloadzorchestrated_resource.idr%T)rzorchestration_scope.selectorsr)rrrrrr!rr$rrrrrr"rr#r%rrHr) unused_refrCrrqrr) update_maskrr(modifieds r#ModifyOrchestrorPolicyUpdateRequestr/!sw %+ ; '(IGGhFFdBB+ [["kk//1x  $F (55%!q !))E? ^^/3~~)),12 ZZ!ZZ--/w6 C;(8D-#x6788K(#. *rc0|xjdz c_|S)z*Extends request with global location part.z/locations/global)rgr, unused_argsrs r!ModifyOrchestrorPolicyListRequestr3Ws**##* *rc|jrAtjjt |j |j |S)zReminds user of the command to check operation status. Args: response: Response from CreateOsPolicyAssignment args: gcloud args Returns: The original response )async_r outPrintrrname)responserCs r&LogOutOperationCommandForAsyncResponser:^s8 [[GGMM)$/66x}}EG /rc.|jdz|_|S)Nz/projectFeatureSettings)r8r1s r$UpdateProjectFeatureSettingsResourcer<ps XX1 1#( *r)F)rEs4&'%,(.<=#*)$ D"+44" 5 > >""#8#A#A " ,55 " .77 ".77"-66"/88"-66"/88"'(=(F(F".77"#$9$B$B"$%:%C%C"  5 > >!"&/88'"(#$9$B$B)"*(='E'E0992;;2;;1::3<<1::3<<,A,J,J2;;!6!?!?"7"@"@$9$B$BC"H4 , / /- 9 A  ; I 9 F F F$NO J #> F   D DD$ D"?Dnb3 l3 l  $ r