o Df#@sdZddlZddlZddlmZddlmZddlmZddl m Z ddl Z ddl m Z m Z ddlmZdd lmZdd lmZdd lmZdd lmZdd lmZzddlmZddlmZddlmZWne ywddZdZ!dZ"YnweZ!eZ"GdddZ#ddZ$Gddde j%Z&Gddde j'Z(Gddde&Z)Gddde(Z*Gdd d e Z+Gd!d"d"e Z,Gd#d$d$e Z-Gd%d&d&e Z.Gd'd(d(e Z/Gd)d*d*e j0Z1e,Z2e-Z3e.Z4e/Z5e1Z6e+Z7dS)+z Click customizations for Celery.N) OrderedDictupdate_wrapper)pformat)Any)Context ParamType)cached_property)get_current_app)user_preload_options)text)mlevel) maybe_iso8601) highlight)Terminal256Formatter) PythonLexercOs|S)z2Place holder function in case pygments is missing.)sargskwargsrrH/home/ubuntu/webapp/venv/lib/python3.10/site-packages/celery/bin/base.pyrsrc@s|eZdZdZdddZeddZeddZdd d Zdd d Z dddZ dddZ ddZ ddZ ddZdddZd S) CLIContextzContext Object for the CLI.FcCs"|pt|_||_||_||_dS)zInitialize the CLI context.N)r appno_colorquietworkdir)selfrrrrrrr__init__%s  zCLIContext.__init__cC|jddddS)NOKgreenTfgboldstylerrrrr,z CLIContext.OKcCr)NERRORredTr!r$r&rrrr(0r'zCLIContext.ERRORNcKs|jr|Stj|fi|SN)rclickr%rmessagerrrrr%4szCLIContext.stylecKs:|jrd|d<tj|fi|dStj|fi|dSNFcolorrr+echosechor,rrrr2:zCLIContext.sechocKs:|jrd|d<tj|fi|dStj|fi|dSr.)rr+r1r,rrrr1Ar3zCLIContext.echocKsBd|d<|jrd|d<tj|fi|dStj|fi|dS)NTerrFr/r0r,rrrerrorHs zCLIContext.errorcCst|tr |j||fSt|tr5d|vsd|vr||Stj|ddd}|js0t |t t }|j|fSt|t r?|j|fS|jt |fS)Nokr5T) sort_keysindent) isinstancelistr pretty_listdictpretty_dict_ok_errorjsondumpsrrLEXER FORMATTERstrr)rnrrrrprettyPs       zCLIContext.prettycs |sdSdfdd|DS)Nz - empty - c3s(|]}jdddd|VqdS)*white)r" Nr$).0itemr&rr bs z)CLIContext.pretty_list..)joinrrDrr&rr<_s zCLIContext.pretty_listcCsVz|jt||dddfWStyYnw|jt||dddfS)Nr6r7r5)rr r9rEKeyErrorr(rNrrrr>fs zCLIContext.pretty_dict_ok_errorcCsf|dkr |jr dS|j r|j|ddddpd}||d||r/|r1||dSdSdS)Nz<-rHTr!rIrQ)rr%r1)r directiontitlebody show_bodydirstrrrrsay_chatos"zCLIContext.say_chat)Fr*)rQF)__name__ __module__ __qualname____doc__rr rr(r%r2r1r5rEr<r>rWrrrrr"s        rcsfdd}t|S)z6Extract preload options and return a wrapped callable.cs\|jj}dd|jdgD}|r#fdd|D}tj||d|g|RiS)NcSsg|]}|jqSr)name)rJorrr }sz:handle_preload_options..caller..preloadcsi|]}||qSrr)rJpreload_optionrrr sz:handle_preload_options..caller..)senderroptions)objr user_optionsgetr send)ctxrrrpreload_optionsrffrarcallerzs z&handle_preload_options..callerr)rlrmrrkrhandle_preload_optionsxs  rnc,eZdZdZfddZfddZZS) CeleryOptionzCustomized option for Celery.cs.|jr |j|j|_tj|g|Ri|Sr*)default_value_from_contextredefaultsuper get_default)rrirr __class__rrrtszCeleryOption.get_defaultcs2|dd|_|dd|_tj|i|dS)zInitialize a Celery option. help_groupNrq)poprwrqrsrrrrrurrrszCeleryOption.__init__)rXrYrZr[rtr __classcell__rrrurrps rpc@seZdZdZddZdS) CeleryCommandzCustomized command for Celery.c Cst}||D])}||}|dur1t|dr(|jr(|t|jg|q|dg|q|D]\}}| | | |Wdn1sOwYq6dS)z7Write all the options into the formatter if they exist.NrwOptions) r get_paramsget_help_recordhasattrrw setdefaultrCappenditemssectionwrite_dl)rri formatteroptsparamrvr\ opts_grouprrrformat_optionss   zCeleryCommand.format_optionsN)rXrYrZr[rrrrrr{s r{cs:eZdZdZfddZdedededefdd ZZ S) DaemonOptionzCommon daemonization optioncs0tj|f|dd|d|jd|dS)NrwzDaemonization Optionscallback)rwr)rsrrxdaemon_settingryrurrrs    zDaemonOption.__init__rioptvaluereturncCs&|pt|jjj|jjd|jdS)z Try to fetch deamonization option from applications settings. Use the daemon command name as prefix (eg. `worker` -> `worker_pidfile`) _N)getattrrerconfcommandr\)rrirrrrrrs&zDaemonOption.daemon_setting) rXrYrZr[rrrprrrzrrrurrs "rcs eZdZdZfddZZS)CeleryDaemonCommandzDaemon commands.c s`tj|i||jtddddtdddtdddtd d dtd d dtd ddfdS)z7Initialize a Celery command with common daemon options.z --logfilez-fz#Log destination; defaults to stderr)helpz --pidfilez&PID file path; defaults to no PID filez--uidz Drops privileges to this user IDz--gidz!Drops privileges to this group IDz--umaskz,Create files and directories with this umaskz --executablez&Override path to the Python executableN)rsrparamsextendrryrurrrs       zCeleryDaemonCommand.__init__)rXrYrZr[rrzrrrurrsrc@eZdZdZdZddZdS)CommaSeparatedListzComma separated list argument.zcomma separated listcCs t|Sr*)r str_to_listrrrrirrrconverts zCommaSeparatedList.convertNrXrYrZr[r\rrrrrr rc@r) JsonArrayzJSON formatted array argument.z json arrayc Cnt|tr|Szt|}Wnty'}z |t|WYd}~nd}~wwt|ts5||d|S)Nz was not an array)r:r;r?loads ValueErrorfailrCrrrriverrrr  zJsonArray.convertNrrrrrrrrc@r) JsonObjectzJSON formatted object argument.z json objectc Cr)Nz was not an object)r:r=r?rrrrCrrrrrrzJsonObject.convertNrrrrrrrrc@r)ISO8601DateTimezISO 8601 Date Time argument.z iso-86091c Cs@zt|WSttfy}z ||WYd}~dSd}~wwr*)r TypeErrorrrrrrrirrrrrs  zISO8601DateTime.convertNrrrrrrrrc@r)ISO8601DateTimeOrFloatz%ISO 8601 Date Time or float argument.ziso-86091 or floatc Csbzt|WSttfyYnwzt|WSttfy0}z ||WYd}~dSd}~wwr*)floatrrrrrrrrrs  zISO8601DateTimeOrFloat.convertNrrrrrr rrcro)LogLevelzLog level option.cstddS)z:Initialize the log level option with the relevant choices.)DEBUGINFOWARNINGr(CRITICALFATALN)rsrr&rurrrr'zLogLevel.__init__cs0t|tjr|S|}t|||}t|Sr*)r:numbersIntegralupperrsrr rrurrr#s zLogLevel.convert)rXrYrZr[rrrzrrrurrs r)8r[r?r collectionsr functoolsrpprintrtypingrr+rrkombu.utils.objectsr celery._stater celery.signalsr celery.utilsr celery.utils.logr celery.utils.timerpygmentsrpygments.formattersrpygments.lexersr ImportErrorrArBrrnOptionrpCommandr{rrrrrrrChoicer JSON_ARRAY JSON_OBJECTISO8601ISO8601_OR_FLOAT LOG_LEVELCOMMA_SEPARATED_LISTrrrrsX             V