&dZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd l m Z dd l mZdd l mZGd d ej ZdZGddej&Zy)z(Utilities for Eventarc MessageBuses API.)absolute_import)division)unicode_literals) list_pager)base)common)common_publishing)apis) exceptions) resourcesceZdZdZy)NoFieldsSpecifiedErrorz:Error when no fields were specified for a Patch operation.N)__name__ __module__ __qualname____doc__4lib/googlecloudsdk/api_lib/eventarc/message_buses.pyrrsBrrcxtjj|jd}|j S)Nz(eventarc.projects.locations.messageBuses) collection)r REGISTRYParseRelativeNamenameSelfLink)resource message_busess rGetMessageBusURIr"s8$$66mm J7-    !!rc|eZdZdZfdZddZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZedZxZS)MessageBusClientV1zGMessageBus Client for interaction with v1 of Eventarc MessageBuses API.ctt| tjtj dt jtjtj }|j|_ |j|_ t jtjtj }|j|_ |j|_y)Nz message bus)superr __init__rAPI_NAME API_VERSION_1r GetClientInstanceMESSAGES_MODULE _messagesprojects_locations_messageBuses_servicer _publishing_messages_publishing_service)selfclientpublishing_client __class__s rr#zMessageBusClientV1.__init__,s d,--}  # #FOOV5I5I JF++DN::DM..""$5$C$C!2 A AD0PPDrc|jj|jj||j |}|j j |S)a_Creates a new MessageBus. Args: message_bus_ref: Resource, the MessageBus to create. message_bus_message: MessageBus, the messageBus message that holds messageBus' name, crypto key name, etc. dry_run: If set, the changes will not be committed, only validated Returns: A long-running operation for create. )parent messageBus messageBusId validateOnly)r(2EventarcProjectsLocationsMessageBusesCreateRequestParent RelativeNameNamer*Create)r-message_bus_refmessage_bus_messagedry_run create_reqs rr:zMessageBusClientV1.Create?s_ II"))+88:*(--/ J  ==   ++rc|jj|j}|jj |S)zGets the requested MessageBus. Args: message_bus_ref: Resource, the MessageBus to get. Returns: The MessageBus message. r)r(/EventarcProjectsLocationsMessageBusesGetRequestr8r*Get)r-r;get_reqs rrBzMessageBusClientV1.GetUsAnnLL  ) ) +MG ==  W %%rc|jj|j|}tj|j |d||dS)avList available messageBuses in location. Args: location_ref: Resource, the location to list MessageBuses 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 MessageBuses in the location. r2pageSize messageBusesrF)servicerequestfieldlimit batch_sizebatch_size_attribute)r(0EventarcProjectsLocationsMessageBusesListRequestr8r YieldFromListr*)r- location_refrK page_sizelist_reqs rListzMessageBusClientV1.ListcsW~~NN((*YOH  # # '  rc|jj|j||}|jj |S)amUpdates the specified MessageBus. Args: message_bus_ref: Resource, the MessageBus to update. message_bus_message: MessageBus, the messageBus message that holds messageBus' name, crypto key name, etc. update_mask: str, a comma-separated list of MessageBus fields to update. Returns: A long-running operation for update. )rr3 updateMask)r(1EventarcProjectsLocationsMessageBusesPatchRequestr8r*Patch)r-r;r< update_mask patch_reqs rrWzMessageBusClientV1.Patch{sJ HH --/*" I  ==  y ))rc|jj|j}|jj |S)zDeletes the specified MessageBus. Args: message_bus_ref: Resource, the MessageBus to delete. Returns: A long-running operation for delete. r@)r(2EventarcProjectsLocationsMessageBusesDeleteRequestr8r*Delete)r-r; delete_reqs rr\zMessageBusClientV1.DeletesD II --/ J  ==   ++rc |jj|j|jj|j |||||||} |j j | y)aPublish a Cloud Event to a MessageBus. Args: message_bus_ref: Resource, the message bus to publish to. json_message: str, the json string to publish. avro_message: byte, the avro payload to publish. event_id: str, the id of the event. event_type: str, the type of the event. event_source: str, the source of the event. event_data: str, the data of the event. event_attributes: dict, the attributes of the event. ) protoMessage avroMessage jsonMessage)r3-googleCloudEventarcPublishingV1PublishRequestN)r+=EventarcpublishingProjectsLocationsMessageBusesPublishRequestr8-GoogleCloudEventarcPublishingV1PublishRequest_BuildCloudEventProtoMessager,Publish) r-r; json_message avro_messageevent_id event_type event_source event_dataevent_attributes publish_reqs rrfzMessageBusClientV1.Publishs~0++ii"//16:6O6O6}6}::*lJ@P%$ 7~7 j K $$[1rc |jj|j|}tj|j d|d||dS)z@List available enrollments attached to the specified messageBus.rEListEnrollments enrollmentsrF)rHmethodrIrJrKrLrM)r(;EventarcProjectsLocationsMessageBusesListEnrollmentsRequestr8rrOr*)r-r;rKrQrRs rrpz"MessageBusClientV1.ListEnrollmentssX~~YY++- ZH  # #  ' rcd}|?|jj|jjj|}|jj|j |||S)N) logSeverity)r loggingConfig cryptoKeyNamelabels)r( LoggingConfigLogSeverityValueValuesEnum MessageBusr8)r-r;logging_configcrypto_key_namerxlogging_config_enums rBuildMessageBusz"MessageBusClientV1.BuildMessageBussx! NN88nn22MM 9 >> $ $  ) ) +)% % rcg}|r|jd|s|r|jd|r|jd|s tddj|S)aBuilds an update mask for updating a MessageBus. Args: logging_config: bool, whether to update the logging config. 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. rvrwrxz*Must specify at least one field to update.,)appendrjoin)r-r| crypto_keyclear_crypto_keyrxrXs rBuildUpdateMaskz"MessageBusClientV1.BuildUpdateMasks]$K)%) "  "#O PP 88K  rcB|jjjS)zReturns the labels value class.)r(r{ LabelsValuer-s rLabelsValueClassz#MessageBusClientV1.LabelsValueClasss >> $ $ 0 00rc t||||y|jj|||d||j|S)Nz1.0)idtypesource specVersiontextData attributes)r+)GoogleCloudEventarcPublishingV1CloudEvent_BuildCloudEventAttributes)r-rirjrkrlrms rrez/MessageBusClientV1._BuildCloudEventProtoMessages\          $ $ N N  223CD O rc 4|y|jjj|jDcgc]P\}}|jjjj ||jj |Rc}}Scc}}w)N)ceString)keyvalue)additionalProperties)r+rAttributesValueitemsAdditionalPropertyAGoogleCloudEventarcPublishingV1CloudEventCloudEventAttributeValue)r-rmrrs rrz-MessageBusClientV1._BuildCloudEventAttributess   $ $ N N ^ ^/446 7 U  % % O O _ _ r r//qq"r s  7  _   sAB cy)Nz message-busesrrs r_resource_label_pluralz)MessageBusClientV1._resource_label_plural*s r)F)rrrrr#r:rBrSrWr\rfrprrrrerpropertyr __classcell__)r0s@rr r )s]OQ&,, &0**, $2L "!<1&   rr N)r __future__rrrapitools.base.pyrgooglecloudsdk.api_lib.eventarcrrr googlecloudsdk.api_lib.utilr googlecloudsdk.corer r ErrorrrEventarcClientBaser rrrrsU/&''02=,*)CZ--C"C00Cr