File: //usr/local/lib/python3.9/site-packages/celery/fixups/__pycache__/django.cpython-39.pyc
a
X>h � @ sL d Z ddlZddlZddlZddlmZmZ ddlmZ ddlm Z m
Z
m Z m Z m
Z
mZ ddlmZ ddlmZ ddlmZmZ dd lmZmZ e
r�dd
lmZ dd lmZ dd l m Z dd
l m Z ddl m!Z! ddl"m#Z# G dd� de�Z$dZ%dZ&e dd�dd�Z'ddd�dd�Z(d$d e)e
d d �d d �Z*G d!d � d �Z+G d"d#� d#�Z,dS )%z Django-specific customization.� N)�datetime�timezone)�
import_module)�IO�
TYPE_CHECKING�Any�List�Optional�cast��symbol_by_name)�cached_property)�_state�signals)� FixupWarning�ImproperlyConfigured)�
ModuleType)�Protocol)�BaseDatabaseWrapper)�ConnectionHandler)�Celery)�Taskc @ s e Zd ZU eed< dS )�DjangoDBModule� connectionsN)�__name__�
__module__� __qualname__r �__annotations__� r r �>/usr/local/lib/python3.9/site-packages/celery/fixups/django.pyr s
r )� DjangoFixup�fixupzpEnvironment variable DJANGO_SETTINGS_MODULE is defined
but Django isn't installed. Won't apply Django fix-ups!
)�fh�returnc
C s0 zt �| �� � W n tttfy* Y n0 d S �N)�os�close�fileno�AttributeError�OSError� TypeError)r" r r r �_maybe_close_fd% s r+ r )�djangor# c C s | j dk rtd��d S )N)� � z)Celery 5.x requires Django 1.11 or later.)�VERSIONr )r, r r r �_verify_django_version- s
r0 �DJANGO_SETTINGS_MODULEr r )�app�envr# c C sd t j�|�}|r`d| j�� vr`z ddl}W n tyJ t�t t
�� Y n0 t |� t | ��
� S dS )z;Install Django fixup if settings module environment is set.r, r N)r% �environ�get�
loader_cls�lowerr, � ImportError�warnings�warnr �ERR_NOT_INSTALLEDr0 r �install)r2 r3 ZSETTINGS_MODULEr, r r r r! 2 s r! c @ s� e Zd ZdZdd�dd�Zd d�dd�Zed d�d
d ��Zejd d d
�dd ��Ze d d�dd�Z
e d d�dd�Z d e e
d�dd�Zee d�dd�Zee
d�dd��Zd S ) r z"Fixup installed when using Django.r )r2 c C s$ || _ tjd u r| j �� d | _d S r$ )r2 r Z default_app� set_default�
_worker_fixup��selfr2 r r r �__init__C s
zDjangoFixup.__init__�r# c C sX t j�dt�� � td�| _| j| jj _| jj
s8d| j_ t j
�| j� t j�| j� | S )Nr �django.conf:settingsz%celery.contrib.django.task:DjangoTask)�sys�path�insertr% �getcwdr � _settings�nowr2 �loaderZ_custom_task_cls_usedZtask_clsr Zimport_modules�connect�on_import_modulesZ worker_init�on_worker_init�r@ r r r r< I s
zDjangoFixup.install�DjangoWorkerFixupc C s | j d u rt| j�| _ | j S r$ )r> rO r2 rN r r r � worker_fixupY s
zDjangoFixup.worker_fixupN)�valuer# c C s
|| _ d S r$ )r> )r@ rQ r r r rP _ s ��kwargsr# c K s | j �� d S r$ )rP �validate_models�r@ rS r r r rL c s z DjangoFixup.on_import_modulesc K s | j �� d S r$ )rP r< rU r r r rM g s zDjangoFixup.on_worker_initF)�utcr# c C s |rt �tj�S | �� S r$ )r rI r rV �_now)r@ rV r r r rI j s zDjangoFixup.nowc C s ddl m} dd� |�� D �S )Nr )�appsc S s g | ]
}|j �qS r )�name)�.0�configr r r �
<listcomp>o � z2DjangoFixup.autodiscover_tasks.<locals>.<listcomp>)Z django.appsrX Zget_app_configs)r@ rX r r r �autodiscover_tasksm s z DjangoFixup.autodiscover_tasksc C s t d�S )Nzdjango.utils.timezone:nowr rN r r r rW q s zDjangoFixup._now)F)r r r �__doc__rA r< �propertyrP �setterr rL rM �boolr rI r �strr^ r
rW r r r r r @ s c @ s� e Zd ZdZddd�dd�Zdd�dd �Zdd�d
d �Zd d�d d
�Zedd�dd�Z ddd�dd�Z
dedd�dd�Z dedd�dd�Z edd�dd �Z
d#edd �d d �Zdd�d!d"�ZdS )$rO r r N)r2 r# c C sV || _ | j j�dd �| _tdtd��| _td�| _td�| _ td�f| _
td�| _ d S )NZCELERY_DB_REUSE_MAXr z django.dbzdjango.core.cacherC z django.db.utils.InterfaceErrorzdjango.db:DatabaseError) r2 �confr5 � db_reuse_maxr
r �_db�_cacher rH �interface_errors�
DatabaseErrorr? r r r rA y s
�zDjangoWorkerFixup.__init__rB c C s dd l }|�� d S )Nr )r, �setup)r@ r, r r r � django_setup� s z DjangoWorkerFixup.django_setupc C s* ddl m} | �� tj�d�s&|� d S )Nr )�
run_checksZCELERY_SKIP_CHECKS)Zdjango.core.checksrl rk r% r4 r5 )r@ rl r r r rT � s z!DjangoWorkerFixup.validate_modelsc C sL t j�| j� t j�| j� t j�| j� t j�| j � | �� | �
� | S r$ ) r Zbeat_embedded_initrK �close_databaseZ task_prerun�on_task_prerunZ task_postrun�on_task_postrunZworker_process_init�on_worker_process_init� close_cacherN r r r r< � s zDjangoWorkerFixup.installrR c K sR t j�d�r| �� | jj�� D ]}|r |jr | �|� q | j dd� | �
� d S )NZFORKED_BY_MULTIPROCESSINGT)�force) r% r4 r5 rT rf r �all�
connection�_maybe_close_db_fd�_close_databaserq )r@ rS �cr r r rp � s
z(DjangoWorkerFixup.on_worker_process_initr )rw r# c C sN z4|j � t|j� W d � n1 s(0 Y W n | jyH Y n0 d S r$ )Zwrap_database_errorsr+ rt rh )r@ rw r r r ru � s
,z$DjangoWorkerFixup._maybe_close_db_fdr )�senderrS r# c K s t |jdd�s| �� dS )zCalled before every task.�is_eagerFN)�getattr�requestrm �r@ rx rS r r r rn � s z DjangoWorkerFixup.on_task_prerunc K s" t |jdd�s | �� | �� d S )Nry F)rz r{ rm rq r| r r r ro � s z!DjangoWorkerFixup.on_task_postrunc K s> | j s| �� S | j| j d kr,d| _| �� | jd7 _d S )N� r r- )re rv � _db_recyclesrU r r r rm � s z DjangoWorkerFixup.close_databaseF)rr r# c C s� | j j�� D ]r}z|r |�� n|�� W q | jy> Y q | jy| } z&t|�}d|vrhd|vrh� W Y d }~q d }~0 0 q d S )N�closedz
not connected)rf r rs r& Z close_if_unusable_or_obsoleterh ri rc )r@ rr �conn�excZstr_excr r r rv � s
z!DjangoWorkerFixup._close_databasec C s* z| j �� W n ttfy$ Y n0 d S r$ )rg Z close_cachesr* r( rN r r r rq � s z DjangoWorkerFixup.close_cache)F)r r r r~ rA rk rT r< r rp ru rn ro rm rb rv rq r r r r rO v s rO )r1 )-r_ r% rD r9 r r � importlibr �typingr r r r r r
Zkombu.utils.importsr Zkombu.utils.objectsr
Zceleryr r Zcelery.exceptionsr r �typesr r Z django.db.backends.base.baser Zdjango.db.utilsr Zcelery.app.baser Zcelery.app.taskr r �__all__r; r+ r0 rc r! r rO r r r r �<module> s2 6