;ò ÎÑAHc@sÙdZdkZdkZdkZdkZdZd„Zeƒdefd„ƒYZdefd„ƒYZ defd „ƒYZ d efd „ƒYZ d efd „ƒYZ dZ d„Zedjo eƒndS(sðModule for interactively running scripts. This module implements classes for interactively running scripts written for any system with a prompt which can be matched by a regexp suitable for pexpect. It can be used to run as if they had been typed up interactively, an arbitrary series of commands for the target system. The module includes classes ready for IPython (with the default prompts), plain Python and SAGE, but making a new one is trivial. To see how to use it, simply run the module as a script: ./irunner.py --help This is an extension of Ken Schutte 's script contributed on the ipython-user list: http://scipy.net/pipermail/ipython-user/2006-May/001705.html NOTES: - This module requires pexpect, available in most linux distros, or which can be downloaded from http://pexpect.sourceforge.net - Because pexpect only works under Unix or Windows-Cygwin, this has the same limitations. This means that it will NOT work under native windows Python. Ns@ Interactive script runner, type: %s runner [opts] script_name cCs5tid djodSnd„}|ti_dS(soPatch pexpect to prevent unhandled exceptions at VM teardown. Calling this function will monkeypatch the pexpect.spawn class and modify its __del__ method to make it more robust in the face of failures that can occur if it is called when the Python VM is shutting down. Since Python may fire __del__ methods arbitrarily late, it's possible for them to execute during the teardown of the Python VM itself. At this point, various builtin modules have been reset to None. Thus, the call to self.close() will trigger an exception because it tries to call os.close(), and os is now None. is2.2NcCs7|i o(y|iƒWq3tj oq3XndS(sTThis makes sure that no system resources are left open. Python only garbage collects Python objects. OS file descriptors are not Python objects, so they must be handled explicitly. If the child file descriptor was opened outside of this class (passed to the constructor) then this does not close it. N(sselfsclosedsclosesAttributeError(sself((s@/u/d/b/dbachman/=/lib/python2.3/site-packages/IPython/irunner.pys__del__Bs  (spexpects __version__s__del__sspawn(s__del__((s@/u/d/b/dbachman/=/lib/python2.3/site-packages/IPython/irunner.pyspexpect_monkeypatch0s  sInteractiveRunnercBsVtZdZeeied„Zd„Ze e d„Z e e d„Z ed„Z RS(sCClass to run a sequence of commands through an interactive program.cCs’||_||_|tjo g}n||_||_||_d|_ti |i|idtƒ}|_ |i|_|i ddƒdS(sdConstruct a runner. Inputs: - program: command to execute the given program. - prompts: a list of patterns to match as valid prompts, in the format used by pexpect. This basically means that it can be either a string (to be compiled as a regular expression) or a list of such (it must be a true list, as pexpect does type checks). If more than one prompt is given, the first is treated as the main program prompt and the others as 'continuation' prompts, like python's. This means that blank lines in the input source are ommitted when the first prompt is matched, but are NOT ommitted when the continuation one matches, since this is how python signals the end of multiline input interactively. Optional inputs: - args(None): optional list of strings to pass as arguments to the child program. - out(sys.stdout): if given, an output stream to be used when writing output. The only requirement is that it must have a .write() method. Public members not parameterized in the constructor: - delaybeforesend(0): Newer versions of pexpect have a delay before sending each new input. For our purposes here, it's typically best to just set this to zero, but if you encounter reliability problems or want an interactive run to pause briefly at each prompt, just increase this value (it is measured in seconds). Note that this variable is not honored at all by older versions of pexpect. istimeouticiÈN( sprogramsselfspromptssargssNonesoutsechosdelaybeforesendspexpectsspawnscschilds setwinsize(sselfsprogramspromptssargssoutsechosc((s@/u/d/b/dbachman/=/lib/python2.3/site-packages/IPython/irunner.pys__init__Ws#       % cCs|iiƒdS(sclose child processN(sselfschildsclose(sself((s@/u/d/b/dbachman/=/lib/python2.3/site-packages/IPython/irunner.pysclose’scCsIt|dƒ}z|i|||ƒ}Wd|iƒX|o|SndS(s»Run the given file interactively. Inputs: -fname: name of the file to execute. See the run_source docstring for the meaning of the optional arguments.srN( sopensfnamesfobjsselfs run_sourcesinteracts get_outputsoutsclose(sselfsfnamesinteracts get_outputsfobjsout((s@/u/d/b/dbachman/=/lib/python2.3/site-packages/IPython/irunner.pysrun_file—s cs>t|tƒ o|itƒ}n|io(ti‰|i i ‰‡‡d†}n d„}|i }|i|iƒ} |i| ƒ} t} |og}|i} nxÜ|D]Ô} | djo | iƒp| iƒidƒo|| ƒq°n||iƒ|i| ƒy|i| ƒ} Wn*tij o||iƒt } PnX||iƒ|o| |it!| dƒƒq°q°W|i i"ƒ| o[|oP|idƒdGy|i#ƒWqót$j o|dƒ|i i"ƒqóXqn|od}t&i'|IJn|idƒ|od i(|ƒSnd S( sKRun the given source code interactively. Inputs: - source: a string of code to be executed, or an open file object we can iterate over. Optional inputs: - interact(False): if true, start to interact with the running program at the end of the script. Otherwise, just exit. - get_output(False): if true, capture the output of the child process (filtering the input commands out) and return it as a string. Returns: A string containing the process output, but only if requested. csˆ|idˆƒƒS(Ns (sstdwritesssreplaceslinesep(ss(sstdwriteslinesep(s@/u/d/b/dbachman/=/lib/python2.3/site-packages/IPython/irunner.pysÇscCstS(N(sNone(ss((s@/u/d/b/dbachman/=/lib/python2.3/site-packages/IPython/irunner.pysÊsis#s s<< Starting interactive mode >>s s;Further interaction is not possible: child process is dead.sN()s isinstancessourcesfiles splitlinessTruesselfsechososslinesepsoutswritesstdwriteschildscscompile_pattern_listspromptss expect_lists prompt_idxs end_normals get_outputsoutputsappends store_outputscmdsisspaceslstrips startswithsafterssendspexpectsEOFsbeforesFalseslensflushsinteractsOSErrorsessyssstderrsjoin(sselfssourcesinteracts get_outputslinesepswritesstdwritesescs end_normalscmdspromptss prompt_idxs store_outputsoutput((slinesepsstdwrites@/u/d/b/dbachman/=/lib/python2.3/site-packages/IPython/irunner.pys run_source©s^      0     #    c Cs¥tidt|iiƒ}|i}|dddddt ddƒ|i |ƒ\}}t|ƒd jotid IJtid ƒn|i|d |iƒd S( sRun as a command-line script.susages-is --interactsactions store_truesdefaultshelps2Interact with the program after the script is run.is(You must supply exactly one file to run.iN(soptparses OptionParsersUSAGEsselfs __class__s__name__sparsers add_optionsnewoptsFalses parse_argssargvsoptssargsslenssyssstderrsexitsrun_filesinteract(sselfsargvsnewoptsargssparsersopts((s@/u/d/b/dbachman/=/lib/python2.3/site-packages/IPython/irunner.pysmains   ( s__name__s __module__s__doc__sNonessyssstdoutsTrues__init__sclosesFalsesrun_files run_sourcesmain(((s@/u/d/b/dbachman/=/lib/python2.3/site-packages/IPython/irunner.pysInteractiveRunnerTs ; gs IPythonRunnercBs&tZdZdeeied„ZRS(sÐInteractive IPython runner. This initalizes IPython in 'nocolor' mode for simplicity. This lets us avoid having to write a regexp that matches ANSI sequences, though pexpect does support them. If anyone contributes patches for ANSI color support, they will be welcome. It also sets the prompts manually, since the prompt regexps for pexpect need to be matched to the actual prompts, so user-customized prompts would break this. sipythoncCskddddddddg}|tjo |}n ||}d d g}ti||||||ƒd S( s:New runner, optionally passing the ipython command to use.s-colorssNoColors-pi1s In [\#]: s-pi2s .\D.: s -noterm_titles -noautoindents In \[\d+\]: s \.*: N( sargs0sargssNonespromptssInteractiveRunners__init__sselfsprogramsoutsecho(sselfsprogramsargssoutsechospromptssargs0((s@/u/d/b/dbachman/=/lib/python2.3/site-packages/IPython/irunner.pys__init__/s   (s__name__s __module__s__doc__sNonessyssstdoutsTrues__init__(((s@/u/d/b/dbachman/=/lib/python2.3/site-packages/IPython/irunner.pys IPythonRunner"s s PythonRunnercBs&tZdZdeeied„ZRS(sInteractive Python runner.spythoncCs,ddg}ti||||||ƒdS(s9New runner, optionally passing the python command to use.s>>> s\.\.\. N(spromptssInteractiveRunners__init__sselfsprogramsargssoutsecho(sselfsprogramsargssoutsechosprompts((s@/u/d/b/dbachman/=/lib/python2.3/site-packages/IPython/irunner.pys__init__@s (s__name__s __module__s__doc__sNonessyssstdoutsTrues__init__(((s@/u/d/b/dbachman/=/lib/python2.3/site-packages/IPython/irunner.pys PythonRunner=s s SAGERunnercBs&tZdZdeeied„ZRS(söInteractive SAGE runner. WARNING: this runner only works if you manually configure your SAGE copy to use 'colors NoColor' in the ipythonrc config file, since currently the prompt matching regexp does not identify color sequences.ssagecCs,ddg}ti||||||ƒdS(s7New runner, optionally passing the sage command to use.ssage: s \s*\.\.\. N(spromptssInteractiveRunners__init__sselfsprogramsargssoutsecho(sselfsprogramsargssoutsechosprompts((s@/u/d/b/dbachman/=/lib/python2.3/site-packages/IPython/irunner.pys__init__Ns (s__name__s __module__s__doc__sNonessyssstdoutsTrues__init__(((s@/u/d/b/dbachman/=/lib/python2.3/site-packages/IPython/irunner.pys SAGERunnerGs s RunnerFactorycBs/tZdZeid„Zd„Zd„ZRS(s Code runner factory. This class provides an IPython code runner, but enforces that only one runner is ever instantiated. The runner is created based on the extension of the first file to run, and it raises an exception if a runner is later requested for a different extension type. This ensures that we don't generate example files for doctest with a mix of python and ipython syntax. cCs||_t|_t|_dS(sInstantiate a code runner.N(soutsselfsNonesrunners runnerClass(sselfsout((s@/u/d/b/dbachman/=/lib/python2.3/site-packages/IPython/irunner.pys__init__as  cCs)||_|d|iƒ|_|iSdS(Nsout(s runnerClasssselfsoutsrunner(sselfs runnerClass((s@/u/d/b/dbachman/=/lib/python2.3/site-packages/IPython/irunner.pys _makeRunnerhs cCs£|idƒo t}n+|idƒo t}ntd|ƒ‚|itjo|i |ƒSn;||ijo |iSn d|i|f}t|ƒ‚dS(s'Return a runner for the given filename.s.pys.ipys Unknown file type for Runner: %rs'A runner of type %r can not run file %rN( sfnamesendswiths PythonRunners runnerClasss IPythonRunners ValueErrorsselfsrunnersNones _makeRunnerse(sselfsfnameses runnerClass((s@/u/d/b/dbachman/=/lib/python2.3/site-packages/IPython/irunner.pys__call__ms   (s__name__s __module__s__doc__ssyssstdouts__init__s _makeRunners__call__(((s@/u/d/b/dbachman/=/lib/python2.3/site-packages/IPython/irunner.pys RunnerFactoryUs  s„ %prog [options] file_to_run This is an interface to the various interactive runners available in this module. If you want to pass specific options to one of the runners, you need to first terminate the main options with a '--', and then provide the runner's options. For example: irunner.py --python -- --help will pass --help to the python runner. Similarly, irunner.py --ipython -- --interact script.ipy will run the script.ipy file under the IPython runner, and then will start to interact with IPython at the end of the script (instead of exiting). The already implemented runners are listed below; adding one for a new program is a trivial task, see the source for examples. WARNING: the SAGE runner only works if you manually configure your SAGE copy to use 'colors NoColor' in the ipythonrc config file, since currently the prompt matching regexp does not identify color sequences. c CsFtidtƒ}|i}|iddƒ|dddddddd d ƒ|d dddddd d d ƒ|dddddddd dƒ|iƒ\}}t dt d t dt ƒ}ytii|dƒd}Wntj o d}nXhdd<dd <dd<}|i||iƒ}||ƒi|ƒdS(sRun as a command-line script.susagesmodesipythons --ipythonsactions store_constsdestsconstshelps%IPython interactive runner (default).s--pythonspythonsPython interactive runner.s--sagessagesSAGE interactive runner.iiÿÿÿÿss.ipys.pys.sageN(soptparses OptionParsers MAIN_USAGEsparsers add_optionsnewopts set_defaultss parse_argssoptssargssdicts IPythonRunners PythonRunners SAGERunnersrunnerssosspathssplitextsexts IndexErrorsmodessgetsmodesmain(sargssmodessparsersmodesextsrunnerssoptssnewopt((s@/u/d/b/dbachman/=/lib/python2.3/site-packages/IPython/irunner.pysmainœs*       !s__main__(s__doc__soptparsesosssysspexpectsUSAGEspexpect_monkeypatchsobjectsInteractiveRunners IPythonRunners PythonRunners SAGERunners RunnerFactorys MAIN_USAGEsmains__name__( s PythonRunners MAIN_USAGEsInteractiveRunnerspexpectsoptparses SAGERunners IPythonRunnerssyssUSAGEs RunnerFactoryspexpect_monkeypatchsmainsos((s@/u/d/b/dbachman/=/lib/python2.3/site-packages/IPython/irunner.pys?s      !Î .