',dZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z dd d Zejej ej"j$ej"j&Gd d ej(Zee_y )z.Create command for Colab Enterprise Schedules.)util) schedules)base) endpoint_util)flags)logzQ Create a schedule to run a Colab Enterprise notebook execution job. a To create a schedule in region `us-central1` with the following schedule properties: - display name: `my-schedule`, - cron schedule: `TZ=America/Los_Angeles * * * * *`, - maximum concurrent runs allowed: 1, - start time: 2025-01-01T00:00:00-06:00, for a notebook execution job: - with display name `my-execution`, - running notebook file from Cloud Storage URI `gs://my-bucket/my-notebook.ipynb`, - compute configured from runtime template `my-runtime-template-id`, - running with service account `my-service-account@my-project.iam.gserviceaccount.com`, - with results uploaded to Cloud Storage bucket `gs://my-bucket/results` Run the following command: $ {command} --region=us-central1 --display-name=my-schedule --cron-schedule='TZ=America/Los_Angeles * * * * *' --max-concurrent-runs=1 --start-time=2025-01-01T00:00:00-06:00 --execution-display-name=my-execution --notebook-runtime-template=my-runtime-template-id --gcs-notebook-uri=gs://my-bucket/my-notebook.ipynb --service-account=my-service-account@my-project.iam.gserviceaccount.com --gcs-output-uri=gs://my-bucket/results ) DESCRIPTIONEXAMPLESc&eZdZdZedZdZy)CreatezCreate a schedule.c2tj|dy)z^Register flags for this command. Args: parser: argparse parser for the command F) is_updateN)rAddCreateOrUpdateScheduleFlags)parsers %lib/surface/colab/schedules/create.pyArgsz Create.Args6s ((5Ac|j}tj|j}|jjj }|j d}tjd|5tj|}|j}|jtj||}tj|j d|cdddS#1swYyxYw)zThis is what gets called when the user runs this command. Args: args: argparse.Namespace, The arguments that this command was invoked with. Returns: The API response for creating the schedule. locationsIdBETA)versionregionschedule)resourcekindN) ReleaseTrackr GetMessagesCONCEPTSrParseAsDictrAiplatformEndpointOverrides GetClientprojects_locations_schedulesr schedules_utilCreateScheduleCreateRequestrCreatedResourcename) selfargs release_trackmessages region_refr api_clientschedules_servicers rRunz Create.Run?s%%'M 1 1 34H%%++-J     /F  2 2v >>-0j$AA"))  4 4T8 Dh 8==zB    s A)C55C>N)__name__ __module__ __qualname____doc__ staticmethodrr/rrr r 1s BBrr N)r3'googlecloudsdk.api_lib.colab_enterpriser(googlecloudsdk.api_lib.notebook_executorrr$googlecloudsdk.calliopergooglecloudsdk.command_lib.air,googlecloudsdk.command_lib.notebook_executorrgooglecloudsdk.corer_DETAILED_HELPDefaultUniverseOnly ReleaseTracksrGAr CreateCommandr detailed_helpr5rrrBs 58P(7># 0D%%(($*;*;*@*@A#T  #B#L&r