U %YU@sPddlZddlZddlmZGdddeZddZddZd d Zd d Z dS) N)utilsc@s@eZdZdZddZddZddZdd Zd d Zd d Z dS)JsonSerializerz8Serializer used to produce JSON macaroon format v1. cCs,ddlm}|j|jkr"||S||S)zSerialize the macaroon in JSON format indicated by the version field. @param macaroon the macaroon to serialize. @return JSON macaroon. r)macaroon) pymacaroonsrversion MACAROON_V1 _serialize_v1 _serialize_v2)selfmrr I/usr/lib/python3/dist-packages/pymacaroons/serializers/json_serializer.py serialize s   zJsonSerializer.serializecCsHt|j|jd}|jr$|j|d<|jr>dd|jD|d<t|S)zSerialize the macaroon in JSON format v1. @param macaroon the macaroon to serialize. @return JSON macaroon. ) identifier signaturelocationcSsg|] }t|qSr )_caveat_v1_to_dict.0caveatr r r !sz0JsonSerializer._serialize_v1..caveats)rZconvert_to_stringrrrrjsondumpsr r serializedr r r rs   zJsonSerializer._serialize_v1cCsZi}t|j|dtt|j|d|jr6|j|d<|jrPdd|jD|d<t|S)zSerialize the macaroon in JSON format v2. @param macaroon the macaroon to serialize. @return JSON macaroon in v2 format. islcSsg|] }t|qSr )_caveat_v2_to_dictrr r r r4sz0JsonSerializer._serialize_v2..c) _add_json_binary_fieldZidentifier_bytesbinasciiZ unhexlifyZsignature_bytesrrrrrr r r r &s   zJsonSerializer._serialize_v2cCs0t|}|ddkr"||S||SdS)zDeserialize a JSON macaroon depending on the format. @param serialized the macaroon in JSON format. @return the macaroon object. rN)rloadsget_deserialize_v2_deserialize_v1)r r deserializedr r r deserialize9s  zJsonSerializer.deserializecCsddlm}m}ddlm}g}|dgD]L}||d|drPt|dnd|drd|dnd|d }||q,||d |d ||d |d S)zDeserialize a JSON macaroon in v1 format. @param serialized the macaroon in v1 JSON format. @return the macaroon object. r)MacaroonrCaveatrcidvidNcl caveat_idverification_key_idrrrrrrrrrr) pymacaroons.macaroonr)rpymacaroons.caveatr+r$rZ raw_b64decodeappend)r r'r)rr+rr rr r r r&Es&  zJsonSerializer._deserialize_v1c Csddlm}m}ddlm}g}|dgD]0}|t|dt|dt|d|d}||q,|t|dt|d|t t|d |d S) zDeserialize a JSON macaroon v2. @param serialized the macaroon in JSON format v2. @return the macaroon object. r)r) MACAROON_V2r*r rvrr/rr2) r3r)r6r4r+r$_read_json_binary_fieldr5r"Zhexlify)r r'r)r6r+rr rr r r r%es(  zJsonSerializer._deserialize_v2N) __name__ __module__ __qualname____doc__rrr r(r&r%r r r r rs   rcCsLi}t|jdkr|j|d<|jr8t|jd|d<|jrH|j|d<|S)zT Return a caveat as a dictionary for export as the JSON macaroon v1 format. rr,utf-8r-r.)lenr0r1rraw_urlsafe_b64encodedecoderr rr r r rs  rcCsHi}t|jdkr t|j|d|jr4t|j|d|jrD|j|d<|S)zT Return a caveat as a dictionary for export as the JSON macaroon v2 format. rrr7r)r>Zcaveat_id_bytesr!r1rrAr r r rs rcCsLz|d}|||<Wn0tk rFt|d}|||d<YnXdS)z Set the given field to the given val (a bytearray) in the serialized dictionary. If the value isn't valid utf-8, we base64 encode it and use field+"64" as the field name. r=64N)r@UnicodeDecodeErrorrr?)brfieldvalr r r r!s   r!cCs@||}|dk rt|S||d}|dkr6dSt|S)zJ Read the value of a JSON field that may be string or base64-encoded. NrB)r$rZconvert_to_bytesZraw_urlsafe_b64decode)r'rErFr r r r8s  r8) r"rrrobjectrrrr!r8r r r r s z