o h@s<dZddlZddlZddlZddlmZddlmZddlmZm Z m Z m Z m Z m Z ddlmZddlmZddlmZmZdd lmZmZe rndd lmZdd lmZdd lmZdd lmZddlm Z GdddeZ!dZ"dZ#deddfddZ$d%ddZ%d&ddde&de dfd d!Z'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__rrF/var/www/html/env/lib/python3.10/site-packages/celery/fixups/django.pyrs r) DjangoFixupfixupzpEnvironment variable DJANGO_SETTINGS_MODULE is defined but Django isn't installed. Won't apply Django fix-ups! fhreturnc Cs0z t|WdStttfyYdSwN)osclosefilenoAttributeErrorOSError TypeError)r rrr_maybe_close_fd$s r)djangorcCs|jdkr tddS)N) z)Celery 5.x requires Django 1.11 or later.)VERSIONr)r*rrr_verify_django_version,s r.DJANGO_SETTINGS_MODULEapprenvrcCsftj|}|r1d|jvr1zddl}Wnty&tt t YdSwt |t | SdS)z;Install Django fixup if settings module environment is set.r*rN)r#environget loader_clslowerr* ImportErrorwarningswarnrERR_NOT_INSTALLEDr.rinstall)r0r1SETTINGS_MODULEr*rrrr1s    rc@seZdZdZdddZdddZedd d Zejddd Zde dd fddZ de dd fddZ d de de fddZdeefddZede fddZd S)!rz"Fixup installed when using Django.r0rcCs$||_tjdur |jd|_dSr")r0r 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 _settingsnowr0loaderrimport_modulesconnecton_import_modules worker_initon_worker_initr@rrrr:Hs   zDjangoFixup.installDjangoWorkerFixupcCs|jdur t|j|_|jSr")r>rPr0rOrrr worker_fixupUs  zDjangoFixup.worker_fixupvalueNcCs ||_dSr")r>)r@rRrrrrQ[s kwargscK|jdSr")rQvalidate_modelsr@rSrrrrL_szDjangoFixup.on_import_modulescKrTr")rQr:rVrrrrNcszDjangoFixup.on_worker_initFutccCs|rtS|Sr")rutcnow_now)r@rWrrrrHfszDjangoFixup.nowcCsddlm}dd|DS)Nr)appscSsg|]}|jqSr)name).0configrrr ksz2DjangoFixup.autodiscover_tasks..) django.appsrZget_app_configs)r@rZrrrautodiscover_tasksis zDjangoFixup.autodiscover_taskscCstdS)Nzdjango.utils.timezone:nowr rOrrrrYmszDjangoFixup._now)r0r)r!rr!rP)rRrPr!NF)rrr__doc__rAr:propertyrQsetterrrLrNboolrrHrstrrar rYrrrrr?s    c@seZdZdZd"ddZd#dd Zd#d d Zd$d d ZdeddfddZ de ddfddZ dddeddfddZ dddeddfddZ deddfddZd%deddfddZd#d d!ZdS)&rPrr0rr!NcCsV||_|jjdd|_tdtd|_td|_td|_ tdf|_ td|_ dS)NCELERY_DB_REUSE_MAXrz django.dbzdjango.core.cacherBzdjango.db.utils.InterfaceErrorzdjango.db:DatabaseError) r0confr3 db_reuse_maxr r_db_cacher rGinterface_errors DatabaseErrorr?rrrrAus  zDjangoWorkerFixup.__init__cCsddl}|dS)Nr)r*setup)r@r*rrr django_setups zDjangoWorkerFixup.django_setupcCs.ddlm}|tjds|dSdS)Nr) run_checksCELERY_SKIP_CHECKS)django.core.checksrrrqr#r2r3)r@rrrrrrUs   z!DjangoWorkerFixup.validate_modelscCsLtj|jtj|jtj|jtj|j || |Sr") rbeat_embedded_initrKclose_database task_prerunon_task_prerun task_postrunon_task_postrunworker_process_initon_worker_process_init close_cacherOrrrr:szDjangoWorkerFixup.installrScKsTtjdr ||jjD] }|r|jr||jq|j dd| dS)NFORKED_BY_MULTIPROCESSINGT)force) r#r2r3rUrlrall connection_maybe_close_db_fd_close_databaser})r@rScrrrr|s     z(DjangoWorkerFixup.on_worker_process_initfdcCs&zt|WdS|jyYdSwr")r)rn)r@rrrrrs z$DjangoWorkerFixup._maybe_close_db_fdsenderrcKst|jdds |dSdS)zCalled before every task.is_eagerFN)getattrrequestrvr@rrSrrrrxs z DjangoWorkerFixup.on_task_preruncKs&t|jdds||dSdS)NrF)rrrvr}rrrrrzs z!DjangoWorkerFixup.on_task_postruncKs>|js|S|j|jdkrd|_||jd7_dS)Nrr+)rkr _db_recyclesrVrrrrvs z DjangoWorkerFixup.close_databaseFrc Cs|jjD]8}z |r|n|Wq|jyYq|jy>}zt|}d|vr4d|vr4WYd}~qd}~wwdS)Nclosedz not connected)rlrrr$close_if_unusable_or_obsoleternrorh)r@rconnexcstr_excrrrrs z!DjangoWorkerFixup._close_databasec Cs*z|jWdSttfyYdSwr")rm close_cachesr(r&rOrrrr}s zDjangoWorkerFixup.close_cache)r0rr!N)r!Nrbrc)rrrrrArqrUr:rr|rrrxrzrvrgrr}rrrrrPrs    rP)r*rr!N)r/)*rdr#rCr7r importlibrtypingrrrrrr kombu.utils.importsr kombu.utils.objectsr celeryr rcelery.exceptionsrrtypesrrdjango.db.utilsrcelery.app.basercelery.app.taskrr__all__r9r)r.rhrrrPrrrrs2           3