U .e@s<dZddlZddlZddlZddlmZddlmZddlmZm Z m Z m Z m Z m Z ddlmZddlmZddlmZmZdd lmZmZe rdd lmZdd lmZdd lmZdd lmZddlm Z GdddeZ!dZ"dZ#eddddZ$dddddZ%d#de&e ddddZ'Gd ddZ(Gd!d"d"Z)dS)$zDjango-specific customization.N)datetime) import_module)IO TYPE_CHECKINGAnyListOptionalcastsymbol_by_name)cached_property)_statesignals) FixupWarningImproperlyConfigured) ModuleType)Protocol)ConnectionHandler)Celery)Taskc@seZdZUeed<dS)DjangoDBModule connectionsN)__name__ __module__ __qualname__r__annotations__rr8/tmp/pip-unpacked-wheel-f4liivr4/celery/fixups/django.pyrs r) DjangoFixupfixupzpEnvironment variable DJANGO_SETTINGS_MODULE is defined but Django isn't installed. Won't apply Django fix-ups! )fhreturnc Cs2zt|Wntttfk r,YnXdSN)osclosefilenoAttributeErrorOSError TypeError)r rrr_maybe_close_fd$sr)r)djangor!cCs|jdkrtddS)N) z)Celery 5.x requires Django 1.11 or later.)VERSIONr)r*rrr_verify_django_version,s r.DJANGO_SETTINGS_MODULErr)appenvr!cCsftj|}|rbd|jkrbz ddl}Wn"tk rLtt t YnXt |t | SdS)z;Install Django fixup if settings module environment is set.r*rN)r#environget loader_clslowerr* ImportErrorwarningswarnrERR_NOT_INSTALLEDr.rinstall)r0r1ZSETTINGS_MODULEr*rrrr1s   rc@seZdZdZddddZddddZed dd d Zejd d d dd Ze d dddZ e d dddZ de e dddZeedddZee dddZd S)rz"Fixup installed when using Django.r)r0cCs$||_tjdkr|jd|_dSr")r0r Z default_app set_default _worker_fixupselfr0rrr__init__Bs  zDjangoFixup.__init__r!cCsHtjdttd|_|j|jj _t j |j t j |j|S)Nrdjango.conf:settings)syspathinsertr#getcwdr _settingsnowr0loaderrZimport_modulesconnecton_import_modulesZ worker_initon_worker_initr>rrrr:Hs   zDjangoFixup.installDjangoWorkerFixupcCs|jdkrt|j|_|jSr")r<rMr0rLrrr worker_fixupUs  zDjangoFixup.worker_fixupN)valuer!cCs ||_dSr")r<)r>rOrrrrN[skwargsr!cKs|jdSr")rNvalidate_modelsr>rQrrrrJ_szDjangoFixup.on_import_modulescKs|jdSr")rNr:rSrrrrKcszDjangoFixup.on_worker_initF)utcr!cCs|r tS|Sr")rutcnow_now)r>rTrrrrGfszDjangoFixup.nowcCsddlm}dd|DS)Nr)appscSsg|] }|jqSr)name).0configrrr ksz2DjangoFixup.autodiscover_tasks..)Z django.appsrWZget_app_configs)r>rWrrrautodiscover_tasksis zDjangoFixup.autodiscover_taskscCstdS)Nzdjango.utils.timezone:nowr rLrrrrVmszDjangoFixup._now)F)rrr__doc__r?r:propertyrNsetterrrJrKboolrrGrstrr\r rVrrrrr?s c@seZdZdZdddddZdddd Zddd d Zddd d ZeddddZ e ddddZ deddddZ deddddZ eddddZd"eddddZddd d!ZdS)#rMrrN)r0r!cCsV||_|jjdd|_tdtd|_td|_td|_ tdf|_ td|_ dS)NZCELERY_DB_REUSE_MAXrz django.dbzdjango.core.cacherAzdjango.db.utils.InterfaceErrorzdjango.db:DatabaseError) r0confr3 db_reuse_maxr r_db_cacher rFinterface_errors DatabaseErrorr=rrrr?us  zDjangoWorkerFixup.__init__r@cCsddl}|dS)Nr)r*setup)r>r*rrr django_setupszDjangoWorkerFixup.django_setupcCs*ddlm}|tjds&|dS)Nr) run_checksZCELERY_SKIP_CHECKS)Zdjango.core.checksrjrir#r2r3)r>rjrrrrRs  z!DjangoWorkerFixup.validate_modelscCsLtj|jtj|jtj|jtj|j || |Sr") rZbeat_embedded_initrIclose_databaseZ task_prerunon_task_prerunZ task_postrunon_task_postrunZworker_process_initon_worker_process_init close_cacherLrrrr:szDjangoWorkerFixup.installrPcKsTtjdr||jjD]}|r |jr ||jq |j dd| dS)NZFORKED_BY_MULTIPROCESSINGT)force) r#r2r3rRrdrall connection_maybe_close_db_fd_close_databasero)r>rQcrrrrns    z(DjangoWorkerFixup.on_worker_process_init)fdr!cCs(z t|Wn|jk r"YnXdSr")r)rf)r>rvrrrrss z$DjangoWorkerFixup._maybe_close_db_fdr)senderrQr!cKst|jdds|dS)zCalled before every task.is_eagerFN)getattrrequestrkr>rwrQrrrrlsz DjangoWorkerFixup.on_task_preruncKs"t|jdds||dS)NrxF)ryrzrkror{rrrrmsz!DjangoWorkerFixup.on_task_postruncKs>|js|S|j|jdkr,d|_||jd7_dS)Nrr+)rcrt _db_recyclesrSrrrrks z DjangoWorkerFixup.close_databaseF)rpr!c Cs|jjD]r}z|r |n|Wq |jk r@Yq |jk r|}zt|}d|krld|krlW5d}~XYq Xq dS)Nclosedz not connected)rdrrqr$Zclose_if_unusable_or_obsoleterfrgra)r>rpconnexcZstr_excrrrrts  z!DjangoWorkerFixup._close_databasec Cs,z|jWnttfk r&YnXdSr")reZ close_cachesr(r&rLrrrroszDjangoWorkerFixup.close_cache)F)rrrr}r?rirRr:rrnrrsrlrmrkr`rtrorrrrrMrs  rM)r/)*r]r#rBr7r importlibrtypingrrrrrr Zkombu.utils.importsr Zkombu.utils.objectsr Zceleryr rZcelery.exceptionsrrtypesrrZdjango.db.utilsrZcelery.app.baserZcelery.app.taskrr__all__r9r)r.rarrrMrrrrs0          3