3
SLjr @ s d Z dZyddlmZ W n ek
r8 ddlmZ Y nX ddlZddlZddlZddl Z ddl
Z
ddl Z ddl m
Z
ddlmZ ddlmZ dZeejjejjejjejjgZeejjejjgZeejjejjgZ d Z d
Z d Z e j e
j!d Z"e j d
Z#G dd de$Z%G dd de%Z&G dd de%Z'd=ddZ(d>ddZ)dd Z*G d d d e+Z,d d Z-d d! Z.d"d# Z/d?d$d%Z0d@d&d'Z1e
j2e3f Z4G d(d) d)e+Z5dAd*d+Z6d,d- Z7d.d/ Z8d0d1 Z9d2d3gd4d5gd4d5gd6d7gd8d9gd4d5gd:d;gd<Z:dS )Bat Contains 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) ) OrderedDictN) methodcaller)
descriptor)symbol_databasez%Y-%m-%dT%H:%M:%SZInfinityz -InfinityZNaNzF[\ud800-\udbff](?![\udc00-\udfff])|(?|D ]6}t|trn|rhd} qrd} n|} j ||| || < qPW |||< q j
t j j
kr fdd|D ||< q jr }
|
jj jr|
jt j jkr|
j
t j jkr|
j}
d|
j|f }j |||< qj |||< qW jr|j} x| jD ] j
t j j
krd jt j jks@ jrpq@jr j}n j}||krq@t ri ||< n, j
t j j
krg ||< nj j ||< q@W W n6 t k
r } zt dj j| W Y dd} ~ X nX |S ) z?Converts normal message according to Proto3 JSON Specification.valuetruefalsec s g | ]}j |qS r )_FieldToJsonObject).0k)r r" r r
s z8_Printer._RegularMessageToJsonObject..z[%s.%s]z#Failed to serialize {0} field: {1}.N)
ListFieldsr name json_namer! r fields_by_name
isinstanceboolr1 labelr r LABEL_REPEATED is_extensionZcontaining_typer Zmessage_set_wire_formatr r ZLABEL_OPTIONALr( r r' fieldscpp_typeCPPTYPE_MESSAGEcontaining_oneofZ
default_value
ValueErrorr
format)
r" r r& r> r. r6 Zv_fieldZjs_mapkeyZ recorded_keyfr- er )r r" r r, sf
"z$_Printer._RegularMessageToJsonObjectc C s |j tjjkr| j|S |j tjjkr^|jjj|d}|dk rD|j S |j
j dkrT|S t dn|j tjj
kr|jtjjkrtj|jdS |S nZ|j tjjkrt|S |j tkrt|S |j tkrtj|r|dk rtS tS tj |rt S |S )zConverts Value message according to Proto3 JSON Specification.ZkindN
null_value
list_value struct_value)Z
WhichOneof _ListValueMessageToJsonObjectra rb getattrr' r8 r1 )r" r Zwhichr. Zoneof_descriptorr r r _ValueMessageToJsonObject1 s
z"_Printer._ValueMessageToJsonObjectc s fdd|j D S )zBConverts ListValue message according to Proto3 JSON Specification.c s g | ]} j |qS r )re )r2 r. )r" r r r4 C s z:_Printer._ListValueMessageToJsonObject..)values)r" r r )r" r rc A s
z&_Printer._ListValueMessageToJsonObjectc C s. |j }i }x |D ]}| j|| ||< qW |S )z?Converts Struct message according to Proto3 JSON Specification.)r> re )r" r r> retrD r r r _StructMessageToJsonObjectF s
z#_Printer._StructMessageToJsonObjectc C s | j |jjd |jS )Nr. )r1 r' r8 r. )r" r r r r r* N s z$_Printer._WrapperMessageToJsonObjectN)FF)r r r r
r# r r r, r1 r^ r_ re rc rh r* r r r r r s
E!r c C s | j jdkS )Nz google/protobuf/wrappers.proto)rL r6 )r- r r r r) S s r) c C s8 i }x.| D ]&\}}||kr(t dj||||< q
W |S )Nz'Failed to load JSON: duplicate key {0}.)r rC )r& resultr6 r. r r r _DuplicateCheckerW s rj c C sZ t j }| jdd }y|jj|}W n" tk
rH tdj| Y nX |j|}| S )N/ z1Can not find message descriptor by type_url: {0}.) r ZDefaultsplitZpoolZFindMessageTypeByNameKeyError TypeErrorrC Z GetPrototype)r[ ZdbZ type_namer- Z
message_classr r r r\ ` s
r\ c C s t | tjs| jd} y(tjdk r.tj| }ntj| td}W n4 t k
rr } zt
dj t |W Y dd}~X nX t
|||S )ab Parses 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.
Returns:
The same message passed as argument.
Raises::
ParseError: On JSON parsing problems.
zutf-8r )Zobject_pairs_hookzFailed to load JSON: {0}.N)r rq )r9 six text_typerQ sys version_infor$ loadsrj rB r rC rT ParseDict)textr ignore_unknown_fieldsr& rF r r r Parsen s
$rz c C s t |}|j| | |S )a. Parses 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.
Returns:
The same message passed as argument.
)_ParserConvertMessage)Zjs_dictr ry parserr r r rw s rw c @ s` e Zd ZdZdd Zdd Zdd Zdd Zd
d Zd d
Z dd Z
dd Z dd Z dd Z
dS )r{ z(JSON format parser for protocol message.c C s
|| _ d S )N)ry )r" ry r r r r# s z_Parser.__init__c C sT |j }|j}t|r"| j|| n.|tkrDtt| d ||| n | j|| dS )zConvert a JSON object into a message.
Args:
value: A JSON object.
message: A WKT or regular protocol message to record the data.
Raises:
ParseError: In case of convert problems.
rl N)r' r( r) _ConvertWrapperMessager+ r _ConvertFieldValuePair)r" r. r r- r( r r r r| s
z_Parser.ConvertMessagec C s g }|j }tdd |jD }xl|D ]b}y|j|d}|sN|jj|d}| rtj|r|jsttdj |j
|dd }dj |j ddd }|j
j|}|s| jrw&tdj |j
||j||krtdj |j j
||j| |jdk r(|jj} | |kr td j |j j
| |j| || }
|
dkrz|jtjjkrl|jj
d
krlt||j} d | _n |j|j w&t|r|j|j | j |
|| n,|j tjj krt|j|j t |
t std j ||
|jtjjkr8x|
D ]B} t||jj! } | dkr$| j j
d
kr$td
| j"| | qW n:x|
D ]0} | dkrTtd
t||jjt#| | q>W nZ|jtjjkr|j$r|j
| } n t||j} | j% | j"|
| nt&||jt#|
| W q& tk
r$ }
z6|r|jdkrtdj ||
n tt'|
W Y dd}
~
X q& t(k
rV }
ztdj ||
W Y dd}
~
X q& t)k
r }
ztdj ||
W Y dd}
~
X q&X q&W dS )zConvert 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.
Raises:
ParseError: In case of problems converting.
c s s | ]}|j |fV qd S )N)r7 )r2 rE r r r s z1_Parser._ConvertFieldValuePair..Nz)Message type {0} does not have extensionsrl .zVMessage type "{0}" has no field named "{1}".
Available Fields(except extensions): {2}z9Message type "{0}" should not have multiple "{1}" fields.z?Message type "{0}" should not have multiple "{1}" oneof fields.zgoogle.protobuf.Valuer z.repeated field {0} must be in [] which is {1}.zAnull is not allowed to be used as an element in a repeated field.z Failed to parse {0} field: {1}z Failed to parse {0} field: {1}.rm rm )*r' dictr> rK r8 _VALID_EXTENSION_NAMEmatchZ
is_extendabler rC r( joinrn Z
ExtensionsZ_FindExtensionByNamery appendrA r6 r? r r r@ r rd r`
ClearFieldr! _ConvertMapFieldValuer; r< r9 listaddr| _ConvertScalarFieldValuer= Z SetInParentsetattrrT rB rp )r" r& r namesr- Zfields_by_json_namer6 r Z
identifierZ
oneof_namer. r] itemrF r r r r s
z _Parser._ConvertFieldValuePairc C s t |tr| rdS y |d }W n tk
r< tdY nX t|}|j}|j}t|rl| j|d | n8|t krt
t | d |d || n|d= | j || |j |_
||_dS )z/Convert a JSON representation into Any message.Nz@typez*@type is missing when parsing any message.r. rl )r9 r ro r r\ r' r( r) r~ r+ r r ZSerializeToStringr. r[ )r" r. r r[ r] r- r( r r r _ConvertAnyMessage s$
z_Parser._ConvertAnyMessagec C s |j | dS )z?Convert a JSON representation into message with FromJsonString.N)ZFromJsonString)r" r. r r r r _ConvertGenericMessage7 s z _Parser._ConvertGenericMessagec C s t |tr| j||j njt |tr4| j||j nP|dkrDd|_n@t |trV||_ n.t |t
j rj||_ nt |t
r|||_ntddS )z1Convert a JSON representation into Value message.Nr z"Unexpected type for Value message.)r9 r _ConvertStructMessagerb r _ConvertListValueMessagera r` r: Z
bool_valuerr Z string_typesZ string_value
_INT_OR_FLOATZ number_valuer )r" r. r r r r _ConvertValueMessage= s
z _Parser._ConvertValueMessagec C sF t |tstdj||jd x |D ]}| j||jj q(W dS )z5Convert a JSON representation into ListValue message.z%ListValue must be in [] which is {0}.rf N)r9 r r rC r r rf r )r" r. r r r r r r N s
z _Parser._ConvertListValueMessagec C s@ t |tstdj|x"|D ]}| j|| |j| q W dS )z2Convert a JSON representation into Struct message.z&Struct must be in a dict which is {0}.N)r9 r r rC r r> )r" r. r rD r r r r W s
z _Parser._ConvertStructMessagec C s" |j jd }t|dt|| dS )z3Convert a JSON representation into Wrapper message.r. N)r' r8 r r )r" r. r r r r r r~ ` s z _Parser._ConvertWrapperMessagec C s t |ts tdj|j||jjd }|jjd }x^|D ]V}t||d}|jt j
j krv| j || t
||j| q:t|| |t
||j|< q:W dS )a5 Convert 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.
Raises:
ParseError: In case of convert problems.
z-Map field {0} must be in a dict which is {1}.rD r. TN)r9 r r rC r6 r r8 r r? r r r@ r| rd )r" r. r r Z key_fieldZ value_fieldrD key_valuer r r r e s
z _Parser._ConvertMapFieldValueN)r r r r
r# r| r r r r r r r~ r r r r r r{ s j r{ c C s$ |j tkrt| S |j tkr$t| S |j tjjkr) s
<
f
0