
                             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l	mZ  ej                  ej                  j                         G d	 d
ej                               Zy)z#`gcloud tasks renew-lease` command.    )absolute_import)division)unicode_literals)GetApiAdapter)base)flags)parsersc                   &    e Zd ZdZed        Zd Zy)
RenewLeasez*Renew the lease on a task in a pull queue.c                     t        j                  | d       t        j                  |        t        j                  | d       t        j                  |        y )Nzto renew the lease ofrenewing)r   AddTaskResourceArgsAddLocationFlagAddTaskLeaseScheduleTimeFlagAddTaskLeaseDurationFlag)parsers     lib/surface/tasks/renew_lease.pyArgszRenewLease.Args   s?    	f&=>	&!	&&vz:	""6*    c                 T   t        | j                               j                  }t        j                  |j
                  |j                        }t        j                  |j                  |      }t        j                  |j                        }|j                  ||j                  |      S )N)r   ReleaseTracktasksr	   
ParseQueuequeuelocation	ParseTasktaskFormatLeaseDurationlease_durationr   schedule_time)selfargstasks_client	queue_reftask_refdurations         r   RunzRenewLease.Run&   s|     !2!2!45;;L""4::t}}=I  I6H**4+>+>?H""8T-?-?JJr   N)__name__
__module____qualname____doc__staticmethodr   r'    r   r   r   r      s    2+ +Kr   r   N)r+   
__future__r   r   r   googlecloudsdk.api_lib.tasksr   googlecloudsdk.callioper    googlecloudsdk.command_lib.tasksr   r	   ReleaseTracksr   ALPHACommandr   r-   r   r   <module>r5      sZ    * &  ' 6 ( 2 4 D%%++,K K -Kr   