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/database/__pycache__/ |
a
��Sh� � @ s� d Z ddlZddlmZ ddlmZ ddlmZ ddlm Z ddl
m Z ddl m
Z
zdd lmZ W n ey� dd lmZ Y n0 e� Zd
Zd Zd d
� ZG dd� d�ZdS )zSQLAlchemy session.� N)�register_after_fork)�
create_engine)�
DatabaseError)� sessionmaker)�NullPool)� get_exponential_backoff_interval)�declarative_base)�SessionManager�
c C s | � � d S �N)� _after_fork)�session� r �J/usr/local/lib/python3.9/site-packages/celery/backends/database/session.py�_after_fork_cleanup_session s r c @ sB e Zd ZdZdd� Zdd� Zdd� Zdd d
�Zd d � Zd
d� Z dS )r zManage SQLAlchemy sessions.c C s. i | _ i | _d| _d| _td ur*t| t� d S )NF)�_engines� _sessions�forked�preparedr r ��selfr r r �__init__ s zSessionManager.__init__c C s
d| _ d S )NT)r r r r r r ( s zSessionManager._after_forkc K sp | j rFz | j| W S tyB t|fi |�� }| j|< | Y S 0 n&dd� |�� D �}t|fdti|��S d S )Nc S s i | ]\}}|� d �s||�qS )�pool)�
startswith)�.0�k�vr r r �
<dictcomp>3 s
�z-SessionManager.get_engine.<locals>.<dictcomp>Z poolclass)r r �KeyErrorr �itemsr )r �dburi�kwargs�enginer r r �
get_engine+ s zSessionManager.get_engineFc K sR | j |fi |��}| jrD|s&|| jvr6t|d�| j|< || j| fS |t|d�fS )N)�bind)r# r r r )r r Zshort_lived_sessionsr! r" r r r �create_session7 s z SessionManager.create_sessionc C sl | j shd}ztj�|� W qb ty\ |tk rVtd|dd�}t�|d � |d7 }n� Y q
0 qbq
d| _ d S )Nr r
i� T� ) r �ResultModelBase�metadataZ
create_allr �PREPARE_MODELS_MAX_RETRIESr �time�sleep)r r" �retriesZsleep_amount_msr r r �prepare_models? s �
z SessionManager.prepare_modelsc K s& | j |fi |��\}}| �|� |� S r )r% r- )r r r! r" r
r r r �session_factoryV s
z SessionManager.session_factoryN)F)
�__name__�
__module__� __qualname__�__doc__r r r# r% r- r. r r r r r s
r )r2 r* Zkombu.utils.compatr Z
sqlalchemyr Zsqlalchemy.excr Zsqlalchemy.ormr Zsqlalchemy.poolr Zcelery.utils.timer r � ImportErrorZsqlalchemy.ext.declarativer' �__all__r) r r r r r r �<module> s