ZdZddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l mZgZGd d ej ZGd d ej ZGddej ZGddej ZGddej ZGddeZddZddZdZGddeZy)z'Utilities for Cloud Pub/Sub Topics API.)absolute_import)division)unicode_literals) list_pager)utils)apis)iam_util) exceptionsceZdZdZy)PublishOperationExceptionz-Error when something went wrong with publish.N__name__ __module__ __qualname____doc__+lib/googlecloudsdk/api_lib/pubsub/topics.pyr r s5rr ceZdZdZy)EmptyMessageExceptionz E 1 = E 1 = E2 = E2 =  E 8t C  E 8t C 'd 2 < $D 0 < +4 7 </d :   44 ? D 1 = D ,D 8 D8t C D7d B }}//04/= 0n ]] 6 6##GG$ 7 !]]77/"T68 . 7*.--*B*B<+C+ ' 06 9*.--*B*B*D' /= @04 0N0N0P- ]] 6 6+#GG$ 7  $ < <A8777CF!=!]] 6 60#GG$ 7 ==''2'3= (j]] 6 6#GG$ 7 #}};;D8/CE < ]] 6 6/#GG$ 7 1 :  rc(|jj|j||}(|r||(_|r,|jj |})|r||)_|)|(_|r<|r:t|j|}*|jj||*| | |(_ |j| | | ||||||||||||||||||| |!|"|#|$|(_ |%r1 tj|jj|%|'|(_|&r|&|(_|j,j/|(S#tj tj"tj$f$r!}+tj&|+f|+_d}+~+wwxYw)aCreates a Topic. Args: topic_ref (Resource): Resource reference to the Topic to create. labels (LabelsValue): Labels for the topic to create. kms_key (str): Full resource name of kms_key to set on Topic or None. message_retention_duration (str): How long to retain messages published to the Topic. message_storage_policy_allowed_regions (list[str]): List of Cloud regions in which messages are allowed to be stored at rest. message_storage_policy_enforce_in_transit (bool): Whether or not to enforce in-transit guarantees for this topic using the allowed regions. schema (Resource): Full resource name of schema used to validate messages published on Topic. message_encoding (str): If a schema is set, the message encoding of incoming messages to be validated against the schema. first_revision_id (str): If a schema is set, the revision id of the oldest revision allowed for validation. last_revision_id (str): If a schema is set, the revision id of the newest revision allowed for validation. kinesis_ingestion_stream_arn (str): The Kinesis data stream ARN to ingest data from. kinesis_ingestion_consumer_arn (str): The Kinesis data streams consumer ARN to use for ingestion. kinesis_ingestion_role_arn (str): AWS role ARN to be used for Federated Identity authentication with Kinesis. kinesis_ingestion_service_account (str): The GCP service account to be used for Federated Identity authentication with Kinesis cloud_storage_ingestion_bucket (str): The Cloud Storage bucket to ingest data from. cloud_storage_ingestion_input_format (str): the format of the data in the Cloud Storage bucket ('text', 'avro', or 'pubsub_avro'). cloud_storage_ingestion_text_delimiter (optional[str]): delimiter to use with text format when partioning the object. cloud_storage_ingestion_minimum_object_create_time (optional[str]): only Cloud Storage objects with a larger or equal creation timestamp will be ingested. cloud_storage_ingestion_match_glob (optional[str]): glob pattern used to match Cloud Storage objects that will be ingested. If unset, all objects will be ingested. azure_event_hubs_ingestion_resource_group (str): The name of the resource group within an Azure subscription. azure_event_hubs_ingestion_namespace (str): The name of the Azure Event Hubs namespace. azure_event_hubs_ingestion_event_hub (str): The name of the Azure event hub. azure_event_hubs_ingestion_client_id (str): The client id of the Azure Event Hubs application used to authenticate Pub/Sub. azure_event_hubs_ingestion_tenant_id (str): The tenant id of the Azure Event Hubs application used to authenticate Pub/Sub. azure_event_hubs_ingestion_subscription_id (str): The id of the Azure Event Hubs subscription. azure_event_hubs_ingestion_service_account (str): The GCP service account to be used for Federated Identity authentication with Azure Event Hubs. aws_msk_ingestion_cluster_arn (str): The ARN that uniquely identifies the MSK cluster. aws_msk_ingestion_topic (str): The name of the MSK topic that Pub/Sub will import from. aws_msk_ingestion_aws_role_arn (str): AWS role ARN to be used for Federated Identity authentication with MSK. aws_msk_ingestion_service_account (str): The GCP service account to be used for Federated Identity authentication with MSK. confluent_cloud_ingestion_bootstrap_server (str): The address of the Confluent Cloud bootstrap server. The format is url:port. confluent_cloud_ingestion_cluster_id (str): The id of the Confluent Cloud cluster. confluent_cloud_ingestion_topic (str): The name of the Confluent Cloud topic that Pub/Sub will import from. confluent_cloud_ingestion_identity_pool_id (str): The id of the identity pool to be used for Federated Identity authentication with Confluent Cloud. confluent_cloud_ingestion_service_account (str): The GCP service account to be used for Federated Identity authentication with Confluent Cloud. ingestion_log_severity (optional[str]): The log severity to use for ingestion. message_transforms_file (str): The file path to the JSON or YAML file containing the message transforms. tags (TagsValue): The tag Keys/Values to be bound to the topic. enable_vertex_ai_smt (bool): Whether or not to enable Vertex AI message transforms. Returns: Topic: The created topic. Raises: InvalidSchemaSettingsException: If an invalid --schema, --message-encoding flag comnbination is specified, or if the --first_revision_id revision is newer than the --last_revision_id specified. )namelabelsmessageRetentionDurationallowedPersistenceRegionsschemaencodingfirstRevisionIdlastRevisionIdrqrrrsrtrurvrwrxryrzr{r|r}r~rrrrrrrrrrrrFN)r7Topic RelativeName kmsKeyNameMessageStoragePolicyenforceInTransitmessageStoragePolicyr:r3schemaSettingsringestionDataSourceSettingsrGetMessageTransformsFromFileMessageTransformmessageTransforms#MessageTransformsInvalidFormatErrorMessageTransformsEmptyFileError!MessageTransformsMissingFileErrorGetErrorMessageargstagsr?Create),r# topic_refrkms_keymessage_retention_duration&message_storage_policy_allowed_regions)message_storage_policy_enforce_in_transitrr8first_revision_idlast_revision_idrqrrrsrtrurvrwrxryrzr{r|r}r~rrrrrrrrrrrrFmessage_transforms_filerenable_vertex_ai_smtr_message_storage_policy encoding_enumes, rrzTopicsClient.CreatesH MM    # # %!;  E  e-#}}AA$J B  3 5 /$:e  "*4==:JKm!]]99 +) :e )-(N(N%A'E#=*K'E-Q/U;m+M2[-Q-Q-Q-Q3]3]&C 7'E*K3]-Q(G3]2[55)O)E%8 "'"D"D MM * * # #  ej ==   &&  3 3  / /  1 1 ''*, s0D))2E<E77E<c|jj|j}|jj |S)zGets a Topic. Args: topic_ref (Resource): Resource reference to the Topic to get. Returns: Topic: The topic. r_)r7PubsubProjectsTopicsGetRequestrr?Get)r#rget_reqs rrzTopicsClient.Gets@mm::$$&;G ==  W %%rc|jj|j}|jj |S)zDeletes a Topic. Args: topic_ref (Resource): Resource reference to the Topic to delete. Returns: Empty: An empty response message. r)r7!PubsubProjectsTopicsDeleteRequestrr?Delete)r#r delete_reqs rrzTopicsClient.DeletesA@@$$&AJ ==   ++rc|jj|j}|jj |S)zDetaches the subscription from its topic. Args: subscription_ref (Resource): Resource reference to the Subscription to detach. Returns: Empty: An empty response message. ) subscription)r7(PubsubProjectsSubscriptionsDetachRequestrrADetach)r#subscription_ref detach_reqs rDetachSubscriptionzTopicsClient.DetachSubscriptionsCGG%224HJ  & & - -j 99rc|jj|j|}tj|j ||ddS)a5Lists Topics for a given project. Args: project_ref (Resource): Resource reference to Project to list Topics from. page_size (int): the number of entries in each batch (affects requests made, but not the yielded results). Returns: A generator of Topics in the Project. )projectpageSizetopicsr batch_sizefieldbatch_size_attribute)r7PubsubProjectsTopicsListRequestrr YieldFromListr?)r# project_ref page_sizelist_reqs rListzTopicsClient.ListsS}}<<((*Y=H  # # '  rc|jj|j|}|jj}t j |||ddS)a7Lists Snapshots for a given topic. Args: topic_ref (Resource): Resource reference to Topic to list snapshots from. page_size (int): the number of entries in each batch (affects requests made, but not the yielded results). Returns: A generator of Snapshots for the Topic. r_r snapshotsrr)r7(PubsubProjectsTopicsSnapshotsListRequestrr-projects_topics_snapshotsrr)r#rrrlist_snaps_services r ListSnapshotszTopicsClient.ListSnapshots s`}}EE$$&FH>>  # #'  rc|jj|j|}|jj}t j |||ddS)aLLists Subscriptions for a given topic. Args: topic_ref (Resource): Resource reference to Topic to list subscriptions from. page_size (int): the number of entries in each batch (affects requests made, but not the yielded results). Returns: A generator of Subscriptions for the Topic.. r subscriptionsrr)r7,PubsubProjectsTopicsSubscriptionsListRequestrr-projects_topics_subscriptionsrr)r#rrrlist_subs_services rListSubscriptionszTopicsClient.ListSubscriptions#s`}}II$$&JH AA  # #'  rc|s |s td|jj||jjj||}|jj |jj |g|j }|jj|}|js td|S)apPublishes a message to the given topic. Args: topic_ref (Resource): Resource reference to Topic to publish to. message_body (bytes): Message to send. attributes (list[AdditionalProperty]): List of attributes to attach to the message. ordering_key (string): The ordering key to associate with this message. Returns: PublishResponse: Response message with message ids from the API. Raises: EmptyMessageException: If neither message nor attributes is specified. PublishOperationException: When something went wrong with the publish operation. zdYou cannot send an empty message. You must specify either a MESSAGE, one or more ATTRIBUTE, or both.)additionalProperties)data attributes orderingKey)r7)publishRequestr_z,Publish operation failed with Unknown error.) rr7 PubsubMessageAttributesValue"PubsubProjectsTopicsPublishRequestPublishRequestrr?Publish messageIdsr )r#r message_bodyr ordering_keymessage publish_reqresults rrzTopicsClient.Publish;s(  ! 5 mm)) ==..>>!+? ! *G--BB}}33gY3G$$&CK]] " "; /F    % 8  Mrc|jj|j|jj|}|jj |S)zSets an IAM policy on a Topic. Args: topic_ref (Resource): Resource reference for topic to set IAM policy on. policy (Policy): The policy to be added to the Topic. Returns: Policy: the policy which was set. )policy)resourcesetIamPolicyRequest)r7'PubsubProjectsTopicsSetIamPolicyRequestrSetIamPolicyRequestr? SetIamPolicy)r#rrrequests rrzTopicsClient.SetIamPolicygsVmmCC'') MM==V=LDG == % %g ..rc|jj|j}|jj |S)zGets the IAM policy for a Topic. Args: topic_ref (Resource): Resource reference for topic to get the IAM policy of. Returns: Policy: the policy for the Topic. )r)r7'PubsubProjectsTopicsGetIamPolicyRequestrr? GetIamPolicy)r#rrs rrzTopicsClient.GetIamPolicywsAmmCC'')DG == % %g ..rc|j|}tj|jj||||j ||S)axAdds an IAM Policy binding to a Topic. Args: topic_ref (Resource): Resource reference for subscription to add IAM policy binding to. member (str): The member to add. role (str): The role to assign to the member. Returns: Policy: the updated policy. Raises: api_exception.HttpException: If either of the requests failed. )rr AddBindingToIamPolicyr7Bindingrr#rmemberrolers rAddIamPolicyBindingz TopicsClient.AddIamPolicyBindingsE  y )F ""4==#8#8&&$O   Y //rcv|j|}tj||||j||S)aRemoves an IAM Policy binding from a Topic. Args: topic_ref (Resource): Resource reference for subscription to remove IAM policy binding from. member (str): The member to remove. role (str): The role to remove the member from. Returns: Policy: the updated policy. Raises: api_exception.HttpException: If either of the requests failed. )rr RemoveBindingFromIamPolicyrr s rRemoveIamPolicyBindingz#TopicsClient.RemoveIamPolicyBindings9  y )F ''=   Y //rc,g},|r|,jtd||r|,jtd||r|,jtd||r|,jtdd|r|,jtddnB|r@|jj|}-|r||-_|,jtd|-| r|,jtddnR| rP| rNt |j| }.|,jtd|jj | |.| | |r|,jtd dnH|j|||||||||||||||||| |!|"|#|$|%|&|'|( }/|/|,jtd |/|)rF |,jtd tj|jj|)|+ |*r|,jtd t |jj#|j% }1g}2|,D]>}3t'|1|3j(|3j*|2j|3j(@|2s t-d|jj/|jj1|1dj3|2|j%}4|j4j7|4S#tjtjtjf$r!}0tj|0f|0_d}0~0wwxYw)aUpdates a Topic. Args: topic_ref (Resource): Resource reference for the topic to be updated. labels (LabelsValue): The Cloud labels for the topic. kms_key_name (str): The full resource name of the Cloud KMS key to associate with the topic, or None. message_retention_duration (str): How long to retain messages. clear_message_retention_duration (bool): If set, remove retention from the topic. recompute_message_storage_policy (bool): True to have the API recalculate the message storage policy. message_storage_policy_allowed_regions (list[str]): List of Cloud regions in which messages are allowed to be stored at rest. message_storage_policy_enforce_in_transit (bool): Whether or not to enforce in-transit guarantees for this topic using the allowed regions. schema (Resource): Full resource name of schema used to validate messages published on Topic. message_encoding (str): If a schema is set, the message encoding of incoming messages to be validated against the schema. first_revision_id (str): If a schema is set, the revision id of the oldest revision allowed for validation. last_revision_id (str): If a schema is set, the revision id of the newest revision allowed for validation. clear_schema_settings (bool): If set, clear schema settings from the topic. clear_ingestion_data_source_settings (bool): If set, clear IngestionDataSourceSettings from the topic. kinesis_ingestion_stream_arn (str): The Kinesis data stream ARN to ingest data from. kinesis_ingestion_consumer_arn (str): The Kinesis data streams consumer ARN to use for ingestion. kinesis_ingestion_role_arn (str): AWS role ARN to be used for Federated Identity authentication with Kinesis. kinesis_ingestion_service_account (str): The GCP service account to be used for Federated Identity authentication with Kinesis cloud_storage_ingestion_bucket (str): The Cloud Storage bucket to ingest data from. cloud_storage_ingestion_input_format (str): the format of the data in the Cloud Storage bucket ('text', 'avro', or 'pubsub_avro'). cloud_storage_ingestion_text_delimiter (optional[str]): delimiter to use with text format when partioning the object. cloud_storage_ingestion_minimum_object_create_time (optional[str]): only Cloud Storage objects with a larger or equal creation timestamp will be ingested. cloud_storage_ingestion_match_glob (optional[str]): glob pattern used to match Cloud Storage objects that will be ingested. If unset, all objects will be ingested. azure_event_hubs_ingestion_resource_group (str): The name of the resource group within an Azure subscription. azure_event_hubs_ingestion_namespace (str): The name of the Azure Event Hubs namespace. azure_event_hubs_ingestion_event_hub (str): The name of the Azure event hub. azure_event_hubs_ingestion_client_id (str): The client id of the Azure Event Hubs application used to authenticate Pub/Sub. azure_event_hubs_ingestion_tenant_id (str): The tenant id of the Azure Event Hubs application used to authenticate Pub/Sub. azure_event_hubs_ingestion_subscription_id (str): The id of the Azure Event Hubs subscription. azure_event_hubs_ingestion_service_account (str): The GCP service account to be used for Federated Identity authentication with Azure Event Hubs. aws_msk_ingestion_cluster_arn (str): The ARN that uniquely identifies the MSK cluster. aws_msk_ingestion_topic (str): The name of the MSK topic that Pub/Sub will import from. aws_msk_ingestion_aws_role_arn (str): AWS role ARN to be used for Federated Identity authentication with MSK. aws_msk_ingestion_service_account (str): The GCP service account to be used for Federated Identity authentication with MSK. confluent_cloud_ingestion_bootstrap_server (str): The address of the Confluent Cloud bootstrap server. The format is url:port. confluent_cloud_ingestion_cluster_id (str): The id of the Confluent Cloud cluster. confluent_cloud_ingestion_topic (str): The name of the Confluent Cloud topic that Pub/Sub will import from. confluent_cloud_ingestion_identity_pool_id (str): The id of the identity pool to be used for Federated Identity authentication with Confluent Cloud. confluent_cloud_ingestion_service_account (str): The GCP service account to be used for Federated Identity authentication with Confluent Cloud. ingestion_log_severity (optional[str]): The log severity to use for ingestion. message_transforms_file (str): The file path to the JSON or YAML file containing the message transforms. clear_message_transforms (bool): If set, clears all message transforms from the topic. enable_vertex_ai_smt (bool): If set, enables Vertex AI message transforms. Returns: Topic: The updated topic. Raises: NoFieldsSpecifiedError: if no fields were specified. PatchConflictingArgumentsError: if conflicting arguments were provided InvalidSchemaSettingsException: If an invalid --schema, --message-encoding flag comnbination is specified, or if the --first_revision_id revision is newer than the --last_revision_id specified. rrrNrrrrrrr)r)rz*Must specify at least one field to update.,)r_ updateMask)updateTopicRequestr)appendrr7rrr:r3rrrrrrrrrCLEAR_MESSAGE_TRANSFORMS_VALUErrsetattrr!r"r PubsubProjectsTopicsPatchRequestUpdateTopicRequestjoinr?Patch)5r#rr kms_key_namer clear_message_retention_duration recompute_message_storage_policyrrrr8rrclear_schema_settings$clear_ingestion_data_source_settingsrqrrrsrtrurvrwrxryrzr{r|r}r~rrrrrrrrrrrrFrclear_message_transformsrupdate_settingsrr new_settingsrr_ update_maskupdate_setting patch_reqs5 rrzTopicsClient.PatchstdO 06BC0|LM! (*D  ( 8$ ?(01GNO /#}}AA$J B  3 5 / 46L M01A4HI $*4==:JKm mm**("3!1 +  , ;T B;;'C)G%?,M)G/S1W=o-O4]/S/S/S/S5_5_(E"9)G,M5_/S*I5_4]!75<l8  !  =| L  #22MM22+)=  $  !#A  MM  Y%;%;%=  >EK) e^..0D0DE223*  "#O PP >>==;;CHH[$9<  # # % ?I ==  y ))=  3 3  / /  1 1 ''*, s0AL2M4MM)NN)NNNNNNNNNNNNNNNNNNNNNNNNNN)&NNNNFNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNF)d)NNN)*NNNFFNFNNNNNFNNNNNNNNNNNNNNNNNNNNNNNNNNNFF)rrrrr$rGrrrrrrrrrrrr rrrrrr<r<Ns;E $(%)!%(,%)+/-19=)-04+/+/+/+/1515$("%)(,15+/&*1504!7]D!%-105 #'%)!%(,%)+/-19=)-04+/+/+/+/1515$("%)(,15+/&*1504!"  QJ'X & , :,.2IM*X/ /0$0*!%',',-105  +0#'%)!%(,%)+/-19=)-04+/+/+/+/1515$("%)(,15+/&*1504!"$ YS*rr<)Fr )r __future__rrrapitools.base.pyrgooglecloudsdk.api_lib.pubsubrgooglecloudsdk.api_lib.utilrgooglecloudsdk.command_lib.iamr googlecloudsdk.corer rErrorr rrrrobjectrr*r.r:r<rrrr/s.&''/,3*!#6 0 06EJ,,ECZ--C4Z%5%547J,<,<7&A o *6o *r