File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/__pycache__/msgpack.cpython-310.pyc
o
�N�g� � @ s0 d Z ddlZe�e�ZdZzddlZejejg d�dd�dd�du r$e �dZW n e yB zddl
ZdZW n e y? Y nw Y nw erNe ed�rNej Z nG d d
� d
e
�ZG d d � d �Ze� Z ejZejZesjd
nejZdd� Zdd� ZG dd� dej�Zdd� Zdd� Zdd� Zdd� ZeZeZeZeZdS ) z$
Functions to work with MessagePack
� NF)� � � )Z use_bin_typeT)Zuse_list�
exceptionsc @ s e Zd ZdZdS )�PackValueErrorzF
older versions of msgpack do not have PackValueError
N)�__name__�
__module__� __qualname__�__doc__� r r �F/opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/msgpack.pyr " s r c @ s e Zd ZdZe� ZdS )� _exceptionszL
older versions of msgpack do not have an exceptions module
N)r r r r
r r r r r r
' s
r
)r r r c C s, t | t�sJ �| �dd�durt�d� | S )z�
Clean up msgpack keyword arguments based on the version
https://github.com/msgpack/msgpack-python/blob/master/ChangeLog.rst
�encodingNz:removing unsupported `encoding` argument from msgpack call)�
isinstance�dict�pop�log�debug��kwargsr r r �_sanitize_msgpack_kwargs6 s
r c C s. t | t�sJ �| �dd� | �dd� t| �S )z�
Clean up msgpack keyword arguments for unpack operations, based on
the version
https://github.com/msgpack/msgpack-python/blob/master/ChangeLog.rst
�rawTZstrict_map_keyF)r r �
setdefaultr r r r r � _sanitize_msgpack_unpack_kwargsB s r c @ s e Zd ZdZdd� ZdS )�UnpackerzG
Wraps the msgpack.Unpacker and removes non-relevant arguments
c O s" t jj| g|�R i t|��� d S )N)�msgpackr �__init__r )�self�argsr r r r r S s ��
�zUnpacker.__init__N)r r r r
r r r r r r N s r c K s t j| |fi t|��� dS )z�
.. versionadded:: 2018.3.4
Wraps msgpack.pack and ensures that the passed object is unwrapped if it is
a proxy.
By default, this function uses the msgpack module and falls back to
msgpack_pure, if the msgpack is not available.
N)r �packr )�o�streamr r r r r Y s r c K � t j| fi t|���S )z�
.. versionadded:: 2018.3.4
Wraps msgpack.packb and ensures that the passed object is unwrapped if it
is a proxy.
By default, this function uses the msgpack module and falls back to
msgpack_pure, if the msgpack is not available.
)r �packbr )r r r r r r# g s
r# c K r" )z�
.. versionadded:: 2018.3.4
Wraps msgpack.unpack.
By default, this function uses the msgpack module and falls back to
msgpack_pure, if the msgpack is not available.
)r �unpackr )r! r r r r r$ t � r$ c K r" )z�
.. versionadded:: 2018.3.4
Wraps msgpack.unpack.
By default, this function uses the msgpack module and falls back to
msgpack_pure.
)r �unpackbr )�packedr r r r r& � r% r& ) r
�logging� getLoggerr r Z HAS_MSGPACKr �loads�dumps� ImportErrorZ msgpack_pure�hasattrr � Exceptionr r
ZPackerZExtType�versionr r r r r# r$ r& �load�dumpr r r r �<module> sN
� ���