o 4@sdZddlZddlZddlZddlZddlmZddlmZddlmZgdZ ddZ e Z Gd d d e j Z Gd d d eZejZejZejZejZdS) zJSON support for message types. Public classes: MessageJSONEncoder: JSON encoder for message objects. Public functions: encode_message: Encodes a message in to a JSON string. decode_message: Merge from a JSON string in to a message. N) message_types)messages)util)ALTERNATIVE_CONTENT_TYPES CONTENT_TYPEMessageJSONEncoderencode_messagedecode_message ProtoJsonc Csd}dD]6}zt|iid}t|ds d|}t|t||WSty:}z |s0|}WYd}~qd}~wwtd|)aTry to load a valid json module. There are more than one json modules that might be installed. They are mostly compatible with one another but some versions may be different. This function attempts to load various json modules in a preferred order. It does a basic check to guess if a loaded version of json is compatible. Returns: Compatible json module. Raises: ImportError if there are no json modules or the loaded json module is not compatible with ProtoRPC. N)json simplejsonr JSONEncoderz1json library "%s" is not compatible with ProtoRPCz0Must use valid json library (json or simplejson)) __import__hasattrloggingwarning ImportErrorerror)first_import_error module_namemodulemessageerrrM/tmp/google-cloud-sdk/lib/third_party/apitools/base/protorpclite/protojson.py_load_json_module/s&    rcs.eZdZdZdfdd ZfddZZS)rziMessage JSON encoder class. Extension of JSONEncoder that can build JSON from a message object. Nc s(tt|jdi||pt|_dS)zVConstructor. Args: protojson_protocol: ProtoJson instance. Nr)superr__init__r get_default'_MessageJSONEncoder__protojson_protocol)selfprotojson_protocolkwargs __class__rrr]s zMessageJSONEncoder.__init__cst|tjr t|Stjrt|tr|dSt|tjrQi}| D]}| |j }|dgdfvr<|j ||||j <q#|D] }||\}}|||<qA|Stt||S)zReturn dictionary instance from a message object. Args: value: Value to get dictionary for. If not encodable, will call superclasses default method. utf8Nr) isinstancerEnumstrsixPY3bytesdecodeMessage all_fieldsget_assigned_valuenamer encode_fieldall_unrecognized_fieldsget_unrecognized_field_inforrdefault)r valueresultfielditem unknown_keyunrecognized_field_r#rrr4gs&        zMessageJSONEncoder.default)N)__name__ __module__ __qualname____doc__rr4 __classcell__rrr#rrWs rc@sdeZdZdZdZgdZddZddZdd Zd d Z d d Z ddZ e ddZ e ddZdS)r aProtoRPC JSON implementation class. Implementation of JSON based protocol used for serializing and deserializing message objects. Instances of remote.ProtocolConfig constructor or used with remote.Protocols.add_protocol. See the remote.py module for more details. zapplication/json)zapplication/x-javascriptztext/javascriptztext/x-javascriptz text/x-jsonz text/jsoncCsbt|tjr|jrdd|D}|St|}|St|tjr/|jr+dd|D}|S|}|S)zEncode a python field value to a JSON value. Args: field: A ProtoRPC field instance. value: A python value supported by field. Returns: A JSON serializable value appropriate for field. cSsg|]}t|qSr)base64 b64encode).0byterrr sz*ProtoJson.encode_field..cSsg|]}|qSr) isoformat)rCirrrrEs) r&r BytesFieldrepeatedrArBr DateTimeFieldrF)r r7r5rrrr1s   zProtoJson.encode_fieldcCs|tj|t|dS)a#Encode Message instance to JSON string. Args: Message instance to encode in to JSON string. Returns: String encoding of Message instance in protocol JSON format. Raises: messages.ValidationError if message is not initialized. )clsr!)check_initializedr dumpsr)r rrrrrs zProtoJson.encode_messagecCs:t|}|s |St|}|||}||S)aMerge JSON structure to Message instance. Args: message_type: Message to decode data to. encoded_message: JSON encoded version of message. Returns: Decoded instance of message_type. Raises: ValueError: If encoded_message is not valid JSON. messages.ValidationError if merged message is not initialized. )r) ensure_strstripr loads_ProtoJson__decode_dictionaryrL)r message_typeencoded_message dictionaryrrrrr s   zProtoJson.decode_messagec Cst|tr tjjSt|tjrtjjSt|trtjj St|tj r&tjj St|t t frbdtjjtjj tjj g}d}|D] }||}z||}Wn tyVd}Ynw||kr]|}q=||SdS)a,Find the messages.Variant type that describes this value. Args: value: The value whose variant type is being determined. Returns: The messages.Variant value that best describes value's type, or None if it's a type we don't know how to handle. Nr)r&boolrVariantBOOLr) integer_typesINT64floatDOUBLE string_typesSTRINGlisttuple_ProtoJson__find_variantindex IndexError)r r5variant_prioritychosen_priorityvvariantpriorityrrr__find_variants6      zProtoJson.__find_variantc Cs|}t|D]\}}|dur"z||Wn ty Ynwqz||}Wnty@||}|r>||||Yqwt|t j }d} |j rt|t sT|g}g} |D]'} z| || } |ri| duriWqXWnt jyy|sud} YqXw| | qXt||j| | r||}|r||||q|gkrqz t||j| ||Wqt jy|s||}|r||||Yqw|S)aMerge dictionary in to message. Args: message: Message to merge dictionary in to. dictionary: Dictionary to extract information from. Dictionary is as parsed from JSON. Nested objects will also be dictionaries. NFT)r) iteritemsresetAttributeError field_by_nameKeyErrorraset_unrecognized_fieldr&r EnumFieldrIr_ decode_field DecodeErrorappendsetattrr0) r rRrTrkeyr5r7rg is_enum_fieldis_unrecognized_field valid_valuer8rfrrr__decode_dictionarysn          zProtoJson.__decode_dictionaryc CsRt|tjrz||WStytd|pdwt|tjr@zt|WSt j tfy?}ztd|d}~wwt|t j r`zt j|ddWSty_}zt|d}~wwt|tjrtt|jtjrt||j|St|tjrt|tjtjfrzt|WSY|St|tjrt|tjrzt|WSY|S|S)zDecode a JSON value to a python value. Args: field: A ProtoRPC field instance. value: A serialized JSON value. Return: A Python value compatible with field. zInvalid enum value "%s"zBase64 decoding error: %sNT) truncate_time)r&rrptype TypeErrorrrrHrA b64decodebinasciiErrorrrJrdecode_datetime ValueError MessageField issubclassr-rQ FloatFieldr)rYr]r[ IntegerFieldint)r r7r5rrrrrqDsT              zProtoJson.decode_fieldcCs*ztjWStytt_tjYSw)z!Get default instanceof ProtoJson.)r _ProtoJson__defaultrlrrrrrus   zProtoJson.get_defaultcCst|ts td|t_dS)zhSet the default instance of ProtoJson. Args: protocol: A ProtoJson instance. z#Expected protocol of type ProtoJsonN)r&r r}r)protocolrrr set_default~s  zProtoJson.set_defaultN)r<r=r>r?rrr1rr rarQrq staticmethodrrrrrrr s &D1 r )r?rArrr)apitools.base.protorpcliterrr__all__rr r robjectr rrrrr rrrrs&     $1