7dZddlmZddlmZddlmZddlZddlmZddlm Z ddlm Z dd l m Z dd l mZdd l mZGd d ej ZGddej ZGddej ZGddej ZGddej ZGddej ZGddej ZdZGdde j2Zy)z(Utilities for Eventarc KafkaSources API.)absolute_import)division)unicode_literalsN) list_pager)base)common)apis) exceptions) resourcesceZdZdZy)NoFieldsSpecifiedErrorz:Error when no fields were specified for a Patch operation.N__name__ __module__ __qualname____doc__4lib/googlecloudsdk/api_lib/eventarc/kafka_sources.pyr r sBrr ceZdZdZy)!InvalidNetworkConfigArgumentErrorzJError when the Kafka Source's network configuration arguments are invalid.Nrrrrrr#sRrrceZdZdZy)InvalidDestinationArgumentErrorz>Error when the Kafka Source's destination argument is invalid.Nrrrrrr'sFrrceZdZdZy)(InvalidAuthenticationMethodArgumentErrorzCError when the Kafka Source's authentication arguments are invalid.Nrrrrrr+sKrrceZdZdZy)InvalidBrokerUrisArgumentErrorzDError when the Kafka Source's bootstrap_servers argument is invalid.Nrrrrrr/sLrrceZdZdZy)InvalidTopicsArgumentErrorz9Error when the Kafka Source's topics argument is invalid.Nrrrrrr3sArrceZdZdZy)!InvalidInitialOffsetArgumentErrorzAError when the Kafka Source's initial offset argument is invalid.Nrrrrr!r!7sIrr!cxtjj|jd}|j S)Nz(eventarc.projects.locations.kafkaSources) collection)r REGISTRYParseRelativeNamenameSelfLink)resource kafka_sourcess rGetKafkaSourceURIr*;s8$$66mm J7-    !!rceZdZdZfdZddZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZxZS)KafkaSourceClientV1zJKafka Source Client for interaction with v1 of Eventarc Kafka Sources API.ctt| tjtj dt jtjtj }|j|_ |j|_ y)N kafkaSource) superr,__init__rAPI_NAME API_VERSION_1r GetClientInstanceMESSAGES_MODULE _messagesprojects_locations_kafkaSources_service)selfclient __class__s rr0zKafkaSourceClientV1.__init__Es\ t---}  # #FOOV5I5I JF++DN::DMrc|jj|jj||j |}|j j |S)aCreates a Kafka Source. Args: kafka_source_ref: Resource, the Kafka Source to create. kafka_source_message: Kafka Source, the Kafka Source message that holds Kafka source's name, destinations, mediations, input payload format, logging config, retry policy, crypto key name, etc. dry_run: If set, the changes will not be committed, only validated Returns: A long-running operation for create. )parentr. kafkaSourceId validateOnly)r52EventarcProjectsLocationsKafkaSourcesCreateRequestParent RelativeNameNamer7Create)r8kafka_source_refkafka_source_messagedry_run create_reqs rrCzKafkaSourceClientV1.CreatePs_ II#**,99;,*//1 J  ==   ++rc|jj|j|}tj|j |d||dS)ayList available Kafka Sources in location. Args: location_ref: Resource, the location to list Kafka Sources in. limit: int or None, the total number of results to return. page_size: int, the number of entries in each batch (affects requests made, but not the yielded results). Returns: A generator of Kafka Sources in the location. )r<pageSize kafkaSourcesrI)servicerequestfieldlimit batch_sizebatch_size_attribute)r50EventarcProjectsLocationsKafkaSourcesListRequestrAr YieldFromListr7)r8 location_refrN page_sizelist_reqs rListzKafkaSourceClientV1.ListgsW~~NN((*YOH  # # '  rc|jj|j}|jj |S)zGets the requested Kafka Source. Args: kafka_source_ref: Resource, the Kafka Source to get. Returns: The Kafka Source message. r&)r5/EventarcProjectsLocationsKafkaSourcesGetRequestrAr7Get)r8rDget_reqs rrZzKafkaSourceClientV1.GetsAnnLL  * * ,MG ==  W %%rc|jj|j||}|jj |S)aUpdates the specified Kafka Source. Args: kafka_source_ref: Resource, the Kafka Source to update. kafka_source_message: Kafka Source, the Kafka Source message that holds Kafka Source's name, destinations, mediations, input payload format, logging config, retry policy, crypto key name, etc. update_mask: str, a comma-separated list of Kafka Source fields to update. Returns: A long-running operation for update. )r& kafka_source updateMask)r51EventarcProjectsLocationsKafkaSourcesPatchRequestrAr7Patch)r8rDrE update_mask patch_reqs rr`zKafkaSourceClientV1.PatchsJ HH!..0-" I  ==  y ))rc|jj|j}|jj |S)zDeletes the specified Kafka Source. Args: kafka_source_ref: Resource, the Kafka Source to delete. Returns: A long-running operation for delete. rX)r52EventarcProjectsLocationsKafkaSourcesDeleteRequestrAr7Delete)r8rD delete_reqs rrezKafkaSourceClientV1.DeletesD II!..0 J  ==   ++rc d|jj|j|j||j ||j ||j ||||| |j|| |j|| |j| |j| | S)N) r& brokerUrisconsumerGroupIdtopicsauthenticationConfig networkConfig destination initialOffset loggingConfiglabels) r5 KafkaSourcerA_BuildBrokerUris_BuildConsumerGroupID _BuildTopics_BuildAuthenticationConfig_BuildNetworkConfig_BuildDestination_BuildInitialOffset_BuildLoggingConfig)r8rDbootstrap_serversconsumer_group_idrjsasl_mechanism sasl_username sasl_passwordtls_client_certificatetls_client_keynetwork_attachment message_businitial_offsetlogging_configrps rBuildKafkaSourcez$KafkaSourceClientV1.BuildKafkaSources" >> % %  * * ,(():;223DE  (!<<    "   .. 0 **+;[I..~>..~>% & rc dg} |r| jd|s|s|r| jd|r| jd|r| jd|s|s|s| r| jd| s| r| jd| r| jd| s tdd j| S) acBuilds an update mask for updating a Kafka Source. Args: destinations: bool, whether to update the destinations. input_payload_format_json: bool, whether to update the input_payload_format_json. input_payload_format_avro_schema_definition: bool, whether to update the input_payload_format_avro_schema_definition. input_payload_format_protobuf_schema_definition: bool, whether to update the input_payload_format_protobuf_schema_definition. mediations: bool, whether to update the mediations. logging_config: bool, whether to update the logging_config. max_retry_attempts: bool, whether to update the max_retry_attempts. min_retry_delay: bool, whether to update the min_retry_delay. max_retry_delay: bool, whether to update the max_retry_delay. crypto_key: bool, whether to update the crypto_key. clear_crypto_key: bool, whether to clear the crypto_key. labels: bool, whether to update the labels. Returns: The update mask as a string. Raises: NoFieldsSpecifiedError: No fields are being updated. destinationsinputPayloadFormat mediationsro retryPolicy cryptoKeyNamerpz*Must specify at least one field to update.,)appendr join)r8rinput_payload_format_json+input_payload_format_avro_schema_definition/input_payload_format_protobuf_schema_definitionrrmax_retry_attemptsmin_retry_delaymax_retry_delay crypto_keyclear_crypto_keyrpras rBuildUpdateMaskz#KafkaSourceClientV1.BuildUpdateMasksRK(! 6 :-.&)   '%) "  "#O PP 88K  rcB|jjjS)N)r5rq LabelsValue)r8s rLabelsValueClassz$KafkaSourceClientV1.LabelsValueClasss >> % % 1 11rc2| tdt|S)Nz+Must specify at least one bootstrap server.)rlist)r8rzs rrrz$KafkaSourceClientV1._BuildBrokerUris"s& * 7  ! ""rc2| tdt|S)Nz Must specify at least one topic.)rr)r8rjs rrtz KafkaSourceClientV1._BuildTopics)s ~ &'I JJ <rcZ| tdd|jd|jd|S)Nzmessage_bus must be set projects/z /locations/z/messageBuses/)r projectsId locationsId)r8rDrs rrwz%KafkaSourceClientV1._BuildDestination.s@ +,E FF '223;?O?[?[>\\jkvjw xxrc| td|jjd|jd|jd|S)Nznetwork_attachment must be setrz /regions/z/networkAttachments/)networkAttachment)rr5 NetworkConfigrr)r8rDrs rrvz'KafkaSourceClientV1._BuildNetworkConfig3sd! -.N OO >> ' '%&6&A&A%B)L\LhLhKii}Q~RS ( rc8|dtjS|S)Nz eventarc-)uuiduuid4)r8r{s rrsz)KafkaSourceClientV1._BuildConsumerGroupID:s!  '' rc6|y|dk7r|dk7r td|S)Nnewestoldestz.initial_offset must be one of newest or oldest)r!)r8rs rrxz'KafkaSourceClientV1._BuildInitialOffset?s2 !n&@ - :  rc |du|duz|duz}|du|duz}|dkDr|dkDr td|dkDrX|dk7r td|jj|jj|j |||dS|dkDrb|dk7r td|jjd|jj |jj ||  Std) NrzExactly one of the following authentication methods must be set: - SASL Authentication (--sasl-mechanism, --sasl-username, --sasl-password) - TLS Authentication (--tls-client-certificate, --tls-client-key)zpWhen using SASL Authentication, all three arguments sasl_mechanism, sasl_username, and sasl_password must be set) mechanismusernameSecretpasswordSecret)saslAuth mutualTlsAuthzYWhen using TLS Authentication, both tls_client_certificate and tls_client_key must be set)clientCertificate clientKey)secretManagerResources)rr5AuthenticationConfigSaslAuthConfig_ConvertSaslMechanismToEnumMutualTlsAuthConfigMutualTlsSecrets)r8r|r}r~rr num_args_sasl num_args_mtlss rruz.KafkaSourceClientV1._BuildAuthenticationConfigHsh t #  $ &  $ & ,47d"Mq]Q. 4  q ! 6 L  ^^ 0 0>>0088H**1  1q ! 6 *  ^^ 0 0::%)^^%D%D$:*&E&; 1 3  rc|y|jj|jjj|S)N) logSeverity)r5 LoggingConfigLogSeverityValueValuesEnum)r8rs rryz'KafkaSourceClientV1._BuildLoggingConfigsB  >> ' 'NN00KK   ( rc|dk(r%|jjjdS|dk(r%|jjjdS|dk(r%|jjjdStd)z)Convert human-readable mechanism to enum.PLAINz SCRAM-SHA-256SHA_256z SCRAM-SHA-512SHA_512zDsasl_mechanism must be one of PLAIN, SCRAM_SHA_256, or SCRAM_SHA_512)r5rMechanismValueValuesEnumr)r8rs rrz/KafkaSourceClientV1._ConvertSaslMechanismToEnums}G ^^ * * C CG LLO# ^^ * * C CI NNO# ^^ * * C CI NN 2  r)F)rrrrr0rCrVrZr`rerrrrrrtrwrvrsrxruryr __classcell__)r:s@rr,r,BseR ;,.0 &*,, $LD!L2# y  ;z rr,)r __future__rrrrapitools.base.pyrgooglecloudsdk.api_lib.eventarcrrgooglecloudsdk.api_lib.utilr googlecloudsdk.corer r Errorr rrrrrr!r*EventarcClientBaser,rrrrs/&' '02,*)CZ--CS (8(8SGj&6&6GLz/?/?LMZ%5%5MB!1!1BJ (8(8J"W$11Wr