WindowList.py 2.23 KB
Newer Older
Guido van Rossum's avatar
Guido van Rossum committed
1 2 3 4 5 6
from Tkinter import *

class WindowList:

    def __init__(self):
        self.dict = {}
7
        self.callbacks = []
Guido van Rossum's avatar
Guido van Rossum committed
8 9

    def add(self, window):
10
        window.after_idle(self.call_callbacks)
Guido van Rossum's avatar
Guido van Rossum committed
11 12 13 14 15 16 17 18
        self.dict[str(window)] = window

    def delete(self, window):
        try:
            del self.dict[str(window)]
        except KeyError:
            # Sometimes, destroy() is called twice
            pass
19
        self.call_callbacks()
Guido van Rossum's avatar
Guido van Rossum committed
20 21 22 23 24

    def add_windows_to_menu(self,  menu):
        list = []
        for key in self.dict.keys():
            window = self.dict[key]
25 26 27 28
            try:
                title = window.get_title()
            except TclError:
                continue
Guido van Rossum's avatar
Guido van Rossum committed
29 30 31 32 33 34 35 36
            list.append((title, window))
        list.sort()
        for title, window in list:
            if title == "Python Shell":
                # Hack -- until we have a better way to this
                continue
            menu.add_command(label=title, command=window.wakeup)

37 38 39 40 41
    def register_callback(self, callback):
        self.callbacks.append(callback)

    def unregister_callback(self, callback):
        try:
Tim Peters's avatar
Tim Peters committed
42
            self.callbacks.remove(callback)
43
        except ValueError:
44 45 46 47 48 49 50 51 52 53
            pass

    def call_callbacks(self):
        for callback in self.callbacks:
            try:
                callback()
            except:
                print "warning: callback failed in WindowList", \
                      sys.exc_type, ":", sys.exc_value

Guido van Rossum's avatar
Guido van Rossum committed
54 55
registry = WindowList()

56 57 58 59
add_windows_to_menu = registry.add_windows_to_menu
register_callback = registry.register_callback
unregister_callback = registry.unregister_callback

Guido van Rossum's avatar
Guido van Rossum committed
60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75

class ListedToplevel(Toplevel):

    def __init__(self, master, **kw):
        Toplevel.__init__(self, master, kw)
        registry.add(self)

    def destroy(self):
        registry.delete(self)
        Toplevel.destroy(self)

    def get_title(self):
        # Subclass can override
        return self.wm_title()

    def wakeup(self):
76
        try:
77 78 79 80
            if self.wm_state() == "iconic":
                self.wm_deiconify()
            else:
                self.tkraise()
81 82 83 84 85
            self.focus_set()
        except TclError:
            # This can happen when the window menu was torn off.
            # Simply ignore it.
            pass