+ SrSSKrSSKJr SSKJr SSKJr SSKJ r SSK J r SSK J r \ Rr\RrS rS rS rS rS rS S\S\R*S\S\4SjjrSrS\S\R*S\R24SjrS!S\S\R*S\R2S\4SjjrSrSrSrSrSr S!S\S\R*S\4Sjjr!Sr"Sr#g)"z'Notebook-executor schedules api helper.N) executions) exceptions)parser_extensions) resources)times)aiplatform_v1beta1_clientcfURRR5R5$)zGet the resource name for the schedule. Args: args: Argparse object from Command.Run Returns: The resource name in the form projects/{project}/locations/{location}/schedules/{schedule_id}. )CONCEPTSscheduleParse RelativeNameargss 9lib/googlecloudsdk/api_lib/notebook_executor/schedules.pyGetScheduleResourceNamers&    % % ' 4 4 66cSU;a[RRUSS9$[RRUSS9$![Ra N4f=f)zParse operation relative resource name to the operation reference object. Args: operation_name: The schedule operation resource name Returns: The operation reference object z /schedules/z2aiplatform.projects.locations.schedules.operations) collectionz(aiplatform.projects.locations.operations)rREGISTRYParseRelativeName WrongResourceCollectionException)operation_names rParseScheduleOperationr,sun$    1 1 B2    - -!K .   5 5   sAAAcn[RRURSS9R 5$)z$Get the URL for a schedule resource.z'aiplatform.projects.locations.schedules) relative_namer)rrrnameSelfLink)resources rGetScheduleUrirDs3    - -MM: .  HJrchUR(a [R"UR5$S$)z$Get the start time for the schedule.N) start_timerFormatDateTimers r GetStartTimer#Ls"26//  doo .KtKrchUR(a [R"UR5$S$)z"Get the end time for the schedule.N)end_timerr"rs r GetEndTimer&Qs"04   dmm ,G4Grrmessages for_update for_workbenchc SnU(d[R"XU5nURUR[ U5[ U5UR URURURUS9$)aMBuilds a Schedule message. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the specified API. for_update: Whether the schedule is to be used in an update request. for_workbench: Whether the schedule is for a Workbench execution. Returns: Instance of the Schedule message. N) displayName startTimeendTime maxRunCountcronmaxConcurrentRunCount allowQueueing!createNotebookExecutionJobRequest) executions_util'CreateExecutionCreateRequestForSchedule$GoogleCloudAiplatformV1beta1Schedule display_namer#r&max_runs cron_schedulemax_concurrent_runsenable_queueing)rr'r(r)execution_create_requests rCreateScheduler<Vs}"" ?? M   6 6##T"--    44(((@ 7  rcV[R"URR5$)zList filter for Workbench schedules. Args: schedule: The schedule item returned from List API to check. Returns: True if the schedule is for a Workbench notebook execution. )r3IsWorkbenchExecutionr2notebookExecutionJob)r s rFilterWorkbenchScheduler@zs&  - -00EE rservicecUR[X55nURnUc[R"SS5e[ R "UR5(a[R"SS5eU$)aChecks if the schedule is for a Colab Enterprise notebook execution and returns the schedule if so. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the specified API. service: The service to use to make the request. Returns: The schedule if it is of Colab Enterprise type. Raises: InvalidArgumentException: If the schedule is not of notebook execution type or is of Workbench type. SCHEDULE+Schedule is not of notebook execution type.zlSchedule is not of Colab Enterprise type. To manage Workbench schedules use `gcloud beta workbench` instead.)GetCreateScheduleGetRequestr2rInvalidArgumentExceptionr3r>r?)rr'rAr notebook_execution_job_requests rValidateAndGetColabSchedulerIs&[[1$A B(#+#M#M #+  - -A ))$99  - - :  /rskip_workbench_checkcUR[X55nURnUc[R"SS5eUR R cU(d[R"SS5eU$)aChecks if the schedule is for a Workbench notebook execution and returns the schedule if so. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the specified API. service: The service to use to make the request. skip_workbench_check: Whether to skip validation of the schedule type. Returns: The schedule if it is of Workbench type. Raises: InvalidArgumentException: If the schedule is not of notebook execution type. rCrDzmSchedule is not of Workbench type. To manage Colab Enterprise schedules use `gcloud colab schedules` instead.)rErFr2rrGr? kernelName)rr'rArJr rHs rValidateAndGetWorkbenchSchedulerMs([[t.($,#M#M #+  - -A %99DDL"  - - ;  /rc2UR[U5S9$)zBuilds a SchedulesGetRequest message. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the specified API. Returns: Instance of the SchedulesGetRequest message. r).AiplatformProjectsLocationsSchedulesGetRequestrrr's rrFrFs$==&t,>rc2UR[U5S9$)zBuilds a SchedulesDeleteRequest message. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the specified API. Returns: Instance of the SchedulesDeleteRequest message. rO)1AiplatformProjectsLocationsSchedulesDeleteRequestrrQs rCreateScheduleDeleteRequestrTs%@@&t,Arc2UR[U5S9$)zBuilds a SchedulesPauseRequest message. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the specified API. Returns: Instance of the SchedulesPauseRequest message. rO)0AiplatformProjectsLocationsSchedulesPauseRequestrrQs rCreateSchedulePauseRequestrWs%??&t,@rcfURURS9nUR[U5US9$)zBuilds a SchedulesResumeRequest message. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the specified API. Returns: Instance of the SchedulesResumeRequest message. )catchUp)r1googleCloudAiplatformV1beta1ResumeScheduleRequest)1GoogleCloudAiplatformV1beta1ResumeScheduleRequestenable_catch_up1AiplatformProjectsLocationsSchedulesResumeRequestr)rr'resume_schedule_requests rCreateScheduleResumeRequestr_ sL@@&&A @@&t,%ArcJUR[R"U5SS9$)zBuilds a SchedulesListRequest message. Args: args: Argparse object from Command.Run messages: Module containing messages definition for the specified API. Returns: Instance of the SchedulesListRequest message. z'create_notebook_execution_job_request:*)parentfilter)/AiplatformProjectsLocationsSchedulesListRequestr3GetParentForExecutionOrSchedulerQs rCreateScheduleListRequestre#s.  A A  < rsD. R.5)*^.??  ' '  70L H ! !!! !H ! !! @ @!P"' % %% @ @% %P$$$2 "HM $//@D*)2r