8dZddlmZddlmZddlmZddlmZddlmZddlm Z ddl m Z ddlm Z dd lmZdd lmZdd lmZdd lmZdd lmZddlZdZdZdZdZdZdZdZdZdZdZ dZ!dZ"dZ#dZ$dZ%Gdde jLZ'Gd d!e jLZ( d>d"Z)d?d#Z*d?d$Z+d?d%Z,d@d&Z-d'Z.d(Z/d)Z0d*Z1d@d+Z2d,Z3d-Z4d.Z5d@d/Z6d0Z7d1Z8d2Z9d3Z:d4Z;d5Zd8Z?d9Z@d:ZAd;ZBd<ZCd=ZDy)Az9A library that is used to support Cloud Pub/Sub commands.)absolute_import)division)unicode_literals) subscriptions)topics) exceptions)util)log) properties) resources)resource_projector)timesNz%Y-%m-%dT%H:%M:%S.%fZzpubsub.projects.topicszpubsub.topics.publishzpubsub.projects.snapshotszpubsub.snapshots.listzpubsub.projects.subscriptionszpubsub.subscriptions.ackzpubsub.subscriptions.listzpubsub.subscriptions.mod_ackzpubsub.subscriptions.mod_configzpubsub.subscriptions.pullzpubsub.subscriptions.seekpubsub.projects.schemasaUsing --push-auth-service-account requires specifying --push-endpoint. This command will continue to run while ignoring --push-auth-service-account, but will fail in a future version. To correct a subscription configuration, run: $ gcloud pubsub subscriptions update SUBSCRIPTION \ --push-endpoint=PUSH_ENDPOINT \ --push-auth-service-account={SERVICE_ACCOUNT_EMAIL} [...] aUsing --push-auth-token-audience requires specifying both --push-endpoint and --push-auth-service-account. This command will continue to run while ignoring --push-auth-token-audience, but will fail in a future version. To correct a subscription configuration, run: $ gcloud pubsub subscriptions update SUBSCRIPTION \ --push-endpoint={PUSH_ENDPOINT} \ --push-auth-service-account={SERVICE_ACCOUNT_EMAIL} \ --push-auth-token-audience={OPTIONAL_AUDIENCE_OVERRIDE} [...] ceZdZdZy)InvalidArgumentErrorz$The user provides invalid arguments.N)__name__ __module__ __qualname____doc__-lib/googlecloudsdk/command_lib/pubsub/util.pyrrJs,rrc"eZdZdZfdZxZS)RequestsFailedErrorz4Indicates that some requests to the API have failed.cltt| dj|dj |y)Nz/Failed to {action} the following: [{requests}].,)actionrequests)superr__init__formatjoin)selfrr __class__s rr zRequestsFailedError.__init__Qs5 t-9@@CHHX$6 A 89r)rrrrr __classcell__)r$s@rrrNs<99rrc|r|S|SNr)original_messagedefault_messages rCreateFailureErrorMessager*Ws. B?Brcht|}tjj|d|itSN projectsIdparams collection) _GetProjectr REGISTRYParseSNAPSHOTS_COLLECTION) snapshot_name project_ids r ParseSnapshotr7]s9:&*    ! !-*6 )C-A " CCrcht|}tjj|d|itSr,)r1r r2r3SUBSCRIPTIONS_COLLECTION)subscription_namer6s rParseSubscriptionr;ds::&*    ! !"3*6 )C-E " GGrcht|}tjj|d|itSr,)r1r r2r3TOPICS_COLLECTION) topic_namer6s r ParseTopicr?ks9:&*    ! !**6 )C-> " @@rcBt|}tj|Sr')r1 projects_util ParseProjectr6s rrBrBrs:&*  # #J //rcr|xs4tjjjj dS)NTrequiredr VALUEScoreprojectGetrCs rr1r1ws-  Hz((--559949HHrcdt|tr|d}n|}t|jS)Nname) isinstancedictr7SelfLink)snapshotrMs rSnapshotUriFuncrR{s/$ F D D t  % % ''rcd}t|tr |d}|d}nt|tr|}n |j}t ||j S)NsubscriptionId projectId)rNrOstrrMr;rP) subscriptionrJrMs rSubscriptionUriFuncrXsV ' d# ( )D;'G,$ D   D 4 ) 2 2 44rcxt|tr|d}n |j}t|j S)NtopicId)rNrOrMr?rP)topicrMs r TopicUriFuncr\s3t  D ::D D  " " $$rch|j}t|dd}t|dd}|8||4tjtj |xsd|xsd|n-|+|)tjt j ||t|r tdy|xstj}d}||jj||}d}t|r)t|d d }|jj| }|jj||| S) z.Parses configs of push subscription from args.SERVICE_ACCOUNT_EMAILNOPTIONAL_AUDIENCE_OVERRIDE PUSH_ENDPOINT)r`r^r_)r^z>argument --push-no-wrapper: --push-endpoint must be specified.)serviceAccountEmailaudiencepush_no_wrapper_write_metadataF) writeMetadata) pushEndpoint oidcToken noWrapper) push_endpointgetattrr warning7PUSH_AUTH_TOKEN_AUDIENCE_MISSING_REQUIRED_FLAGS_WARNINGr!2PUSH_AUTH_SERVICE_ACCOUNT_MISSING_ENDPOINT_WARNING HasNoWrapperrrSubscriptionsClientmessages OidcToken NoWrapper PushConfig)argsclientrhservice_account_emailrb oidc_token no_wrapperwrite_metadatas rParsePushConfigrysd$$-!$(?F T7 >(4<KK?FF':?"7#'&'/ G (]-BKK:AA"7 B  D J    8]668&*&**1H+FJ*$T#CUKN***HJ  # # J* $ NNrct|ddS)Npush_no_wrapperF)ri)rss rrmrms (% 00rcVtj|ttjS)N)fmttzinfo)rFormatDateTimeSEEK_TIME_FORMATUTC)times rFormatSeekTimers   d(8 KKrc2tj|dzS)zFormats a duration argument to be a string with units. Args: duration (int): The duration in seconds. Returns: unicode: The formatted duration. s)six text_type)durations rFormatDurationrs x 3 &&rc|xstj}g}|r\ttj|D];\}}|j |j jj||=|S)a&Parses attribute_dict into a list of AdditionalProperty messages. Args: attribute_dict (Optional[dict]): Dict containing key=value pairs to parse. messages (Optional[module]): Module containing pubsub proto messages. Returns: list: List of AdditionalProperty messages. )keyvalue) rGetMessagesModulesortedr iteritemsappend PubsubMessageAttributesValueAdditionalProperty)attribute_dictro attributesrrs rParseAttributesrsw 3113(*S]]>:; U  0 0 C C D < rcTtj|}|j|d<|d=|S)zCreates a serializable from a Cloud Pub/Sub Topic operation for display. Args: topic: (Cloud Pub/Sub Topic) Topic to be serialized. Returns: A serialized object representing a Cloud Pub/Sub Topic operation (create, delete). rZrM)r MakeSerializablerM)r[topic_display_dicts rTopicDisplayDictrs2*::5A"'**Y rc d}d}|jr.|jjr|jj}d}|j|j|||jt |j |j|jdS)aCreates a serializable from a Cloud Pub/Sub Subscription op for display. Args: subscription: (Cloud Pub/Sub Subscription) Subscription to be serialized. Returns: A serialized object representing a Cloud Pub/Sub Subscription operation (create, delete, update). pullpush)rTr[typereackDeadlineSecondsretainAckedMessagesmessageRetentionDurationenableExactlyOnceDelivery) pushConfigrerMr[rboolrrr)rWrhsubscription_types rSubscriptionDisplayDictrs-++"--::m %))!!#(;;!,"B"BC"."G"G#/#I#I  rcJ|j|j|jdS)aCreates a serializable from a Cloud Pub/Sub Snapshot operation for display. Args: snapshot: (Cloud Pub/Sub Snapshot) Snapshot to be serialized. Returns: A serialized object representing a Cloud Pub/Sub Snapshot operation (create, delete). ) snapshotIdr[ expireTime)rMr[r)rQs rSnapshotDisplayDictrs%MM~~'' rctj|}|jjrdnd|d<t |j }|j |d<|j|d<t|j}|j|d<|S)z3Returns a subscription dict with additional fields.PUSHPULLrrUrTrZ) r rrrer;rMr-subscriptionsIdr?r[topicsId)rWresultsubscription_ref topic_infos rListSubscriptionDisplayDictr/s  . .| <&)44AA6v&.&|'8'89(33&-==& ,,,-* ))& -rctj|}t|j}|j|d<|j|d<|d=|S)Nr[rZrM)r rr?rMr)r[ topic_dict topic_refs rListTopicDisplayDictr;sL!2259*$) *W#,,*Y rctjd|i}t|}|j|d<|j|d<|S)z9Returns a topic_subscription dict with additional fields.rWrUrT)r rr;r-r)topic_subscriptionrrs r ListTopicSubscriptionDisplayDictrDsO  . .)* ,&''9:(33&-==&  -rctj|}t|j}|j|d<|j |d<t |j}|j|d<|j|d<|S)z/Returns a snapshot dict with additional fields.rUrrZr) r rr7rMr- snapshotsIdr?r[rr)rQr snapshot_refrs rListSnapshotDisplayDictrOss  . .x 8&x}}-,$//&%11&()((&!,,& -rcjtjjjj dS)zReturns the value of the core/project config property. Config properties can be overridden with command line flags. If the --project flag was provided, this will return the value provided with the flag. TrErGrrr GetProjectr[s*      ' ' + +T + ::rcntjj|dtidj S)zParses a schema name using configuration properties for fallback. Args: schema: str, the schema's ID, fully-qualified URL, or relative name Returns: str: the relative name of the schema resource r-rr.)r r2r3r RelativeName)schemas rParseSchemaNamerds;    ! !  - " /0<|~>rcBtjjdy)z1Logs a message indicating that a schema is valid.zSchema is valid.Nr statusPrintunused_response unused_argss rOutputSchemaValidatedrss**%&rcBtjjdy)z2Logs a message indicating that a message is valid.zMessage is valid.Nrrs rOutputMessageValidatedrxs**&'rcj|Dcgc]}|d }}|Dcgc] }||vs| }}||fScc}wcc}w)NAckIdr)ack_ids_and_failure_reasonsack_idsackfailed_ack_idsack_idsuccessfully_processed_ack_idss r'ParseExactlyOnceAckIdsAndFailureReasonsr}sX,GH,GSCL,G.H"$"fN&Bf7!$ 7 77 I$s + 00cvtj|}t|jj}|s||Sr')exc HttpExceptionParseExactlyOnceErrorInfopayloaddetails)errorers rHandleExactlyOnceDeliveryErrorrs7 ! 9!)):K:K L % K $$rcg}|D]z}d|vsd|dvrd|vst|dts'|djD]>\}}d|vsd|vstji}||d<||d<|j |@||S)zParses error metadata for exactly once ack/modAck failures. Args: error_metadata: error metadata as dict of format ack_id -> failure_reason. Returns: list: error metadata with only exactly once failures. reason EXACTLY_ONCEmetadataPERMANENT_FAILURETEMPORARY_FAILUREr FailureReason)rNrOitemsr rr)error_metadatarerror_mdrfailure_reasonrs rrrs!# hx>(9K#K!HZ4H$)O"*:"6"<"<">  .3F6D4E#44R8 w"0#**62 #? ! %$r)zInternal Error)rr')Er __future__rrrgooglecloudsdk.api_lib.pubsubrrgooglecloudsdk.api_lib.utilrr#googlecloudsdk.command_lib.projectsr rAgooglecloudsdk.corer r r googlecloudsdk.core.resourcer googlecloudsdk.core.utilrrrr=TOPICS_PUBLISH_COLLECTIONr4SNAPSHOTS_LIST_COLLECTIONr9SUBSCRIPTIONS_ACK_COLLECTIONSUBSCRIPTIONS_LIST_COLLECTION SUBSCRIPTIONS_MOD_ACK_COLLECTION#SUBSCRIPTIONS_MOD_CONFIG_COLLECTIONSUBSCRIPTIONS_PULL_COLLECTIONSUBSCRIPTIONS_SEEK_COLLECTIONSCHEMAS_COLLECTIONrlrkErrorrrr*r7r;r?rBr1rRrXr\ryrmrrrrrrrrrrrrrrrrrrrrrsT @&'709E*#*);* +-323:9 ;#A &G# ; ;.62 ;7-:++-9***9'7C CG@0 I( 5%/Nd1L',  8"  ; >' ( 8%%r