gdZddlmZddlmZddlmZddlmZddlm Z ddl mZ ddl m Z ddl mZddlmZdd lmZdd lmZdd lZd Zd ZddZe j,e j.j0Gdde j2Ze j,e j.j6e j.j8GddeZy )z(Cloud Pub/Sub subscription pull command.)absolute_import)division)unicode_literals) exceptions) subscriptions)base)flags) resource_args)utilNztable[box]( message.data.decode(base64).decode(utf-8), message.messageId, message.orderingKey, message.attributes.list(separator=' '), deliveryAttempt, ackId.if(NOT auto_ack) ) ztable[box]( message.data.decode(base64).decode(utf-8), message.messageId, message.orderingKey, message.attributes.list(separator=' '), deliveryAttempt, ackId.if(NOT auto_ack), ackStatus.if(auto_ack) ) ctj}|jjj }|j |||}i}g}|j rV|jrJ|jDcgc]}|j} } |j| ||s|D] } | d|| d<|j s |jSg}|jD]x}i}|j+D]*}t-||j.}|s|||j.<,|j|vr||j|d<nd|d<|j1|z|Scc}w#tj$rw} tj| } tj| j j"}|s,| D]'} dt%j&| j(z|| <)Yd} ~ Ad} ~ wwxYw)z#Pulls messages from a subscription.FAILURE_N FailureReasonAckId ackStatusSUCCESS)rSubscriptionsClientCONCEPTS subscriptionParsePullauto_ackreceivedMessagesackIdAckapi_ex HttpErrorutil_ex HttpExceptionr ParseExactlyOnceErrorInfopayloaddetailssix text_type status_code all_fieldsgetattrnameappend)args max_messagesreturn_immediatelyclientsubscription_ref pull_responsefailed_ack_idsack_ids_and_failure_reasonsmessageack_idserrorexcack_idack_ids_and_failure_reason return_val message_copyfieldvalues (lib/surface/pubsub/subscriptions/pull.py_Runr<8s  , , .&]]//557++. 02-. " ]]}55,9,J,JK,Jw}},JGK Q jj*+ (C $2?C 1  )D   ) ))*//gL##%guzz*e #( UZZ &}}&"0"?l;"+l;l#0 IL    Q  ! !% (c$($B$B ++  %! )F#- e>O>O0P#P.  Qs6E EG$-A,GG$c.eZdZdZddiZedZdZy)r=Pulls one or more Cloud Pub/Sub messages from a subscription. DESCRIPTIONa  Returns one or more messages from the specified Cloud Pub/Sub subscription, if there are any messages enqueued. By default, this command returns only one message from the subscription. Use the `--limit` flag to specify the max messages to return. Please note that this command is not guaranteed to return all the messages in your backlog or the maximum specified in the --limit argument. Receiving fewer messages than available occasionally is normal.c|jjttj|dt j |tjj|dy)Nto pull messages from.) display_info AddFormatMESSAGE_FORMAT_WITH_ACK_STATUSr AddSubscriptionResourceArgr AddPullFlagsr LIMIT_FLAG SetDefaultparsers r;Argsz Pull.Args~sJ !!"@A,,V5MN vOOvq)c.t||jS)N)r<limit)selfr)s r;RunzPull.Runs djj !!rMN)__name__ __module__ __qualname____doc__ detailed_help staticmethodrLrQrMr;rrjs/E - **"rMrc&eZdZdZedZdZy)PullBetar>c|jjttj|dt j |dddy)NrAT)add_deprecatedadd_waitadd_return_immediately)rCrDrEr rFr rGrJs r;rLz PullBeta.Argss? !!"@A,,V5MN td4rMc:|jdr4|jdrtjdd|j}n |j}d}|jdr |j }n|jdr |j }t|||S)NrOr*z--max-messagesz--limitFr+wait) IsSpecifiedrConflictingArgumentsExceptionrOr*r+r`r<)rPr)r*r+s r;rQz PullBeta.Runs   . )667G7@B BZZl&&l ,-22  & !#yy= l$6 77rMN)rRrSrTrUrWrLrQrXrMr;rZrZsE8rMrZ)F)rU __future__rrrapitools.base.pyrrgooglecloudsdk.api_lib.pubsubrgooglecloudsdk.api_lib.utilrgooglecloudsdk.callioper!googlecloudsdk.command_lib.pubsubr r r r"MESSAGE_FORMATrEr< ReleaseTracks ReleaseTrackGA ListCommandrBETAALPHArZrXrMr;rps/&'17=(.3;2   "/dD%%(()"4  "*">D%%**D,=,=,C,CD8t8E8rM