dZddlmZddlmZddlmZGddejZGddejZ dd Z dd Z Gd d e Z y )z3Utilities for Cloud Pub/Sub Message Transforms API.)utils)apis) exceptionsceZdZdZy)EmptyMessageExceptionz9Error when no message was specified for a Test operation.N__name__ __module__ __qualname____doc__7lib/googlecloudsdk/api_lib/pubsub/message_transforms.pyrrsArrceZdZdZy)EmptyFilePathExceptionzMError when no message transforms file was specified for a Validate operation.Nrr rrrrsUrrc2tjdd|S)Npubsubv1no_http)rGetClientInstancers rrrs  $ @@rNc6|xs t}|jSN)rMESSAGES_MODULE)clients rGetMessagesModuler"s  (&(&   rc2eZdZdZddZddZ ddZy) MessageTransformsClientz?Client for message transforms service in the Cloud Pub/Sub API.Nc|xs t|_|xs t||_|jj|_yr)rrrmessagesprojects_service)selfrr s r__init__z MessageTransformsClient.__init__*s5/-/DK9 1& 9DMKK((DMrc|s td tj|jj|}|jj|j|jj|}|jj|S#tj tj tjf$r!}tj|f|_ d}~wwxYw)aValidates a message transform. Args: project_ref (Resource): Resource reference for the project. message_transform_file (str): The file path to the JSON or YAML file containing the message transform. Returns: ValidateMessageTransformResponse (success) if the message transform is valid, otherwise an error. Raises: EmptyFilePathException: If no message transform file was specified. z]You need to specify a path to JSON or YAML file containing the message transform to validate.N)messageTransform)projectvalidateMessageTransformRequest)rr(GetMessageTransformFromFileForValidationr MessageTransform#MessageTransformsInvalidFormatErrorMessageTransformsEmptyFileError!MessageTransformsMissingFileErrorGetErrorMessageargs-PubsubProjectsValidateMessageTransformRequest RelativeNameValidateMessageTransformRequestr"ValidateMessageTransform)r# project_refmessage_transform_filemessage_transformevalidate_requests rValidatez MessageTransformsClient.Validate/s " " ,   HH -- ( (*@}}RR((*(, (U(U.)V) S == 1 12B CC 11 -- //  %%a(*af  s*B2C, C''C,c |s |s td|jj||jjj|}d}|r- t j |jj |d}|r|jj|nd} |jj|j|jj|| |r|jnd|r|jnd } |j j#| S#tjtjtjf$r!} t j| f| _ d} ~ wwxYw) aTests applying message transforms to a message. Args: project_ref (Resource): Resource reference for the project. message_body (bytes): The message to test. attributes (list[AdditionalProperty]): List of attributes to attach to the message. message_transforms_file (str): The file path to the JSON or YAML file containing the message transforms. topic_ref (Resource): The topic containing the message transforms to test against. subscription_ref (Resource): The subscription containing the message transforms to test against. Returns: TestMessageTransformsResponse which contains a list of TransformedMessage. Raises: EmptyMessageException: If no message body or attributes were specified. EmptyMessageTransformsException: If no message transforms file/topic/subscription were specified. zdYou cannot send an empty message. You must specify either a MESSAGE, one or more ATTRIBUTE, or both.)additionalProperties)data attributesNF)enable_vertex_ai_smt)messageTransforms)messager?topic subscription)r'testMessageTransformsRequest)rr PubsubMessageAttributesValuerGetMessageTransformsFromFiler*r+r,r-r.r/MessageTransforms*PubsubProjectsTestMessageTransformsRequestr1TestMessageTransformsRequestr"TestMessageTransforms) r#r4 message_bodyr=message_transforms_file topic_refsubscription_refr@message_transformsr7message_transforms_msg test_requests rTestzMessageTransformsClient.TestWs|>  ! 5  mm)) ==..>>!+? *G  "?? MM * * #!&   '':L'M  ==KK((*%)]]%O%O4.7)((*T*668 &P& L L == . .| <<1  3 3  / /  1 1 ''*, s,D))2E<E77E<)NNr)NNNNN)r r r r r$r9rRr rrrr's(G) &DV"L=rr)Fr)r googlecloudsdk.api_lib.pubsubrgooglecloudsdk.api_lib.utilrgooglecloudsdk.corerErrorrrrrobjectrr rrrXsT:/,*BJ,,BVZ--VA |=f|=r