|
Server : Apache System : Linux iad1-shared-b8-43 6.6.49-grsec-jammy+ #10 SMP Thu Sep 12 23:23:08 UTC 2024 x86_64 User : dh_edsupp ( 6597262) PHP Version : 8.2.26 Disable Function : NONE Directory : /lib/python3/dist-packages/pymacaroons/serializers/__pycache__/ |
Upload File : |
o
�%�YU � @ sP d dl Z d dlZd dlmZ G dd� de�Zdd� Zdd� Zd d
� Zd d � Z dS )
� N)�utilsc @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�JsonSerializerz8Serializer used to produce JSON macaroon format v1.
c C s, ddl m} |j|jkr| �|�S | �|�S )z�Serialize the macaroon in JSON format indicated by the version field.
@param macaroon the macaroon to serialize.
@return JSON macaroon.
r )�macaroon)� pymacaroonsr �version� MACAROON_V1�
_serialize_v1�
_serialize_v2)�self�mr � r �I/usr/lib/python3/dist-packages/pymacaroons/serializers/json_serializer.py� serialize s
zJsonSerializer.serializec C sH t �|j�|jd�}|jr|j|d< |jr dd� |jD �|d< t�|�S )z�Serialize the macaroon in JSON format v1.
@param macaroon the macaroon to serialize.
@return JSON macaroon.
)�
identifier� signature�locationc S � g | ]}t |��qS r )�_caveat_v1_to_dict��.0�caveatr r r
�
<listcomp>! � �z0JsonSerializer._serialize_v1.<locals>.<listcomp>�caveats)r �convert_to_stringr r r r �json�dumps�r
r �
serializedr r r
r s
�
�
z JsonSerializer._serialize_v1c C sZ i }t |j|d� t t�|j�|d� |jr|j|d< |jr(dd� |jD �|d< t�|�S )z�Serialize the macaroon in JSON format v2.
@param macaroon the macaroon to serialize.
@return JSON macaroon in v2 format.
�i�s�lc S r r )�_caveat_v2_to_dictr r r r
r 4 r z0JsonSerializer._serialize_v2.<locals>.<listcomp>�c) �_add_json_binary_field�identifier_bytes�binascii� unhexlify�signature_bytesr r r r r r r r
r & s �
�
z JsonSerializer._serialize_v2c C s, t �|�}|�d�du r| �|�S | �|�S )z�Deserialize a JSON macaroon depending on the format.
@param serialized the macaroon in JSON format.
@return the macaroon object.
r N)r �loads�get�_deserialize_v2�_deserialize_v1)r
r � deserializedr r r
� deserialize9 s
zJsonSerializer.deserializec C s� ddl m}m} ddlm} g }|�dg �D ]&}||d |�d�r(t�|d �nd|�d�r2|d nd|d �}|�|� q||�d
�|d ||d |d
�S )z�Deserialize a JSON macaroon in v1 format.
@param serialized the macaroon in v1 JSON format.
@return the macaroon object.
r )�Macaroonr ��Caveatr �cid�vidN�cl�� caveat_id�verification_key_idr r r r r �r r r r r ) �pymacaroons.macaroonr/ r �pymacaroons.caveatr1 r* r �
raw_b64decode�append)r
r- r/ r r1 r r# r r r r
r, E s& � �z JsonSerializer._deserialize_v1c C s� ddl m}m} ddlm} g }|�dg �D ]}|t|d�t|d�t|d�|d�}|�|� q|t|d�t|d�|t� t|d ��|d
�S ) z�Deserialize a JSON macaroon v2.
@param serialized the macaroon in JSON format v2.
@return the macaroon object.
r )r/ � MACAROON_V2r0 r# r �vr! r5 r r8 )
r9 r/ r= r: r1 r* �_read_json_binary_fieldr< r& �hexlify)r
r- r/ r= r1 r r# r r r r
r+ e s( � ��z JsonSerializer._deserialize_v2N)
�__name__�
__module__� __qualname__�__doc__r r r r. r, r+ r r r r
r s r c C sL i }t | j�dkr| j|d< | jr t�| j��d�|d< | jr$| j|d< |S )zT Return a caveat as a dictionary for export as the JSON
macaroon v1 format.
r r2 �utf-8r3 r4 )�lenr6 r7 r �raw_urlsafe_b64encode�decoder �r# r r r r
r � s
��
r c C sH i }t | j�dkrt| j|d� | jrt| j|d� | jr"| j|d< |S )zT Return a caveat as a dictionary for export as the JSON
macaroon v2 format.
r r r> r! )rF �caveat_id_bytesr$ r7 r rI r r r
r" � s
r" c C sJ z | � d�}|||< W dS ty$ t�| �� d�}|||d < Y dS w )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.
rE �64N)rH �UnicodeDecodeErrorr rG )�br �field�valr r r
r$ � s
�r$ c C s@ | � |�}|durt�|�S | � |d �}|du rdS t�|�S )zJ Read the value of a JSON field that may be string or base64-encoded.
NrK )r* r �convert_to_bytes�raw_urlsafe_b64decode)r- rN rO r r r
r? � s
r? )
r& r r r �objectr r r" r$ r? r r r r
�<module> s z