JdZddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z dd lmZej Ze j"Zd d d Zej&ej(ej*j,Gd dej.Zee_y)z'Create command for Workbench Schedules.)util) schedules)base)parser_arguments)parser_extensions) endpoint_util)flags)logzJ Create a schedule to run a Workbench notebook execution job. al 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`, - running with service account `my-service-account@my-project.iam.gserviceaccount.com`, - with results uploaded to Cloud Storage bucket `gs://my-bucket/results` - with machine type `n1-standard-4`, 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 --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 --machine-type=n1-standard-4 ) DESCRIPTIONEXAMPLESc2eZdZdZedefdZdefdZy)CreatezCreate a schedule.parserc4tj|ddy)z^Register flags for this command. Args: parser: argparse parser for the command FT) is_update for_workbenchN)r AddCreateOrUpdateScheduleFlags)rs )lib/surface/workbench/schedules/create.pyArgsz Create.Args<s ((%targsc|j}tj|j}|jjj }|j d}tjd|5tj|}|j}|jtj||d}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)versionregionT)rschedule)resourcekindN) ReleaseTrackr GetMessagesCONCEPTSrParseAsDictrAiplatformEndpointOverrides GetClientprojects_locations_schedulesrschedules_utilCreateScheduleCreateRequestr CreatedResourcename) selfr release_trackmessages region_refr api_clientschedules_servicers rRunz Create.RunGs%%'M 1 1 34H%%++-J     /F  2 2v >>-0j$AA"))  4 4HD h  8==zB    s A+C77DN) __name__ __module__ __qualname____doc__ staticmethodArgumentInterceptorr Namespacer2rrrr7s-&irrN)r6'googlecloudsdk.api_lib.colab_enterpriser(googlecloudsdk.api_lib.notebook_executorrr(googlecloudsdk.callioperrrgooglecloudsdk.command_lib.air,googlecloudsdk.command_lib.notebook_executorr googlecloudsdk.corer r8r9_DETAILED_HELPDefaultUniverseOnly ReleaseTracksr r CreateCommandr detailed_helpr:rrrFs .8P(457>#'::  ' '  0D%%**+'T  ','T&r