U bh#@sdZddlZddlZddlmZddlmZddlmZddl m Z ddl Z ddl m Z m Z ddlmZdd lmZdd lmZdd lmZdd lmZdd lmZz(ddlmZddlmZddlmZWn$e k rddZdZ!dZ"YnXeZ!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.)sargskwargsrr3/tmp/pip-unpacked-wheel-kcem4wq5/celery/bin/base.pyrsrc@s|eZdZdZdddZeddZeddZdd d Zdd d Z dddZ dddZ ddZ ddZ ddZd ddZd S)! CLIContextzContext Object for the CLI.FcCs"|pt|_||_||_||_dS)zInitialize the CLI context.N)r appno_colorquietworkdir)selfrrrrrrr__init__%s zCLIContext.__init__cCs|jddddS)NOKZgreenTfgZboldstylerrrrr,sz CLIContext.OKcCs|jddddS)NERRORZredTrr!r#rrrr$0szCLIContext.ERRORNcKs|jr |Stj|f|SdSN)rclickr"rmessagerrrrr"4szCLIContext.stylecKs0|jrd|d<tj|f|ntj|f|dSNFcolorrr&echosechor'rrrr-:szCLIContext.sechocKs0|jrd|d<tj|f|ntj|f|dSr))rr&r,r'rrrr,AszCLIContext.echocKs8d|d<|jr&d|d<tj|f|ntj|f|dS)NTerrFr*r+r'rrrerrorHs zCLIContext.errorcCst|tr|j||fSt|trjd|ks4d|kr>||Stj|ddd}|js`t |t t }|j|fSt|t r~|j|fS|jt |fS)Nokr/T) sort_keysindent) isinstancelistr pretty_listdictpretty_dict_ok_errorjsondumpsrrLEXER FORMATTERstrr)rnrrrrprettyPs       zCLIContext.prettycs |sdSdfdd|DS)Nz - empty - c3s&|]}jdddd|VqdS)*white)r  Nr!).0itemr#rr bsz)CLIContext.pretty_list..)joinrr>rr#rr6_s zCLIContext.pretty_listcCsXz"|jt||dddfWStk r6YnX|jt||dddfS)Nr0r1r/)rr r3r?KeyErrorr$rHrrrr8fszCLIContext.pretty_dict_ok_errorcCs^|dkr|jrdS|j r0|j|ddddp2d}||d||rZ|rZ||dS)Nz<-rBTrrCrK)rr"r,)r directiontitlebodyZ show_bodyZdirstrrrrsay_chatos "zCLIContext.say_chat)F)N)N)N)N)rKF)__name__ __module__ __qualname____doc__rr rr$r"r-r,r/r?r6r8rOrrrrr"s        rcsfdd}t|S)z6Extract preload options and return a wrapped callable.csV|jj}dd|jdgD}|rFfdd|D}tj||d|f|S)NcSsg|] }|jqSr)name)rDorrr }sz:handle_preload_options..caller..Zpreloadcsi|]}||qSrr)rDZpreload_optionrrr sz:handle_preload_options..caller..)Zsenderroptions)objr user_optionsgetr send)ctxrrrZpreload_optionsr[frWrcallerzs z&handle_preload_options..callerr)r`rarr_rhandle_preload_optionsxs rbcs,eZdZdZfddZfddZZS) CeleryOptionzCustomized option for Celery.cs(|jr|j|j|_tj|f||Sr%)default_value_from_contextrZdefaultsuper get_default)rr^rr __class__rrrgszCeleryOption.get_defaultcs.|dd|_|dd|_tj||dS)zInitialize a Celery option. help_groupNrd)poprjrdrfrrrrrhrrrszCeleryOption.__init__)rPrQrRrSrgr __classcell__rrrhrrcs rcc@seZdZdZddZdS) CeleryCommandzCustomized command for Celery.c Cst}||D]R}||}|dk rt|drP|jrP|t|jg|q|dg|q|D](\}}| || |W5QRXqldS)z7Write all the options into the formatter if they exist.NrjOptions) r get_paramsZget_help_recordhasattrrj setdefaultr=appenditemssectionZwrite_dl)rr^ formatteroptsparamrvrTZ opts_grouprrrformat_optionss  zCeleryCommand.format_optionsN)rPrQrRrSrzrrrrrnsrncs4eZdZdZfddZeeeedddZZ S) DaemonOptionzCommon daemonization optioncs0tj|f|dd|d|jd|dS)NrjzDaemonization Optionscallback)rjr|)rfrrkdaemon_settingrlrhrrrs    zDaemonOption.__init__)r^optvaluereturncCs&|p$t|jjj|jjd|jdS)z Try to fetch daemonization option from applications settings. Use the daemon command name as prefix (eg. `worker` -> `worker_pidfile`) _N)getattrrZrconfcommandrT)rr^r~rrrrr}szDaemonOption.daemon_setting) rPrQrRrSrrrcrr}rmrrrhrr{s r{cs eZdZdZfddZZS)CeleryDaemonCommandzDaemon commands.c s\tj|||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)rfrparamsextendr{rlrhrrrs      zCeleryDaemonCommand.__init__)rPrQrRrSrrmrrrhrrsrc@seZdZdZdZddZdS)CommaSeparatedListzComma separated list argument.zcomma separated listcCs t|Sr%)r Z str_to_listrrrxr^rrrconvertszCommaSeparatedList.convertNrPrQrRrSrTrrrrrrsrc@seZdZdZdZddZdS) JsonArrayzJSON formatted array argument.z json arrayc Cslt|tr|Szt|}Wn0tk rL}z|t|W5d}~XYnXt|tsh||d|S)Nz was not an array)r4r5r9loads ValueErrorfailr=rrrxr^verrrrs   zJsonArray.convertNrrrrrrsrc@seZdZdZdZddZdS) JsonObjectzJSON formatted object argument.z json objectc Cslt|tr|Szt|}Wn0tk rL}z|t|W5d}~XYnXt|tsh||d|S)Nz was not an object)r4r7r9rrrr=rrrrrs   zJsonObject.convertNrrrrrrsrc@seZdZdZdZddZdS)ISO8601DateTimezISO 8601 Date Time argument.z iso-86091c Cs@z t|WSttfk r:}z||W5d}~XYnXdSr%)r TypeErrorrrrrrxr^rrrrrs zISO8601DateTime.convertNrrrrrrsrc@seZdZdZdZddZdS)ISO8601DateTimeOrFloatz%ISO 8601 Date Time or float argument.ziso-86091 or floatc Csdz t|WSttfk r"YnXz t|WSttfk r^}z||W5d}~XYnXdSr%)floatrrrrrrrrrs  zISO8601DateTimeOrFloat.convertNrrrrrr srcs,eZdZdZfddZfddZZS)LogLevelzLog level option.cstddS)z:Initialize the log level option with the relevant choices.)DEBUGINFOWARNINGr$CRITICALFATALN)rfrr#rhrrr szLogLevel.__init__cs0t|tjr|S|}t|||}t|Sr%)r4numbersIntegralupperrfrr rrhrrr$s  zLogLevel.convert)rPrQrRrSrrrmrrrhrrs r)8rSr9r collectionsr functoolsrpprintrtypingrr&rrZkombu.utils.objectsr Z celery._stater Zcelery.signalsr Z celery.utilsr Zcelery.utils.logr Zcelery.utils.timerZpygmentsrZpygments.formattersrZpygments.lexersr ImportErrorr;r<rrbOptionrcCommandrnr{rrrrrrZChoicerZ JSON_ARRAYZ JSON_OBJECTZISO8601ZISO8601_OR_FLOATZ LOG_LEVELZCOMMA_SEPARATED_LISTrrrrsT             V