dZddlmZddlmZddlmZddlmZddlmZGddejZ Gd d e Z Gd d e Z Gd de Zy)zAPI Library for gcloud tasks.)absolute_import)division)unicode_literals) list_pager) exceptionsceZdZdZy)"ModifyingPullAndAppEngineTaskErrorzEError for when attempt to create a queue as both pull and App Engine.N)__name__ __module__ __qualname____doc__)lib/googlecloudsdk/api_lib/tasks/tasks.pyr r sMrr c2eZdZdZdZddZd dZdZdZy) BaseTasks!API client for Cloud Tasks tasks.c ||_||_yN)messages tasks_service)selfrrs r__init__zBaseTasks.__init__ sDM&DrNc|jj|j}tj|j |||ddS)N)parenttaskspageSize) batch_sizelimitfieldbatch_size_attribute)r1CloudtasksProjectsLocationsQueuesTasksListRequest RelativeNamer YieldFromListr)r parent_refr page_sizerequests rListzBaseTasks.List$sT GG**, H .   # # G J 88rc|jj|j|}|jj |S)N)name responseView)r0CloudtasksProjectsLocationsQueuesTasksGetRequestr#rGet)rtask_ref response_viewr's rr-z BaseTasks.Get,sG FF&&(& G (     ! !' **rc|jj|j}|jj |SN)r*)r3CloudtasksProjectsLocationsQueuesTasksDeleteRequestr#rDeleterr.r's rr3zBaseTasks.Delete3sD II&&( J *     $ $W --rc|jj|j}|jj |Sr1)r0CloudtasksProjectsLocationsQueuesTasksRunRequestr#rRunr4s rr7z BaseTasks.Run9sD FF&&( G *     ! !' **r)Ndr) r r r r rr(r-r3r7rrrrrs)'8+. +rrc$eZdZdZ ddZddZy)TasksrNc8|r|jnd}|jj|||}|r||_|jj |jj ||j}|j j|S)8Prepares and sends a Create request for creating a task.N)r* scheduleTimeappEngineHttpRequesttaskcreateTaskRequestr)r#rTask httpRequest3CloudtasksProjectsLocationsQueuesTasksCreateRequestCreateTaskRequestrCreate) rr%r. schedule_timeapp_engine_http_request http_requestr*r@r's rrGz Tasks.CreateCs'/8 "DD ==    4  6D%d II"mm==4=H**, J .     $ $W --rc|jj|j|}|jj |Sz9Prepares and sends a Create request for buffering a task.)queuetaskIdr3CloudtasksProjectsLocationsQueuesTasksBufferRequestr#rBufferrr%task_idr's rrQz Tasks.BufferRsCmmOO%%'PG    $ $W --rNNNN)r r r r rGrQrrrr:r:@s)<@8< ..rr:cBeZdZdZd dZ d dZdZdZ d dZdZ y) AlphaTasksrc|jj|j|}|jj |SrLrOrRs rrQzAlphaTasks.Buffer]sCmmOO%%'P9G    $ $W --rNcF|r |r td|r|jnd}|jj||||}|jj |jj ||j}|j j|S)r<zFAttempting to send PullMessage and AppEngineHttpRequest simultaneouslyN)r*r= pullMessager>r?rA)r r#rrCrErFrrG) rr%r.rH pull_messagerIr*r@r's rrGzAlphaTasks.Createds/ .  '/8 "DD ==    <4  6D II"mm==4=H**, J .     $ $W --rc|jj||}|jj}|||j}|jj |S)aConstructs and sends a tasks RenewLease request to the Cloud Tasks API. Args: task_ref: A cloudtasks.projects.locations.queues.tasks resource reference schedule_time: string formatted as an ISO 8601 datetime with timezone lease_duration: string of an integer followed by 's', (e.g. '10s') for the number of seconds for the new lease Returns: The response of the tasks RenewLease request )r= leaseDuration)renewLeaseRequestr*)rRenewLeaseRequest7CloudtasksProjectsLocationsQueuesTasksRenewLeaseRequestr#r RenewLease)rr.rHlease_durationrenew_lease_request request_clsr's rrbzAlphaTasks.RenewLeaseusf--99".:B==JJ,?'4468G    ( ( 11rc|jj|}|jj}|||j}|jj |S)a6Constructs and sends a tasks CancelLease request to the Cloud Tasks API. Args: task_ref: A cloudtasks.projects.locations.queues.tasks resource reference schedule_time: string formatted as an ISO 8601 datetime with timezone Returns: The response of the tasks CancelLease request r=)cancelLeaseRequestr*)rCancelLeaseRequest8CloudtasksProjectsLocationsQueuesTasksCancelLeaseRequestr#r CancelLease)rr.rHcancel_lease_requestrer's rrkzAlphaTasks.CancelLeasesc ==;;"<$==KK-A'4468G    ) )' 22rc|jj|||}|jj||j}|jj |S)aConstructs and sends a LeaseTasks request to the Cloud Tasks API. Args: queue_ref: A cloudtasks.projects.locations.queues resource reference lease_duration: string of an integer followed by 's', (e.g. '10s') for the number of seconds for the new leases filter_string: string with an expression to filter which tasks are leased max_tasks: the maximum number of tasks to lease Returns: The response of the LeaseTasks request )filterr^maxTasks)leaseTasksRequestr)rLeaseTasksRequest2CloudtasksProjectsLocationsQueuesTasksLeaseRequestr#rLease)r queue_refrc filter_string max_taskslease_tasks_requestr's rrszAlphaTasks.Leasesm--99NY:P HH1))+ I -     # #G ,,rc|jj|}|jj}|||j}|jj |S)a6Constructs and sends a tasks Acknowledge request to the Cloud Tasks API. Args: task_ref: A cloudtasks.projects.locations.queues.tasks resource reference schedule_time: string formatted as an ISO 8601 datetime with timezone Returns: The response of the tasks Acknowledge request rg)acknowledgeTaskRequestr*)rAcknowledgeTaskRequest8CloudtasksProjectsLocationsQueuesTasksAcknowledgeRequestr#r Acknowledge)rr.rHacknowledge_task_requestrer's rr|zAlphaTasks.Acknowledgesd $}}CC" D $==KK1I'4468G    ) )' 22rrUrT)NN) r r r r rQrGrbrkrsr|rrrrXrXZs5).=A8<."2&3$<@-,3rrXN)r __future__rrrapitools.base.pyrgooglecloudsdk.corer InternalErrorr objectrr:rXrrrrsT$&''*N)A)AN + +F.I.4f3f3r