o :DiI@sdZdZddlZddlmZddlZddlZddlmZddl Z ddl m Z ddl m Z ddl mZdd l mZee jje jje jje jjgZee jje jjgZee jje jjgZd Zd Zd Ze d Ze dZ Gddde!Z"Gddde"Z#Gddde"Z$        d@ddZ%     dAddZ&ddZ'Gddde(Z)d d!Z*d"d#Z+d$d%Z,   &dBd'd(Z-   &dBd)d*Z.e/e0fZ1Gd+d,d,e(Z2dCd-d.Z3d/d0Z4d1d2Z5d3d4Z6d5d6gd7d8gd7d8gd9d:gd;dgd?Z7dS)DatContains routines for printing protocol messages in JSON format. Simple usage example: # Create a proto object and serialize it to a json format string. message = my_proto_pb2.MyMessage(foo='bar') json_string = json_format.MessageToJson(message) # Parse a json format string to proto object. message = json_format.Parse(json_string, my_proto_pb2.MyMessage()) zjieluo@google.com (Jie Luo)N) OrderedDict) methodcaller) type_checkers) descriptor)message_factory)symbol_databaseInfinityz -InfinityNaNu.[-](?![-])|(?rrFTc Cs"t|||||d} | ||||S)aConverts protobuf message to JSON format. Args: message: The protocol buffers message instance to serialize. including_default_value_fields: If True, singular primitive fields, repeated fields, and map fields will always be serialized. If False, only serialize non-empty fields. Singular message fields and oneof fields are not affected by this option. preserving_proto_field_name: If True, use the original proto field names as defined in the .proto file. If False, convert the field names to lowerCamelCase. indent: The JSON object will be pretty-printed with this indent level. An indent level of 0 or negative will only insert newlines. If the indent level is None, no newlines will be inserted. sort_keys: If True, then the output will be sorted by field names. use_integers_for_enums: If true, print integers instead of enum names. descriptor_pool: A Descriptor Pool for resolving types. If None use the default. float_precision: If set, use this to specify float field valid digits. ensure_ascii: If True, strings with non-ASCII characters are escaped. If False, Unicode strings are returned unchanged. Returns: A string containing the JSON formatted protocol buffer message. float_precision)_Printer ToJsonString) messageincluding_default_value_fieldspreserving_proto_field_nameindent sort_keysuse_integers_for_enumsdescriptor_poolr ensure_asciiprinterrrr MessageToJsonBs#r$cCst|||||d}||S)aConverts protobuf message to a dictionary. When the dictionary is encoded to JSON, it conforms to proto3 JSON spec. Args: message: The protocol buffers message instance to serialize. including_default_value_fields: If True, singular primitive fields, repeated fields, and map fields will always be serialized. If False, only serialize non-empty fields. Singular message fields and oneof fields are not affected by this option. preserving_proto_field_name: If True, use the original proto field names as defined in the .proto file. If False, convert the field names to lowerCamelCase. use_integers_for_enums: If true, print integers instead of enum names. descriptor_pool: A Descriptor Pool for resolving types. If None use the default. float_precision: If set, use this to specify float field valid digits. Returns: A dict representation of the protocol buffer message. r)r_MessageToJsonObject)rrrr r!rr#rrr MessageToDictns r&cCs"|jtjjko|jjo|jjSN)typerFieldDescriptor TYPE_MESSAGE message_type has_options GetOptions map_entry)fieldrrr _IsMapEntrys  r0c@steZdZdZ     dddZddZdd Zd d Zd d ZddZ ddZ ddZ ddZ ddZ ddZdS)rz)JSON format printer for protocol message.FNcCs6||_||_||_||_|rd||_dSd|_dS)Nz.{}g)rrr r!format float_format)selfrrr r!rrrr__init__s z_Printer.__init__cCs||}tj||||dS)N)rrr")r%jsondumps)r3rrrr"jsrrrrs z_Printer.ToJsonStringcCsL|j}|j}t|r||S|tvrtt|d||Si}|||S)zEConverts message to an object according to Proto3 JSON Specification.r) DESCRIPTOR full_name_IsWrapperMessage_WrapperMessageToJsonObject_WKTJSONMETHODSr_RegularMessageToJsonObject)r3rmessage_descriptorr9r7rrrr%s  z_Printer._MessageToJsonObjectc s|}z|D]l\}jrj}nj}trFjjd}i}|D]}t|tr2|r/d} nd} nt |}  ||||| <q#|||<qj t j jkrZfdd|D||<qjrkdj} |||<q |||<qjr|j} | jD]Dj t j jkrjt j jksjrq}jrj}nj}||vrq}tri||<q}j t j jkrg||<q} j||<q}W|SW|Sty} z tdj| | d} ~ ww) z?Converts normal message according to Proto3 JSON Specification.valuetruefalsecsg|]}|qSr)_FieldToJsonObject).0kr/r3rr sz8_Printer._RegularMessageToJsonObject..z[%s]z#Failed to serialize {0} field: {1}.N) ListFieldsrname json_namer0r+fields_by_name isinstanceboolstrrBlabelrr)LABEL_REPEATED is_extensionr9rr8fieldscpp_typeCPPTYPE_MESSAGEcontaining_oneof default_value ValueErrorrr1) r3rr7rQr?rHv_fieldjs_mapkey recorded_keyr>errErr=sr           z$_Printer._RegularMessageToJsonObjectcCs.|jtjjkr ||S|jtjjkr9|jr|S|jjdkr dS|jj |d}|dur/|j S|jj r7t d|S|jtjjkrQ|jtjjkrOt|dS|S|jtjjkr\t|S|jtvret|S|jtvrt|rw|dkrutStSt|r~tS|jtjjkr|j rt!t"||j St#$|S|S)zr9rrr_AnyMessageToJsonObject)s(    z _Printer._AnyMessageToJsonObjectcCs|S)z8Converts message according to Proto3 JSON Specification.)rr3rrrr_GenericMessageToJsonObject>sz$_Printer._GenericMessageToJsonObjectcCs|d}|dus |dkrdS|dkr||jS|dkr3|j}t|r)tdt|r2tdnt||}|j j |}| ||S)z>Converts Value message according to Proto3 JSON Specification.kindN null_value list_value number_valuezTFail to serialize Infinity for Value.number_value, which would parse as string_valuezOFail to serialize NaN for Value.number_value, which would parse as string_value) WhichOneof_ListValueMessageToJsonObjectrrrkrlrVrogetattrr8rJrB)r3rwhichr?oneof_descriptorrrr_ValueMessageToJsonObjectDs       z"_Printer._ValueMessageToJsonObjectcsfdd|jDS)zBConverts ListValue message according to Proto3 JSON Specification.csg|]}|qSr)r)rCr?r3rrrF\sz:_Printer._ListValueMessageToJsonObject..)valuesr{rrrrZs z&_Printer._ListValueMessageToJsonObjectcCs*|j}i}|D] }|||||<q|S)z?Converts Struct message according to Proto3 JSON Specification.)rQr)r3rrQretrYrrr_StructMessageToJsonObject_s z#_Printer._StructMessageToJsonObjectcCs||jjd|jS)Nr?)rBr8rJr?r{rrrr;gsz$_Printer._WrapperMessageToJsonObjectFFFNN)r rrrr4rr%r=rBrzr|rrrr;rrrrrs$  @, rcCs |jjdkS)Nzgoogle/protobuf/wrappers.proto)filerH)r>rrrr:ls r:cCs4i}|D]\}}||vrtd||||<q|S)Nz'Failed to load JSON: duplicate key {0}.)rr1)r7resultrHr?rrr_DuplicateCheckerps   rc Cspt}|dur |jn|}|dd}z||}Wnty/}ztd||d}~wwt |}|S)z"Creates a message from a type URL.N/z0Can not find message descriptor by type_url: {0}) rDefaultpoolsplitFindMessageTypeByNameKeyError TypeErrorr1rGetMessageClass)rvr!dbr type_namer>r[ message_classrrrrwys rwdc Csdt|ts |d}z tj|td}Wnty)}z tdt||d}~wwt |||||S)alParses a JSON representation of a protocol message into a message. Args: text: Message JSON representation. message: A protocol buffer message to merge into. ignore_unknown_fields: If True, do not raise errors for unknown fields. descriptor_pool: A Descriptor Pool for resolving types. If None use the default. max_recursion_depth: max recursion depth of JSON message to be deserialized. JSON messages over this depth will fail to be deserialized. Default value is 100. Returns: The same message passed as argument. Raises:: ParseError: On JSON parsing problems. r])object_pairs_hookzFailed to load JSON: {0}.N) rKrMrgr5loadsrrVrr1 ParseDict)textrignore_unknown_fieldsr!max_recursion_depthr7r[rrrParses   rcCst|||}|||d|S)a8Parses a JSON dictionary representation into a message. Args: js_dict: Dict representation of a JSON message. message: A protocol buffer message to merge into. ignore_unknown_fields: If True, do not raise errors for unknown fields. descriptor_pool: A Descriptor Pool for resolving types. If None use the default. max_recursion_depth: max recursion depth of JSON message to be deserialized. JSON messages over this depth will fail to be deserialized. Default value is 100. Returns: The same message passed as argument. )_ParserConvertMessage)js_dictrrr!rparserrrrrs rc@s`eZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ dS)rz(JSON format parser for protocol message.cCs||_||_||_d|_dS)Nr)rr!rrecursion_depth)r3rr!rrrrr4s z_Parser.__init__cCs|jd7_|j|jkrtd|j|j}|j}|s |j}t|r,||||n|t vr>t t |d||||n| ||||jd8_dS)aConvert a JSON object into a message. Args: value: A JSON object. message: A WKT or regular protocol message to record the data. path: parent path to log parse error info. Raises: ParseError: In case of convert problems. z,Message too deep. Max recursion depth is {0}N) rrrr1r8r9rHr:_ConvertWrapperMessager<r_ConvertFieldValuePair)r3r?rpathr>r9rrrrs z_Parser.ConvertMessagec Cs2g}|j}tdd|jD}|D]}z||d}|s%|j|d}|sXt|rX|js8td |j ||dd} |j | }|sXd | ddd} |j | }|sp|jr_Wqtd |j ||d d |jD||vrtd |jj ||||||} |jdur| dur|jj} | |vrtd |jj | ||| | dur|jtjjkr|jj d krt||j} d| _n|jtjjkr|jj dkrt||jdn||jWqt|r||j|| ||d ||n|j tjj!kr|||jt"| t#std || ||jtjjkrQt$| D].\} }t||j%} |durB| jj d krBtd ||| |&|| d ||| q!ntt$| D]%\} }|durgtd ||| t||jt'||d ||| qUnI|jtjjkr|j(r|j |} nt||j} | )|&| | d ||n!|j(rt'| |d |||j |<nt||jt'| |d ||Wqty}z|r|jdurtd |||tt*||d}~wt+y}z td |||d}~wt,y}z td |||d}~wwdS)a'Convert field value pairs into regular message. Args: js: A JSON object to convert the field value pairs. message: A regular protocol message to record the data. path: parent path to log parse error info. Raises: ParseError: In case of problems converting. css|]}|j|fVqdSr'rIrCfrrr sz1_Parser._ConvertFieldValuePair..Nz0Message type {0} does not have extensions at {1}rr.zaMessage type "{0}" has no field named "{1}" at "{2}". Available Fields(except extensions): "{3}"cSsg|]}|jqSrrrrrrrFsz2_Parser._ConvertFieldValuePair..zBMessage type "{0}" should not have multiple "{1}" fields at "{2}".zHMessage type "{0}" should not have multiple "{1}" oneof fields at "{2}".google.protobuf.Valuerr\{0}.{1}z4repeated field {0} must be in [] which is {1} at {2}zPnull is not allowed to be used as an element in a repeated field at {0}.{1}[{2}]z {0}.{1}[{2}]zFailed to parse {0} field: {1}.)-r8dictrQrarJ_VALID_EXTENSION_NAMEmatch is_extendablerr1r9 Extensions_FindExtensionByNamejoinrrappendrTrHrRrr)rSr+rr~r^r_setattr ClearFieldr0_ConvertMapFieldValuerNrOrKlist enumerateaddr_ConvertScalarFieldValuerP SetInParentrMrVr)r3r7rrnamesr>fields_by_json_namerHr/ identifierr? oneof_nameryindexitemr[rrrrs                              z_Parser._ConvertFieldValuePairc Cst|tr |s dSz|d}Wnty#}ztd||d}~wwzt||j}WntyA}z td|||d}~ww|j}|j }t |rY| |d|d|n%|t vrpt t |d|d|d||n|d=||||||d<||_||_dS)z/Convert a JSON representation into Any message.Nruz0@type is missing when parsing any message at {0} {0} at {1}r? {0}.valuer)rKrrrr1rwr!rr8r9r:rr<rrSerializeToStringr?rv) r3r?rrrvr[ryr>r9rrr_ConvertAnyMessagersF    z_Parser._ConvertAnyMessagec Cs<z||WdSty}z td|||d}~ww)z?Convert a JSON representation into message with FromJsonString.rN)FromJsonStringrVrr1)r3r?rrr[rrr_ConvertGenericMessages z_Parser._ConvertGenericMessagecCst|tr|||j|dSt|tr|||j|dS|dur'd|_dSt|tr1||_ dSt|t r;||_ dSt|t rE||_ dStd|t||)z1Convert a JSON representation into Value message.Nrz(Value {0} has unexpected type {1} at {2})rKr_ConvertStructMessage struct_valuer_ConvertListValueMessagerr~rL bool_valuerM string_value _INT_OR_FLOATrrr1r()r3r?rrrrr_ConvertValueMessages          z_Parser._ConvertValueMessagec CsVt|ts td|||dt|D]\}}|||jd||qdS)z5Convert a JSON representation into ListValue message.z+ListValue must be in [] which is {0} at {1}r{0}[{1}]N) rKrrr1rrrrr)r3r?rrrrrrrrs   z _Parser._ConvertListValueMessagec CsPt|ts td||||D]}||||j|d||qdS)z2Convert a JSON representation into Struct message.z,Struct must be in a dict which is {0} at {1}rN)rKrrr1ClearrrQ)r3r?rrrYrrrrs  z_Parser._ConvertStructMessagec Cs,|jjd}t|dt||d|ddS)z3Convert a JSON representation into Wrapper message.r?rrN)r8rJrrr1)r3r?rrr/rrrrs z_Parser._ConvertWrapperMessagec Cst|tstd|j|||jjd}|jjd}|D]:}t||d|d}|jt j j krD| ||t ||j|d||qt|||d||dt ||j|<qdS) afConvert map field value for a message map field. Args: value: A JSON object to convert the map field value. message: A protocol message to record the converted data. field: The descriptor of the map field to be converted. path: parent path to log parse error info. Raises: ParseError: In case of convert problems. z3Map field {0} must be in a dict which is {1} at {2}rYr?z{0}.keyTrrN)rKrrr1rHr+rJrrRrr)rSrr) r3r?rr/r key_field value_fieldrY key_valuerrrrs*      z_Parser._ConvertMapFieldValueN)r rrrr4rrrrrrrrrrrrrrs     rc Csz|jtvr t|WS|jtvrt||WS|jtjjkr#t||WS|jtjj kr\|j tjj krPt |t r<|d}n|}|ddt|d}t|WSt|rYtd|WS|jtjjkr|jj|d}|durzt|}|jj|d}Wnty}z td||jj|d}~ww|dur|jjrtd||jj|WS|jWSWdSty}z td|||d}~ww)aqConvert a single scalar field value. Args: value: A scalar value to convert the scalar field value. field: The descriptor of the field to convert. path: parent path to log parse error info. require_str: If True, the field value must be a str. Returns: The converted scalar field value Raises: ParseError: In case of convert problems. r]=zUnpaired surrogateNz(Invalid enum value {0} for enum type {1}r) rR _INT_TYPES_ConvertIntegerrj _ConvertFloatrr)rh _ConvertBoolrcr(rdrKrMencodelenreurlsafe_b64decode_UNPAIRED_SURROGATE_PATTERNsearchrr^r_values_by_nameraintr`rVr1r9rbnumber) r?r/r require_strencoded padded_valuertrr[rrrrsV         rcCsft|tr|std|t|tr#|ddkr#td|t|tr/td|t|S)zConvert an integer. Args: value: A scalar value to convert. Returns: The integer value. Raises: ParseError: If an integer couldn't be consumed. zCouldn't parse integer: {0} rzCouldn't parse integer: "{0}"z2Bool value {0} is not acceptable for integer field) rKrr is_integerrr1rMfindrLr)r?rrrr#s  rc Cst|tr8t|rtdt|r|dkrtdtd|jtjj kr8|t j kr/td|t j kr8td|dkr@tdzt|WSt y}z2|tkr[td WYd }~S|tkritd WYd }~S|tkrwtdWYd }~Std ||d }~ww) z!Convert an floating point number.z,Couldn't parse NaN, use quoted "NaN" insteadrzICouldn't parse Infinity or value too large, use quoted "Infinity" insteadzKCouldn't parse -Infinity or value too small, use quoted "-Infinity" insteadzFloat value too largezFloat value too smallnanz-Couldn't parse float "nan", use "NaN" insteadz-infNinfzCouldn't parse float: {0})rKrrrkrorrlrRrr)rqr _FLOAT_MAX _FLOAT_MINrVrmrnrpr1)r?r/r[rrrr<s4      rcCs@|r|dkrdS|dkrdStd|t|tstd|S)zConvert a boolean value. Args: value: A scalar value to convert. require_str: If True, value must be a str. Returns: The bool parsed. Raises: ParseError: If a boolean value couldn't be consumed. r@TrAFz#Expected "true" or "false", not {0}z%Expected true or false without quotes)rr1rKrL)r?rrrrr`s  rrzrr|rrrrrrr)zgoogle.protobuf.Anyzgoogle.protobuf.Durationzgoogle.protobuf.FieldMaskzgoogle.protobuf.ListValuezgoogle.protobuf.Structzgoogle.protobuf.Timestampr)FFrFFNNTr)FNr)F)8r __author__re collectionsrr5rkoperatorrregoogle.protobuf.internalrgoogle.protobufrrr frozensetr) CPPTYPE_INT32CPPTYPE_UINT32 CPPTYPE_INT64CPPTYPE_UINT64rrirqCPPTYPE_DOUBLErjrnrmrpcompilerr Exceptionr rrr$r&r0objectrr:rrwrrrrrrrrrrrr<rrrrs         . &S  #  &:$