File: //usr/local/lib/python3.9/site-packages/celery/utils/__pycache__/serialization.cpython-39.pyc
a
X>h � @ sd d Z ddlZddlZddlZddlmZ ddlmZ ddl m
Z
ddl m Z ddl
mZ ddlmZmZmZ z ddlZW n ey� ddlZY n0 d ZeeefZd
d
d
d d d d d
d �Zd
d� Z ej ej fdd�Z dd� Z d+dd�Z!dd� Z"G dd� de�Z#dd� Z$ej ej fdd �Z%d d � Z&d d � Zd!d"� Zd,d#d$�Z'd%d&� Z(ej)e*fdddfd'd(�Z+d)d*� Z,dS )-z)Utilities for safely pickling exceptions.� N)� b64decode)� b64encode)�partial)�getmro)� takewhile)� bytes_to_str� safe_repr� str_to_bytes)� UnpickleableExceptionWrapper�subclass_exception�find_pickleable_exception�create_exception_cls�get_pickleable_exception�get_pickleable_etype�get_pickled_exception� strtoboolFT)�false�no�0�true�yes�1�on�offc C s t | |fd|i�S )zCreate new exception class.�
__module__)�type)�name�parent�module� r �D/usr/local/lib/python3.9/site-packages/celery/utils/serialization.pyr ! s r c C sV t | dg �}t| jt�D ]8}z||� }|||�� W n tyF Y q0 | S qdS )a� Find first pickleable exception base class.
With an exception instance, iterate over its super classes (by MRO)
and find the first super exception that's pickleable. It does
not go below :exc:`Exception` (i.e., it skips :exc:`Exception`,
:class:`BaseException` and :class:`object`). If that happens
you should use :exc:`UnpickleableException` instead.
Arguments:
exc (BaseException): An exception instance.
loads: decoder to use.
dumps: encoder to use
Returns:
Exception: Nearest pickleable parent exception class
(except :exc:`Exception` and parents), or if the exception is
pickleable it will return :const:`None`.
�argsN)�getattr�itermro� __class__�unwanted_base_classes� Exception)�exc�loads�dumps�exc_argsZsuperclsZsuperexcr r r r & s r c s t � fdd�t| ��S )Nc s | � vS �Nr )�sup��stopr r �<lambda>F � zitermro.<locals>.<lambda>)r r )�clsr. r r- r r# E s r# c C s |st }t| ||�S )z&Dynamically create an exception class.)r& r )r r r r r r r
I s r
c C sN g }| D ]<}z||� |� |� W q tyB |� t|�� Y q0 qt|�S )a Ensure items will serialize.
For a given list of arbitrary objects, return the object
or a string representation, safe for serialization.
Arguments:
items (Iterable[Any]): Objects to serialize.
encoder (Callable): Callable function to serialize with.
)�appendr& r �tuple)�items�encoder�
safe_exc_args�argr r r �ensure_serializableP s
r8 c sJ e Zd ZdZdZdZdZd � fdd� Zdd� Zdd� Z e
d d
� �Z � Z S ) r
at Wraps unpickleable exceptions.
Arguments:
exc_module (str): See :attr:`exc_module`.
exc_cls_name (str): See :attr:`exc_cls_name`.
exc_args (Tuple[Any, ...]): See :attr:`exc_args`.
Example:
>>> def pickle_it(raising_function):
... try:
... raising_function()
... except Exception as e:
... exc = UnpickleableExceptionWrapper(
... e.__class__.__module__,
... e.__class__.__name__,
... e.args,
... )
... pickle.dumps(exc) # Works fine.
Nc s< t |dd� �}|| _|| _|| _|| _t� �||||� d S )Nc S s t �t �| ��S r+ )�pickler( r) )�vr r r r/ � r0 z7UnpickleableExceptionWrapper.__init__.<locals>.<lambda>)r8 �
exc_module� exc_cls_namer* �text�super�__init__)�selfr; r< r* r= r6 �r$ r r r? � s � �z%UnpickleableExceptionWrapper.__init__c C s t | j| j�| j� S r+ )r
r< r; r* �r@ r r r �restore� s
��z$UnpickleableExceptionWrapper.restorec C s | j S r+ )r= rB r r r �__str__� s z$UnpickleableExceptionWrapper.__str__c C s< | |j j|j jt|dg �t|��}t|d�r8|�|j�}|S )Nr! �
__traceback__)r$ r �__name__r" r �hasattr�with_tracebackrE )r1 r' �resr r r �from_exception� s
�
z+UnpickleableExceptionWrapper.from_exception)N)
rF r � __qualname__�__doc__r; r< r* r? rC rD � classmethodrJ �
__classcell__r r rA r r
d s r
c C sF zt �t �| �� W n ty& Y n0 | S t| �}|r<|S t�| �S )z"Make sure exception is pickleable.)r9 r( r) r& r r
rJ )r' Znearestr r r r � s r c C s0 z||| �� W n t y& t Y S 0 | S dS )z Get pickleable exception type.N)r& )r1 r( r) r r r r � s
r c C s t | t�r| �� S | S )z,Reverse of :meth:`get_pickleable_exception`.)�
isinstancer
rC )r' r r r r � s
r c C s t tt| ���S r+ )r � base64encoder ��sr r r r � s r c C s t t| ��S r+ )� base64decoder rQ r r r r � s r c C sL |du r t }t| t�rHz|| �� W S tyF td| �d���Y n0 | S )z_Convert common terms for true/false to bool.
Examples (true/false/yes/no/on/off/1/0).
NzCannot coerce z
to type bool)�STRTOBOOL_DEFAULT_TABLErO �str�lower�KeyError� TypeError)Zterm�tabler r r r � s
r c C s� t | tj�rP| �� }| jr2|d d� |dd � }|�d�rL|d d� d }|S t | tj�rz| �� }| jrv|d d� }|S | �� S d S )N� � z+00:00i�����Z� )rO �datetime� isoformat� microsecond�endswith�time)�dt�rr r r �_datetime_to_json� s
re c s� ddl m} tt||�|d�� t| |�r4| jdd�} | du sFt| |�rJ| S t| ttf�rj� fdd�| D �S t| t�r�� �fd d
�| � � D �S t| t
j t
j f�r�t
| �S t| t
j�r�t| �S |du r�td t| ��d | �d
|� d���|| �S dS )z;Transform object making it suitable for json serialization.r )�Object)�
builtin_types�key� keyfilter�unknown_type_filterT)ZrecurseNc s g | ] }� |��qS r r )�.0r: )�_jsonifyr r �
<listcomp>� r0 zjsonify.<locals>.<listcomp>c s, i | ]$\}}�r�|�rn |� ||d��qS )� )rh r )rk �kr: �rl ri r r �
<dictcomp>� s �zjsonify.<locals>.<dictcomp>zUnsupported type: � z
(parent: �))Zkombu.abstractrf r �jsonifyrO Zas_dictr3 �list�dictr4 r^ �daterb re � timedeltarU �
ValueErrorr )�objrg rh ri rj Z
KombuDictTyper rp r rt � s0 �
� �rt c C s. t �� }|s| �n|d | u r � | |d �d S )Nrn )�sys�exc_info)r' r| r r r �raise_with_context s r} )N)N)-rL r^ Znumbersr{ �base64r rS r rP � functoolsr �inspectr � itertoolsr Zkombu.utils.encodingr r r �cPickler9 � ImportError�__all__r& �
BaseException�objectr% rT r r( r) r r# r
r8 r
r r r r re �RealrU rt r} r r r r �<module> sN
��
=
�
"