ScriptBinding.py 2.89 KB
Newer Older
1 2 3 4 5
"""Extension to execute code outside the Python shell window.

This adds two commands (to the Edit menu, until there's a separate
Python menu):

6
- Import module (F5) is equivalent to either import or reload of the
7
current module.  The window must have been saved previously. The
8 9
module is added to sys.modules, and is also added to the __main__
namespace.  Output goes to the shell window.
10

11 12
- Run module (Control-F5) does the same but executes the module's
code in the __main__ namespace.
13 14 15 16

"""

import sys
Guido van Rossum's avatar
Guido van Rossum committed
17 18
import os
import imp
19
import tkMessageBox
20

Guido van Rossum's avatar
Guido van Rossum committed
21 22

class ScriptBinding:
23 24
    
    keydefs = {
25 26
        '<<import-module>>': ['<F5>'],
        '<<run-script>>': ['<Control-F5>'],
27 28 29 30
    }
    
    menudefs = [
        ('edit', [None,
31 32
                  ('Import module', '<<import-module>>'),
                  ('Run script', '<<run-script>>'),
33 34 35
                 ]
        ),
    ]
Guido van Rossum's avatar
Guido van Rossum committed
36 37 38

    def __init__(self, editwin):
        self.editwin = editwin
39 40 41 42
        # Provide instance variables referenced by Debugger
        # XXX This should be done differently
        self.flist = self.editwin.flist
        self.root = self.flist.root
Guido van Rossum's avatar
Guido van Rossum committed
43

44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75
    def import_module_event(self, event):
        filename = self.getfilename()
        if not filename:
            return

        modname, ext = os.path.splitext(os.path.basename(filename))
        if sys.modules.has_key(modname):
            mod = sys.modules[modname]
        else:
            mod = imp.new_module(modname)
            sys.modules[modname] = mod
        mod.__file__ = filename
        setattr(sys.modules['__main__'], modname, mod)

        dir = os.path.dirname(filename)
        dir = os.path.normpath(os.path.abspath(dir))
        if dir not in sys.path:
            sys.path.insert(0, dir)

        flist = self.editwin.flist
        shell = flist.open_shell()
        interp = shell.interp
        interp.runcode("reload(%s)" % modname)

    def run_script_event(self, event):
        filename = self.getfilename()
        if not filename:
            return

        flist = self.editwin.flist
        shell = flist.open_shell()
        interp = shell.interp
76 77 78
        if (not sys.argv or
            os.path.basename(sys.argv[0]) != os.path.basename(filename)):
            sys.argv = [filename]
79 80 81 82
        interp.execfile(filename)

    def getfilename(self):
        # Logic to make sure we have a saved filename
83 84 85 86 87 88
        if not self.editwin.get_saved():
            tkMessageBox.showerror("Not saved",
                                   "Please save first!",
                                   master=self.editwin.text)
            self.editwin.text.focus_set()
            return
Guido van Rossum's avatar
Guido van Rossum committed
89 90 91 92 93
        filename = self.editwin.io.filename
        if not filename:
            tkMessageBox.showerror("No file name",
                                   "This window has no file name",
                                   master=self.editwin.text)
94
            self.editwin.text.focus_set()
Guido van Rossum's avatar
Guido van Rossum committed
95
            return
96
        return filename