U Z+d@s&dZddlZddlZddlmZddlZddlmZmZm Z ddZ ddZ d d Z d d Z d dZddZddZddZddZejedejdddedddejdddedd dejd!dedd"dejd#d$dedd%dejd&dedd'dejd(dedd)deje d-d+d,ZdS).z3The ``celery shell`` program, used to start a REPL.N) import_module) CeleryCommand CeleryOptionhandle_preload_optionscCs^ddl}z ddl}Wntk r(Yn&Xddl}|||j|d|j|ddS)Nrz tab:complete)local) codereadline ImportError rlcompleterZ set_completerZ CompleterZcompleteparse_and_bindZinteract)localsrrr r 4/tmp/pip-unpacked-wheel-ucduq0nd/celery/bin/shell.py_invoke_fallback_shell s   rcCsddl}||dSNr)bpythonembed)r rr r r_invoke_bpython_shellsrc Cs<tttttfD](}z||WStk r4YqXqdS)N)_ipython_ipython_pre_10_ipython_terminal _ipython_010 _no_ipythonr )r ipr r r_invoke_ipython_shell srcCsddlm}|g|ddS)Nr) start_ipythonargvuser_ns)IPythonr)r rr r rr*s rcCs:ddlm}|}|jgd|jj||dS)Nr)TerminalIPythonApp)r)ZIPython.frontend.terminal.ipappr instanceZ initializeshellrupdatestart)r r appr r rr/s   rcCs ddlm}|j|ddS)Nr)r)r)ZIPython.terminalrZTerminalInteractiveShellmainloop)r rr r rr7s rcCs ddlm}|g|ddS)Nr)IPShellr)Z IPython.Shellr'r&)r r'r r rr<s rcCs tddS)Nzno suitable ipython found)r )selfr r rrAsrc Cs`z ddl}WnFtk rRz ddl}Wntk rDt|Yn Xt|Yn Xt|dSr)rr rrrr)r rrr r r_invoke_default_shellEs  r))clsz-Iz --ipythonTz Shell OptionszForce IPython.)Zis_flagr*Z help_grouphelpz-Bz --bpythonzForce bpython.z--pythonzForce default Python shell.z-Tz--without-taskszDon't add tasks to locals.z --eventletz Use eventlet.z--geventz Use gevent.Fc Cs:tjdt|rtd|r*tdddl}|jj}|j |||j |j |j |j|j|j|j|j|jd } |s| dd|jD|rt| n|rz t| Wn*tk r|j|jjdYnXnR|r.z t| Wn>tk r,} z|j|jjd | W5d} ~ XYnXt| dS) a.Start shell session with convenient access to celery symbols. The following symbols will be added to the main globals: - ``celery``: the current application. - ``chord``, ``group``, ``chain``, ``chunks``, ``xmap``, ``xstarmap`` ``subtask``, ``Task`` - all registered tasks. rzcelery.concurrency.eventletzcelery.concurrency.geventN) r%celeryTaskchordgroupchainchunksxmapxstarmapsubtask signaturecSs i|]}|jds|j|qS)zcelery.)name startswith__name__).0Ztaskr r r s zshell..z: bpython is not installedz: )syspathinsertosgetcwdrr,objr%loaderZimport_default_modulesr-r.r/r0r1r2r3r4r5r#Ztasksvaluesrrr ZechoERRORrr)) ctxZipythonrpythonZ without_tasksZeventletZgeventr,r%r er r rr"SsJ/     ,r")FFFFFF)__doc__r>r; importlibrZclickZcelery.bin.baserrrrrrrrrrrr)commandoptionZ pass_contextr"r r r rs|