
                         ,   d Z ddlmZ ddlmZ ddlmZ ddlm	Z	 ddl
mZ ddlmZ dd	d
Zej                   ej                   ej"                  j$                  ej"                  j&                         G d dej(                                      Zee_        y)z.Update command for Colab Enterprise Schedules.    )util)	schedules)base)endpoint_util)flags)logzD
        Update a Colab Enterprise notebook execution schedule.
    z
        To update the cron schedule and max runs of a schedule with id `my-schedule`, in region `us-central1`, run:

        $ {command} my-schedule --region=us-central1 --cron-schedule='TZ=America/Los_Angeles 0 0 * * 0' --max-runs=2
    )DESCRIPTIONEXAMPLESc                   &    e Zd ZdZed        Zd Zy)UpdatezUpdate a schedule.c                 2    t        j                  | d       y)z^Register flags for this command.

    Args:
      parser: argparse parser for the command
    T)	is_updateN)r   AddCreateOrUpdateScheduleFlags)parsers    %lib/surface/colab/schedules/update.pyArgszUpdate.Args*   s     
((4@    c                 0   | j                         }t        j                  | j                               }|j                  j                  j                         }|j                         d   }t        j                  d|      5  t        j                  |      }|j                  }t        j                  |||       |j                  t        j                  ||            }t        j                   |j"                  d       |cddd       S # 1 sw Y   yxY w)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 updating the schedule.
    locationsIdBETA)versionregionschedule)resourcekindN)ReleaseTrackr   GetMessagesCONCEPTSr   ParseAsDictr   AiplatformEndpointOverrides	GetClientprojects_locations_schedulesschedules_utilValidateAndGetColabSchedulePatchCreateSchedulePatchRequestr   UpdatedResourcename)	selfargsrelease_trackmessagesschedule_refr   
api_clientschedules_servicer   s	            r   Runz
Update.Run3   s     %%'M 1 1 34H==))//1L  "=1F		2	2v
 >>-0j

1
1 
 00
+ #((

3
3D(
Ch 
8==zB!
 
 
s   B DDN)__name__
__module____qualname____doc__staticmethodr   r1    r   r   r   r   %   s      A Ar   r   N)r5   'googlecloudsdk.api_lib.colab_enterpriser   (googlecloudsdk.api_lib.notebook_executorr   r$   googlecloudsdk.callioper   googlecloudsdk.command_lib.air   ,googlecloudsdk.command_lib.notebook_executorr   googlecloudsdk.corer   _DETAILED_HELPDefaultUniverseOnlyReleaseTracksr   GAr   UpdateCommandr   detailed_helpr7   r   r   <module>rD      s     5 8 P ( 7 > #		 D%%(($*;*;*@*@A*T * B *X & r   