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
��Sh1 � @ s� d Z ddlmZ ddlmZ ddlmZ ddlmZ ddl m
Z
z ddl m Z
dd lmZ W n eyv d
Z
ZY n0 d ZG d d
� d
e
�Zd
S )z ArangoDb result store backend.� )� timedelta)�cached_property)�
_parse_url)�ImproperlyConfigured� )�KeyValueStoreBackend)�
connection)�
AQLQueryErrorN)�ArangoDbBackendc s� e Zd ZdZdZdZdZdZdZdZ dZ
dZ e Z
d� fdd � Zed
d � �Zed d
� �Zedd� �Zdd� Zdd� Zdd� Zdd� Zdd� Z� ZS )r
a� ArangoDb backend.
Sample url
"arangodb://username:password@host:port/database/collection"
*arangodb_backend_settings* is where the settings are present
(in the app.conf)
Settings should contain the host, port, username, password, database name,
collection name else the default will be chosen.
Default database name and collection name is celery.
Raises
------
celery.exceptions.ImproperlyConfigured:
if module :pypi:`pyArango` is not available.
z 127.0.0.1Z8529ZceleryN�httpFc sv t � j|i |�� tdu r"td��|| _|du rJd } } } } }} n6t|�\}
}}}} } } | du rrd }}n| �d�\}}| jj� dd�}
|
dur�t
|
t �s�td��ni }
|p�|
� d| j �| _ t
|p�|
� d| j��| _|
� d| j�| _|
� d | j�| _|�p|
� d
| j�| _|�p$|
� d | j�| _|�p:|
� d | j�| _| �pP|
� d
| j�| _dj| j| j | jd�| _d| _dS )z8Parse the url or load the settings from settings object.NzEYou need to install the pyArango library to use the ArangoDb backend.�/Zarangodb_backend_settingsz5ArangoDb backend settings should be grouped in a dict�host�port�
http_protocol�verify�database�
collection�username�passwordz {http_protocol}://{host}:{port})r r
r )�super�__init__�py_arango_connectionr �urlr �split�app�conf�get�
isinstance�dictr
�intr r r r r r r �format� arangodb_url� _connection)�selfr �args�kwargsr
r r r r r Z_schemaZdatabase_collectionZ_query�config�� __class__� �B/usr/local/lib/python3.9/site-packages/celery/backends/arangodb.pyr 6 sF ��
�� �zArangoDbBackend.__init__c C s, | j du r&tj| j| j| j| jd�| _ | j S )z Connect to the arangodb server.N)Z arangoURLr r r )r" r �
Connectionr! r r r �r# r) r) r* r e s
�zArangoDbBackend.connectionc C s | j | j S )z&Database Object to the given database.)r r r, r) r) r* �dbo s zArangoDbBackend.dbc C s t | jd u rdn| jd�S )Nr )�seconds)r �expiresr, r) r) r* �
expires_deltat s z ArangoDbBackend.expires_deltac C s> |d u r d S | j jdd| j|d�d�}t|�dkr:t|�S d S )Nz(RETURN DOCUMENT(@@collection, @key).taskT�� @collection�key�Z
rawResults�bindVarsr )r- �AQLQueryr �len�next)r# r3 �queryr) r) r* r x s ��zArangoDbBackend.getc C s | j jd| j||d�d� d S )Nz�
UPSERT {_key: @key}
INSERT {_key: @key, task: @value}
UPDATE {task: @value} IN @@collection
)r2 r3 �value�r5 �r- r6 r )r# r3 r: r) r) r* �set� s ��zArangoDbBackend.setc c sl |d u r d S | j jdd| jt|t�r(|nt|�d�d�}|E d H z |�� W q8 tyd Y qhY q80 q8d S )Nz4FOR k IN @keys RETURN DOCUMENT(@@collection, k).taskT)r2 �keysr4 )r- r6 r r �listZ nextBatch�
StopIteration)r# r> r9 r) r) r* �mget� s ��
zArangoDbBackend.mgetc C s( |d u r d S | j jd| j|d�d� d S )Nz#REMOVE {_key: @key} IN @@collectionr1 r; r< )r# r3 r) r) r* �delete� s ��zArangoDbBackend.deletec C s: | j s
d S | j�� | j �� }| jjd| j|d�d� d S )Nz�
FOR record IN @@collection
FILTER record.task.date_done < @checkpoint
REMOVE record IN @@collection
)r2 �
checkpointr; )r/ r �nowr0 � isoformatr- r6 r )r# rC r) r) r* �cleanup� s ��zArangoDbBackend.cleanup)N)�__name__�
__module__� __qualname__�__doc__r
r r r r r r r �strZkey_tr �propertyr r- r r0 r r= rA rB rF �
__classcell__r) r) r' r* r
s, /
r
)rJ �datetimer Zkombu.utils.objectsr Zkombu.utils.urlr Zcelery.exceptionsr �baser ZpyArangor r ZpyArango.theExceptionsr � ImportError�__all__r
r) r) r) r* �<module> s