+EZSrSSKJr SSKJr SSKJr SSKJr SSKJ r SSK J r SSK J r SS K Jr SS KJr SS KJr SS KJr S!S jrS!SjrSrSrS!SjrSrS"SjrS!SjrS!SjrS!SjrSrSr Sr!Sr"S!S\ RFS\$4Sjjr%S"S\ RFS\$S\$4Sjjr&g )#z->-D-D-K-KLML  2 2 2 c^SU4Sjjn[RRSU"5SRU5SS9R U5 g)zAdd a resource argument for an execution to the parser. Args: parser: argparse parser for the command. verb: str, the verb to describe the resource, such as 'to update'. for_workbench: bool, whether the flag is added for a workbench execution. c`>[R"SU[R[T5S9$)Nz3aiplatform.projects.locations.notebookExecutionJobs resource_name projectsId locationsIdr ResourceSpec DEFAULT_PROJECT_ATTRIBUTE_CONFIGr)rrs rGetExecutionResourceSpec:_AddExecutionResourceArg..GetExecutionResourceSpec3s-  =#<<-m<  r executionz)Unique resource name of the execution {}.TrequiredN)znotebook execution jobr ConceptParser ForResourceformat AddToParser)parserverbrr$s ` r_AddExecutionResourceArgr0*sI++ 188> , KrcSSjn[RRSU"5SRU5SS9R U5 g)zAdd a resource argument for a schedule to the parser. Args: parser: argparse parser for the command. verb: str, the verb to describe the resource, such as 'to update'. schedulec\[R"SU[R[5S9$)zAdd a resource argument for a schedule to the parser. Args: resource_name: str, the name of the resource to use in attribute help text. Returns: A concepts.ResourceSpec for a schedule. z'aiplatform.projects.locations.schedulesrr!rs rGetScheduleResourceSpec7AddScheduleResourceArg..GetScheduleResourceSpecKs,  1#<<-/  rz:Unique, system-generated resource name of the schedule {}.Tr'N)r2r))r.r/r5s rAddScheduleResourceArgr7CsH"++BII$O , KrcS Sjn[R"SU"5SSSS0S9n[R"U/SS /0S 9R U5 g ) ztAdd a resource argument for a Dataform repository to the parser. Args: parser: argparse parser for the command. c\[R"SU[R[5S9$)Nz(dataform.projects.locations.repositoriesrr!r4s r!GetDataformRepositoryResourceSpecKAddDataformRepositoryResourceArg..GetDataformRepositoryResourceSpecks*  2#<<-/  rz--dataform-repository-namezEUnique name of the Dataform repository to source input notebook from.Trr(flag_name_overridesz!--dataform-repository-name.region--regioncommand_level_fallthroughsN)zdataform repositoryr ResourcePresentationSpecr r*r-)r.r:dataform_repository_resources r AddDataformRepositoryResourceArgrEdsb"4!L!L"')M$R. "#$ . |" Krc[R"SS[U5[RS9n[R R SUSRU5SS9RU5 g) aAdd a resource argument for a Vertex AI region to the parser. Args: parser: argparse parser for the command. verb: str, the verb to describe the resource, such as 'to update'. for_workbench: bool, whether the flag is added for a workbench execution. zaiplatform.projects.locationsr)rr rr?zCloud region {}.Tr'N) rr"rr#r r*r+r,r-)r.r/rregion_resource_specs r_AddRegionResourceArgrHsj"..%+M::: ++% , KrcS Sjn[R"SU"5SSSS0S9n[R"U/SS /0S 9R U5 g ) zpAdd a resource argument for a runtime template to the parser. Args: parser: argparse parser for the command. c\[R"SU[R[5S9$)Nz6aiplatform.projects.locations.notebookRuntimeTemplatesrr!r4s rGetRuntimeTemplateResourceSpecF_AddRuntimeTemplateResourceArg..GetRuntimeTemplateResourceSpecs*  @#<<-/  rz--notebook-runtime-templatezThe runtime template specifying the compute configuration for the notebook execution. The runtime template should be in the same region as the execution.Trr<r=z"--notebook-runtime-template.regionr?r@N)znotebook runtime templaterB)r.rKruntime_template_resources r_AddRuntimeTemplateResourceArgrNsf1II#$&$R.  ! / " KrcURSSS9nU(aURSSSS9 ON[USU5 URSSSS9 URSS SS 9 [RR U5 URS SSS 9nU(d*URS S9n[ U5 URSSS9 URSS9nURSSSS9 URSSS9 U(dURSSS9 URSS[R"5SS9 U(a_URSS9n[R"U5 [R"US5 URSSS S!9 URS"S#SS9 O=[U5 URS$SSS%9nURS&S'S9 URS"S(S)S9 URS*S+SS9 g,)-z3Adds flags for creating an execution to the parser.z#Configuration of the execution job.Thelpr(z--execution-display-namez"The display name of the execution. to create--display-namez--execution-job-idzWThe id to assign to the execution job. If not specified, a random id will be generated.)rQhiddenz"Source of the notebook to execute.)rQr(mutexaThe Dataform repository containing the notebook. Any notebook created from the Colab UI is automatically stored in a Dataform repository. The repository name can be found via the Dataform API by listing repositories in the same project and region as the notebook.)rQz --commit-shaz_The commit SHA to read from the Dataform repository. If unset, the file will be read from HEAD.z"The Cloud Storage notebook source.z--gcs-notebook-urizWThe Cloud Storage uri pointing to the notebook. Format: gs://bucket/notebook_file.ipynbz --generationzjThe version of the Cloud Storage object to read. If unset, the current version of the object will be used.z--direct-contentzThe direct notebook content as IPYNB. This can be a local filepath to an .ipynb file or can be set to `-` to read content from stdin.z--execution-timeoutzThe max running time of the execution job, as a duration. See '$ gcloud topic datetimes' for details on formatting the input duration.24hrQtypedefaultz+Compute configuration of the execution job.zThe Cloud KMS encryption key (customer-managed encryption key) used to protect the execution. The key must be in the same region as the execution. If not specified, Google-managed encryption will be used.z --kernel-namez)The kernel name to use for the execution.python3)rQrYz--service-accountz+The service account to run the execution asz!Identity to run the execution as.)rQrUr(z --user-emailzyThe user email to run the execution as. This requires the provided runtime template to have end user credentials enabled.z,The service account to run the execution as.Fz--gcs-output-uriz]The Cloud Storage location to upload notebook execution results to. Format: gs://bucket-name.N) add_group add_argumentrHr ASYNC_FLAGr-rErDuration colab_flagsAddCustomEnvSpecFlagsAddKmsKeyResourceArgrN) r. is_schedulerexecution_groupnotebook_source_groupdataform_source_groupgcs_source_groupcustom_env_spec_groupexecution_identity_groups rAddCreateExecutionFlagsris$$ 0%/  " 1! &+}=   1!   %  OO')33 / 4 1;;<%%:;&&,' +44 /5 ,   9  && '     ! +55 :6%%&;<$$ P    8!    :! #?3.88 0 9  ))K* )) ;*  ' rc\[USU5 [RRU5 g)zAdds flags for deleting an execution to the parser. Args: parser: argparse parser for the command. for_workbench: bool, whether the flags are for a workbench execution. to deleteN)r0rr]r-r.rs rAddDeleteExecutionFlagsrmGs"6; >//f%rc[USU5 g)zAdds flags for describing an execution to the parser. Args: parser: argparse parser for the command. for_workbench: bool, whether the flag is added for a workbench execution. to describeN)r0rls rAddDescribeExecutionFlagsrpSs6=-@rcp[USU5 URR[R5 g)zConstruct groups and arguments specific to listing executions. Args: parser: argparse parser for the command. for_workbench: bool, whether the flag is added for a workbench execution. z for which to list all executionsN)rH display_info AddUriFuncexecutions_utilGetExecutionUrirls rAddListExecutionsFlagsrv]s0 0-   !@!@Arc[US5 g)z2Add flags for describing a schedule to the parser.roNr7r.s rAddDescribeScheduleFlagsrzjs /rcZ[US5 [RRU5 g)z1Adds flags for deleting a schedule to the parser.rkN)r7rr]r-rys rAddDeleteScheduleFlagsr|os-//f%rc[US5 g)z0Adds flags for pausing a schedule to the parser.zto pauseNrxrys rAddPauseScheduleFlagsr~us ,rcB[US5 URSSSSSS9 g) z1Adds flags for resuming a schedule to the parser.z to resumez--enable-catch-upzEnables backfilling missed runs when the schedule is resumed from PAUSED state. If enabled, all missed runs will be scheduled and new runs will be scheduled after the backfill is complete. store_trueenable_catch_upFrQactiondestrYN)r7r\rys rAddResumeScheduleFlagsrzs3- D  rr.rcp[USU5 URR[R5 g)zConstruct groups and arguments specific to listing schedules. Args: parser: argparse parser for the command. for_workbench: whether the flags are for listing workbench schedules. zfor which to list all schedulesN)rHrrrsschedules_utilGetScheduleUrirls rAddListSchedulesFlagsrs0 /   !>!>?r is_updatec$URSSS9nU(d[USUS9 [USUS9 O [US5 UR SS U(dSOS S9 UR S S [ R RS 9 UR SS[ R RS 9 UR SS[S 9 UR SSU(dSOS S9 UR SS[U(dSOSS9 UR SSSSU(dS OSS9 g)zAdds flags for creating or updating a schedule to the parser. Args: parser: argparse parser for the command. is_update: whether the flags are for updating a schedule. for_workbench: whether the flags are for a workbench schedule. zConfiguration of the schedule.TrPrR)r)rbrz to updaterSz!The display name of the schedule.Fz --start-timezThe timestamp after which the first run can be scheduled. Defaults to the schedule creation time. Must be in the RFC 3339 (https://www.rfc-editor.org/rfc/rfc3339.txt) format. E.g. "2026-01-01T00:00:00Z" or "2026-01-01T00:00:00-05:00")rQrXz --end-timeaTimestamp after which no new runs can be scheduled. If specified, the schedule will be completed when either end_time is reached or when scheduled_run_count >= max_run_count. If neither end time nor max_run_count is specified, new runs will keep getting scheduled until this Schedule is paused or deleted. Must be in the RFC 3339 (https://www.rfc-editor.org/rfc/rfc3339.txt) format. E.g. "2026-01-01T00:00:00Z" or "2026-01-01T00:00:00-05:00"z --max-runszThe max runs for the schedule.z--cron-scheduleawCron schedule (https://en.wikipedia.org/wiki/Cron) to launch scheduled runs. To explicitly set a timezone to the cron tab, apply a prefix in the cron tab: "CRON_TZ=${IANA_TIME_ZONE}" or "TZ=${IANA_TIME_ZONE}". The ${IANA_TIME_ZONE} may only be a valid string from IANA time zone database. For example, "CRON_TZ=America/New_York 1 * * * *", or "TZ=America/New_York 1 * * * *".z--max-concurrent-runszMaximum number of runs that can be started concurrently for this Schedule. This is the limit for starting the scheduled requests and not the execution of the notebook execution jobs created by the requests.NrWz--enable-queueingzEnables new scheduled runs to be queued when max_concurrent_runs limit is reached. If set to true, new runs will be queued instead of skipped.renable_queueingr) r[rHrir7r\rDatetime ParseUtcTimeint)r.rrschedule_groups rAddCreateOrUpdateScheduleFlagsrsd## +$. &+]KD 6;/ .$t%  C    , ,  C    , ,  +   %t%    ad  ( $e$ rN)F)FF)'__doc__(googlecloudsdk.api_lib.notebook_executorrrtrrgooglecloudsdk.callioperrr googlecloudsdk.calliope.conceptsrr +googlecloudsdk.command_lib.colab_enterpriser r_(googlecloudsdk.command_lib.util.conceptsr r googlecloudsdk.corer rr0r7rErHrNrirmrprvrzr|r~rArgumentInterceptorboolrrrrrsCRP/(451LDG* 2BD0!HBJ &A B0 & - "IN @  0 0 @AE @"\  0 0\\\r