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
��Sh6# � @ s� d Z ddlZddlmZ ddlmZ ddlmZ ddlm Z z$ddl
Z
ddl Z
ddl Z
ddl
Z
W n eyv dZ
Y n0 dZee�Zd Zd
Zd Zd Zd
ZdZdZdZdd� ZG dd� de �ZdS )z@Apache Cassandra result store backend using the DataStax driver.� N)�states)�ImproperlyConfigured)�
get_logger� )� BaseBackend)�CassandraBackendz
You need to install the cassandra-driver library to
use the Cassandra backend. See https://github.com/datastax/python-driver
z�
CASSANDRA_AUTH_PROVIDER you provided is not a valid auth_provider class.
See https://datastax.github.io/python-driver/api/cassandra/auth.html.
z(Cassandra backend improperly configured.z!Cassandra backend not configured.z�
INSERT INTO {table} (
task_id, status, result, date_done, traceback, children) VALUES (
%s, %s, %s, %s, %s, %s) {expires};
z]
SELECT status, result, date_done, traceback, children
FROM {table}
WHERE task_id=%s
LIMIT 1
z�
CREATE TABLE {table} (
task_id text,
status text,
result blob,
date_done timestamp,
traceback blob,
children blob,
PRIMARY KEY ((task_id), date_done)
) WITH CLUSTERING ORDER BY (date_done DESC);
z
USING TTL {0}
c C s
t | d�S )N�utf8)�bytes)�x� r �C/usr/local/lib/python3.9/site-packages/celery/backends/cassandra.py�buf_tC s r
c sb e Zd ZdZdZdZdZd� fdd� Zddd�Zdd d
�Z dd d �Z
d
d� Z d� fdd� Z � Z
S )r aG Cassandra/AstraDB backend utilizing DataStax driver.
Raises:
celery.exceptions.ImproperlyConfigured:
if module :pypi:`cassandra-driver` is not available,
or not-exactly-one of the :setting:`cassandra_servers` and
the :setting:`cassandra_secure_bundle_path` settings is set.
NTc s� t � jf i |�� ts tt��| jj}|p4|�dd �| _|pF|�dd �| _ |p\|�dd �p\d| _
|pn|�dd �| _ |p�|�dd �| _ |�di �| _
| jp�| j } | r�| j r�| j s�tt��| jr�| j r�tt��|p�|�dd �}
|
d ur�t�|
�nd | _|�d
�p�d } |�d ��pd } ttj| tjj�| _ttj| tjj�| _d | _|�d
d �}
|�dd �}|
�r�|�r�ttj|
d �}|�s~tt��|f i |��| _d | _d | _ d | _ d | _ t � � | _!d S )NZcassandra_serversZ cassandra_secure_bundle_pathZcassandra_portiR# Zcassandra_keyspaceZcassandra_table�cassandra_optionsZcassandra_entry_ttl� Zcassandra_read_consistency� LOCAL_QUORUMZcassandra_write_consistencyZcassandra_auth_providerZcassandra_auth_kwargs)"�super�__init__� cassandrar �E_NO_CASSANDRA�app�conf�get�servers� bundle_path�port�keyspace�tabler �E_CASSANDRA_NOT_CONFIGURED�E_CASSANDRA_MISCONFIGURED� Q_EXPIRES�format�
cqlexpires�getattrZConsistencyLevelr �read_consistency�write_consistency�
auth_provider�auth�!E_NO_SUCH_CASSANDRA_AUTH_PROVIDER�_cluster�_session� _write_stmt�
_read_stmt� threading�RLock�_lock)�selfr r r Z entry_ttlr r �kwargsr Z
db_directions�expiresZ read_consZ
write_consr% Z auth_kwargsZauth_provider_class�� __class__r r r X sV � ��� zCassandraBackend.__init__Fc C s� | j durdS | j�� �z`�z| j dur<W W | j�� dS | jrhtjj| jf| j| j d�| j
��| _ n$tjjf d| j i| j d�| j
��| _ | j �
| j�| _ tj�tj| j| jd��| _| j| j_tj�tj| jd��| _| j| j_|�r0tj�tj| jd��}| j|_z| j � |� W n tj �y. Y n0 W n: tj �yl | j du�rZ| j � � d| _ d| _ � Y n0 W | j�� n | j�� 0 dS )zjPrepare the connection for action.
Arguments:
write (bool): are we a writer?
N)r r% Zsecure_connect_bundle)Zcloudr% )r r1 )r ) r) r. �acquire�releaser r ZclusterZClusterr r% r r( r �connectr �queryZSimpleStatement�Q_INSERT_RESULTr r r! r* r$ Zconsistency_level�Q_SELECT_RESULTr+ r# �Q_CREATE_RESULT_TABLE�executeZ
AlreadyExistsZOperationTimedOut�shutdown)r/ �writeZ make_stmtr r r �_get_connection� sf
;
����
�����
�
�
z CassandraBackend._get_connectionc K sV | j dd� | j�| j||t| �|��| j�� t| �|��t| �| �|���f� dS )z1Store return value and state of an executed task.T)r= N) r> r) r; r* r
�encoder �nowZcurrent_task_children)r/ �task_id�result�state� traceback�requestr0 r r r �
_store_result� s
�z CassandraBackend._store_resultc C s dS )Nz cassandra://r )r/ Zinclude_passwordr r r �as_uri� s zCassandraBackend.as_uric
C sf | � � | j�| j|f��� }|s.tjdd�S |\}}}}}| �||| �|�|| �|�| �|�d��S )z$Get task meta-data for a task by id.N)�statusrB )rA rH rB � date_donerD �children) r> r) r; r+ Zoner �PENDINGZmeta_from_decoded�decode)r/ rA �resrH rB rI rD rJ r r r �_get_task_meta_for� s �z#CassandraBackend._get_task_meta_forr c s2 |si n|}|� | j| j| jd�� t� �||�S )N)r r r )�updater r r r �
__reduce__)r/ �argsr0 r2 r r rP � s ��zCassandraBackend.__reduce__)NNNNNN)F)NN)T)r N)�__name__�
__module__� __qualname__�__doc__r r Zsupports_autoexpirer r> rF rG rN rP �
__classcell__r r r2 r r G s
�6
I �
r ) rU r, Zceleryr Zcelery.exceptionsr Zcelery.utils.logr �baser r Zcassandra.authZcassandra.clusterZcassandra.query� ImportError�__all__rR �loggerr r' r r r8 r9 r: r r
r r r r r �<module> s0