TdZddlZddlmZddlmZddlmZddlmZddl Z GddejZ Gd d ejZ Gd d ejZGd dejZdZddZdZddZ ddZy)zUtilities for Pub/Sub.N)encoding) exceptions)yaml)filesceZdZdZdZdZdZy) ErrorCauselist yaml_or_jsonunrecognized_fieldsmultiple_smts_validateN)__name__ __module__ __qualname__LIST YAML_OR_JSONUNRECOGNIZED_FIELDSMULTIPLE_SMTS_VALIDATE*lib/googlecloudsdk/api_lib/pubsub/utils.pyrrs $,-3rrc"eZdZdZfdZxZS)!MessageTransformsMissingFileErrorz2Error when the message transforms file is missing.c2t||||_yNsuper__init__path)selfmessager __class__s rrz*MessageTransformsMissingFileError.__init__$ GWDIrr rr__doc__r __classcell__r!s@rrr!s:rrc$eZdZdZdfd ZxZS)MessageTransformsEmptyFileErrorz0Error when the message transforms file is empty.c2t||||_yrr)rrr r!s rrz(MessageTransformsEmptyFileError.__init__,r"rr#r&s@rr(r()s8rr(c$eZdZdZdfd ZxZS)#MessageTransformsInvalidFormatErrorz=Error when the message transforms file has an invalid format.c@t||||_||_yr)rrr error_cause)rrr/r r!s rrz,MessageTransformsInvalidFormatError.__init__4s GWDI"Drr*r#r&s@rr-r-1sE##rr-ct|trdj|jSt|trdj|jSt|t r|j tjk(rdj|jS|j tjk(r/dj|jtj|S|j tjk(rdj|jSdj|jtj|St|S)zReturns the formatted error string for an error type. Args: err: Error raised during the GetMessageTransformsFromFile execution. Returns: Formatted error message as a string. z:Message transforms file [{0}] is missing or does not existz#Empty message transforms file [{0}]z>Message transforms file [{0}] not properly formatted as a listzTMessage transforms file [{0}] is not properly formatted in YAML or JSON due to [{1}]zMessage transform file [{0}] contains a list of message transforms instead of a single (1) message transform. Please edit your message-transform-file to contain a single element.zAMessage transforms file [{0}] contains unrecognized fields: [{1}]) isinstancerformatrr(r-r/rrrsix text_typerstr)errs rGetErrorMessager7:s67 G N N  #67 0 7 7 AA#:; *//) J 6#((  J33 3 ""(&3==3E"F J== = AAGhhB  N 6#((CMM#. / s8Orcrtj|}g}|D]E\}}djd|D}|D]#}|jdj ||%G|r*t |t jdj||s(|jrt |t jdyy)z0Validate all parsed message from file are valid..c3FK|]}tj|ywr)r3r4).0es r z2ValidateMessageTransformMessage..ks!Mi%i7KL mmI' -    P .dJ4K4KQ OO PsA=B #C0C  Cct|} tj|}t|tr.t |dk(r|d}nt |tjtj||}t||||S#ttttjf$r }t |tj |d}~wwxYw)a>Reads a YAML or JSON object of type message from local path. Parses a single message transform. Args: message: The message type to be parsed from the file. path: A local path to an object specification in YAML or JSON format. enable_vertex_ai_smt: Whether or not to enable Vertex AI message transforms. Returns: Object of type message, if successful. Raises: MessageTransformsMissingFileError: If file is missing. MessageTransformsEmptyFileError: If file is empty. MessageTransformsInvalidFormat: If file's format is invalid. rN)rOrrQr1r lenr-rrrrRrJrSrTrUrVr)r rrCrNmessage_transformrXr<s r(GetMessageTransformFromFileForValidationr`s*d #(P (+#T*  1 $-a01 *33   & &w0A BF#FD2FG -    P .dJ4K4KQ OO PsA6B#C(CC)F)r$enumapitools.base.pyrgooglecloudsdk.corerrgooglecloudsdk.core.utilrr3EnumrErrorrr(r-r7rJrOr[r`rrrrgs %*$* 44 (8(8j&6&6#**:*:#)X00)Z).+r