U .e@s,dZddlZddlZddlmZddlZddlmZmZm Z ddZ ddZ d d Z d d Z d dZddZddZddZddZejeddidejdddedddejdd dedd!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-f4liivr4/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_shellsrcCsddlm}|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 rr6s 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 rr@src Cs`z ddl}WnFtk rRz ddl}Wntk rDt|Yn Xt|Yn Xt|dSr)rr rrrr)r rrr r r_invoke_default_shellDs  r)Zallow_extra_argsT)clsZcontext_settingsz-Iz --ipythonz 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 Ks: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_tasksZeventletZgeventkwargsr,r%r er r rr"RsJ1     ,r")FFFFFF)__doc__r>r; importlibrZclickZcelery.bin.baserrrrrrrrrrrr)commandoptionZ pass_contextr"r r r rs