Server : LiteSpeed System : Linux barito.iixcp.rumahweb.net 5.14.0-611.49.1.el9_7.x86_64 #1 SMP PREEMPT_DYNAMIC Tue Apr 21 16:39:08 EDT 2026 x86_64 User : elvh3918 ( 1528) PHP Version : 8.2.31 Disable Function : mail Directory : /usr/local/lib/python3.9/site-packages/celery/backends/__pycache__/ |
a
��Sh�, � @ s d Z ddlmZmZmZ ddlmZ ddlmZ ddlm Z m
Z
ddl m Z ddl
mZ dd lmZ z dd
lZW n ey� d
ZY n0 er�zdd lmZ W n ey� dd lmZ Y n0 dd lmZ nd
ZG d
d� de�ZdZeddg�ZG dd� de�Z d
S )z MongoDB result store backend.� )�datetime� timedelta�timezone)� EncodeError)�cached_property)�maybe_sanitize_url�urlparse)�states)�ImproperlyConfigured� )� BaseBackendN)�Binary)�InvalidDocumentc @ s e Zd ZdS )r N)�__name__�
__module__� __qualname__� r r �A/usr/local/lib/python3.9/site-packages/celery/backends/mongodb.pyr s r )� MongoBackend�pickle�msgpackc s e Zd ZdZdZdZdZdZdZdZ dZ
dZ dZ dZ
d ZdZd3� fd
d � Zed d
� �Zdd� Zdd� Z� fdd�Z� fdd�Zd4dd�Zdd� Zdd� Zd d � Zd d � Zd d!� Z d"d#� Z d5� fd%d&� Z d'd(� Z e d)d*� �Z!e d+d,� �Z"e d-d.� �Z#e d/d0� �Z$d6d1d2�Z%� Z&S )7r z�MongoDB result backend.
Raises:
celery.exceptions.ImproperlyConfigured:
if module :pypi:`pymongo` is not available.
N� localhosti�i �celeryZcelery_taskmetaZcelery_groupmeta�
Fc s� i | _ t� j|fi |�� ts&td��| �� �� D ]\}}| j �||� q2| jr�| � | j�| _tj
� | j�}dd� |d D �}|d | _ |d | _
|| _|d r�|d | _| j �|d � | jj�d �}|d u�r�t|t�s�td
��t|�}d |v s�d |v �rd | _|�d | j�| _|�d | j�| _|�d
| j�| _|�d| j �| _ |�d| j
�| _
|�d| j�| _|�d| j�| _|�d| j�| _| j �|�di �� | j �|� d S )NzCYou need to install the pymongo library to use the MongoDB backend.c S s" g | ]}|d � d|d � ��qS )r �:r r )�.0�xr r r �
<listcomp>N s z)MongoBackend.__init__.<locals>.<listcomp>Znodelist�username�password�database�optionsZmongodb_backend_settingsz4MongoDB backend settings should be grouped in a dict�host�port�
mongo_host�user�taskmeta_collection�groupmeta_collection)r! �super�__init__�pymongor
�_prepare_client_options�items�
setdefault�url� _ensure_mongodb_uri_complianceZ
uri_parser� parse_urir% r r$ �
database_name�update�app�conf�get�
isinstance�dict�popr" r# r& r' )�selfr3 �kwargs�key�valueZuri_dataZ hostslist�config�� __class__r r r) : sV ��
���zMongoBackend.__init__c C s2 t | �}|j�d�s d| � �} | dkr.| d7 } | S )NZmongodbzmongodb+�
mongodb://r )r �scheme�
startswith)r. �
parsed_urlr r r r/ v s
z+MongoBackend._ensure_mongodb_uri_compliancec C s$ t jdkrd| jiS | jdd�S d S )N)� Z maxPoolSizeF)�
max_pool_sizeZauto_start_request)r* Z
version_tuplerE �r9 r r r r+ � s
�z$MongoBackend._prepare_client_optionsc C s� | j du r�ddlm} | j}|sL| j}t|t�rL|�d�sLd|� d| j� �}t | j
�}||d< | j rn| j |d< | j r~| j |d< |f i |��| _ | j S ) z Connect to the MongoDB server.Nr )� MongoClientr@ r r" r r )
� _connectionr* rG r$ r" r6 �strrB r# r7 r! r% r )r9 rG r" r4 r r r �_get_connection� s"
�
z MongoBackend._get_connectionc s0 | j dkr|S t� �|�}| j tv r,t|�}|S �NZbson)�
serializerr( �encode�
BINARY_CODECSr
)r9 �data�payloadr> r r rM � s
zMongoBackend.encodec s | j dkr|S t� �|�S rK )rL r( �decode)r9 rO r> r r rQ � s
zMongoBackend.decodec
K sn | j | �|�|||dd�}||d< z| jjd|i|dd� W n, tyh } zt|��W Y d}~n
d}~0 0 |S )z1Store return value and state of an executed task.F)�result�state� traceback�requestZ format_date�_idT�ZupsertN)Z_get_result_metarM �
collection� replace_oner r ) r9 �task_idrR rS rT rU r: �meta�excr r r �
_store_result� s � zMongoBackend._store_resultc C s� | j �d|i�}|r�| jj�dd�r~| �|d |d |d |d |d |d |d |d
|d |d |d
| �|d �d� �S | �|d |d | �|d �|d |d
|d d��S tjdd�S )z$Get task meta-data for a task by id.rV �extendedrR �name�args�queuer: �status�worker�retries�children� date_donerT ) r_ r` rZ ra r: rb rc rd re rf rT rR )rZ rb rR rf rT re N)rb rR ) rX �find_oner3 r4 Zfind_value_for_keyZmeta_from_decodedrQ r �PENDING)r9 rZ �objr r r �_get_task_meta_for� s4 � �z MongoBackend._get_task_meta_forc C s> || � dd� |D ��t�tj�d�}| jjd|i|dd� |S )zSave the group result.c S s g | ]
}|j �qS r )�id)r �ir r r r � � z,MongoBackend._save_group.<locals>.<listcomp>)rV rR rf rV TrW )rM r �nowr �utc�group_collectionrY )r9 �group_idrR r[ r r r � _save_group� s
�zMongoBackend._save_groupc sD � j �d|i�}|r@|d |d � fdd�� �|d �D �d�S dS )z!Get the result for a group by id.rV rf c s g | ]}� j �|��qS r )r3 Z AsyncResult)r �taskrF r r r � s �z/MongoBackend._restore_group.<locals>.<listcomp>rR )rZ rf rR N)rp rg rQ )r9 rq ri r rF r �_restore_group� s
��zMongoBackend._restore_groupc C s | j �d|i� dS )zDelete a group by id.rV N)rp �
delete_one)r9 rq r r r �
_delete_group� s zMongoBackend._delete_groupc C s | j �d|i� dS )z�Remove result from MongoDB.
Raises:
pymongo.exceptions.OperationsError:
if the task_id could not be removed.
rV N)rX ru )r9 rZ r r r �_forget� s
zMongoBackend._forgetc C sN | j s
dS | j�dd| j�� | j ii� | j�dd| j�� | j ii� dS )zDelete expired meta-data.Nrf z$lt)�expiresrX Z delete_manyr3 rn �
expires_deltarp rF r r r �cleanup s ��zMongoBackend.cleanupr c s( |si n|}t � �|t|| j| jd��S )N)rx r. )r( �
__reduce__r7 rx r. )r9 r` r: r> r r r{ s �zMongoBackend.__reduce__c C s | � � }|| j S )N)rJ r1 )r9 �connr r r �
_get_database s zMongoBackend._get_databasec C s | � � S )z]Get database from MongoDB connection.
performs authentication if necessary.
)r} rF r r r r s zMongoBackend.databasec C s | j | j }|jddd� |S �z"Get the meta-data task collection.rf T)Z
background)r r& � create_index�r9 rX r r r rX & s zMongoBackend.collectionc C s | j | j }|jddd� |S r~ )r r' r r� r r r rp 0 s z MongoBackend.group_collectionc C s t | jd�S )N)�seconds)r rx rF r r r ry : s zMongoBackend.expires_deltac C sL | j s
dS |r| j S d| j vr(t| j �S | j �dd�\}}d�t|�|g�S )z~Return the backend as an URI.
Arguments:
include_password (bool): Password censored if disabled.
r@ �,r )r. r �split�join)r9 Zinclude_passwordZuri1� remainderr r r �as_uri> s
zMongoBackend.as_uri)N)NN)r N)F)'r r r �__doc__r$ r" r# r% r r1 r& r' rE r! Zsupports_autoexpirerH r) � staticmethodr/ r+ rJ rM rQ r] rj rr rt rv rw rz r{ r} r r rX rp ry r� �
__classcell__r r r> r r # sN <
�
r ) r� r r r Zkombu.exceptionsr Zkombu.utils.objectsr Zkombu.utils.urlr r r r Zcelery.exceptionsr
�baser r* � ImportErrorZ bson.binaryr
Zpymongo.binaryZpymongo.errorsr � Exception�__all__� frozensetrN r r r r r �<module> s,