| """Extension to execute code outside the Python shell window. |
| |
| This adds the following commands: |
| |
| - Check module does a full syntax check of the current module. |
| It also runs the tabnanny to catch any inconsistent tabs. |
| |
| - Run module executes the module's code in the __main__ namespace. The window |
| must have been saved previously. The module is added to sys.modules, and is |
| also added to the __main__ namespace. |
| |
| XXX GvR Redesign this interface (yet again) as follows: |
| |
| - Present a dialog box for ``Run Module'' |
| |
| - Allow specify command line arguments in the dialog box |
| |
| """ |
| |
| import os |
| import re |
| import string |
| import tabnanny |
| import tokenize |
| import tkMessageBox |
| from idlelib.EditorWindow import EditorWindow |
| from idlelib import PyShell |
| |
| from idlelib.configHandler import idleConf |
| |
| indent_message = """Error: Inconsistent indentation detected! |
| |
| 1) Your indentation is outright incorrect (easy to fix), OR |
| |
| 2) Your indentation mixes tabs and spaces. |
| |
| To fix case 2, change all tabs to spaces by using Edit->Select All followed \ |
| by Format->Untabify Region and specify the number of columns used by each tab. |
| """ |
| |
| class ScriptBinding: |
| |
| menudefs = [ |
| ('run', [None, |
| ('Check Module', '<<check-module>>'), |
| ('Run Module', '<<run-module>>'), ]), ] |
| |
| def __init__(self, editwin): |
| self.editwin = editwin |
| # Provide instance variables referenced by Debugger |
| # XXX This should be done differently |
| self.flist = self.editwin.flist |
| self.root = self.editwin.root |
| |
| def check_module_event(self, event): |
| filename = self.getfilename() |
| if not filename: |
| return 'break' |
| if not self.checksyntax(filename): |
| return 'break' |
| if not self.tabnanny(filename): |
| return 'break' |
| |
| def tabnanny(self, filename): |
| f = open(filename, 'r') |
| try: |
| tabnanny.process_tokens(tokenize.generate_tokens(f.readline)) |
| except tokenize.TokenError as msg: |
| msgtxt, (lineno, start) = msg |
| self.editwin.gotoline(lineno) |
| self.errorbox("Tabnanny Tokenizing Error", |
| "Token Error: %s" % msgtxt) |
| return False |
| except tabnanny.NannyNag as nag: |
| # The error messages from tabnanny are too confusing... |
| self.editwin.gotoline(nag.get_lineno()) |
| self.errorbox("Tab/space error", indent_message) |
| return False |
| return True |
| |
| def checksyntax(self, filename): |
| self.shell = shell = self.flist.open_shell() |
| saved_stream = shell.get_warning_stream() |
| shell.set_warning_stream(shell.stderr) |
| f = open(filename, 'r') |
| source = f.read() |
| f.close() |
| if '\r' in source: |
| source = re.sub(r"\r\n", "\n", source) |
| source = re.sub(r"\r", "\n", source) |
| if source and source[-1] != '\n': |
| source = source + '\n' |
| editwin = self.editwin |
| text = editwin.text |
| text.tag_remove("ERROR", "1.0", "end") |
| try: |
| # If successful, return the compiled code |
| return compile(source, filename, "exec") |
| except (SyntaxError, OverflowError) as value: |
| msg = value.msg or "<no detail available>" |
| lineno = value.lineno or 1 |
| offset = value.offset or 0 |
| if offset == 0: |
| lineno += 1 #mark end of offending line |
| pos = "0.0 + %d lines + %d chars" % (lineno-1, offset-1) |
| editwin.colorize_syntax_error(text, pos) |
| self.errorbox("SyntaxError", "%-20s" % msg) |
| return False |
| finally: |
| shell.set_warning_stream(saved_stream) |
| |
| def run_module_event(self, event): |
| """Run the module after setting up the environment. |
| |
| First check the syntax. If OK, make sure the shell is active and |
| then transfer the arguments, set the run environment's working |
| directory to the directory of the module being executed and also |
| add that directory to its sys.path if not already included. |
| |
| """ |
| filename = self.getfilename() |
| if not filename: |
| return 'break' |
| code = self.checksyntax(filename) |
| if not code: |
| return 'break' |
| if not self.tabnanny(filename): |
| return 'break' |
| shell = self.shell |
| interp = shell.interp |
| if PyShell.use_subprocess: |
| shell.restart_shell() |
| dirname = os.path.dirname(filename) |
| # XXX Too often this discards arguments the user just set... |
| interp.runcommand("""if 1: |
| _filename = %r |
| import sys as _sys |
| from os.path import basename as _basename |
| if (not _sys.argv or |
| _basename(_sys.argv[0]) != _basename(_filename)): |
| _sys.argv = [_filename] |
| import os as _os |
| _os.chdir(%r) |
| del _filename, _sys, _basename, _os |
| \n""" % (filename, dirname)) |
| interp.prepend_syspath(filename) |
| # XXX KBK 03Jul04 When run w/o subprocess, runtime warnings still |
| # go to __stderr__. With subprocess, they go to the shell. |
| # Need to change streams in PyShell.ModifiedInterpreter. |
| interp.runcode(code) |
| return 'break' |
| |
| def getfilename(self): |
| """Get source filename. If not saved, offer to save (or create) file |
| |
| The debugger requires a source file. Make sure there is one, and that |
| the current version of the source buffer has been saved. If the user |
| declines to save or cancels the Save As dialog, return None. |
| |
| If the user has configured IDLE for Autosave, the file will be |
| silently saved if it already exists and is dirty. |
| |
| """ |
| filename = self.editwin.io.filename |
| if not self.editwin.get_saved(): |
| autosave = idleConf.GetOption('main', 'General', |
| 'autosave', type='bool') |
| if autosave and filename: |
| self.editwin.io.save(None) |
| else: |
| reply = self.ask_save_dialog() |
| self.editwin.text.focus_set() |
| if reply == "ok": |
| self.editwin.io.save(None) |
| filename = self.editwin.io.filename |
| else: |
| filename = None |
| return filename |
| |
| def ask_save_dialog(self): |
| msg = "Source Must Be Saved\n" + 5*' ' + "OK to Save?" |
| mb = tkMessageBox.Message(title="Save Before Run or Check", |
| message=msg, |
| icon=tkMessageBox.QUESTION, |
| type=tkMessageBox.OKCANCEL, |
| default=tkMessageBox.OK, |
| parent=self.editwin.text) |
| return mb.show() |
| |
| def errorbox(self, title, message): |
| # XXX This should really be a function of EditorWindow... |
| tkMessageBox.showerror(title, message, parent=self.editwin.text) |
| self.editwin.text.focus_set() |