wxPython/src/_app_ex.py

   1
   2 #----------------------------------------------------------------------
   3
   4 class PyOnDemandOutputWindow:
   5     """
   6     A class that can be used for redirecting Python's stdout and
   7     stderr streams.  It will do nothing until something is wrriten to
   8     the stream at which point it will create a Frame with a text area
   9     and write the text there.
  10     """
  11     def __init__(self, title = "wxPython: stdout/stderr"):
  12         self.frame  = None
  13         self.title  = title
  14         self.parent = None
  15
  16     def SetParent(self, parent):
  17         """Set the window to be used as the popup Frame's parent."""
  18         self.parent = parent
  19
  20
  21     def CreateOutputWindow(self, st):
  22         self.frame = wx.Frame(self.parent, -1, self.title,
  23                               style=wx.DEFAULT_FRAME_STYLE | wx.NO_FULL_REPAINT_ON_RESIZE)
  24         self.text  = wx.TextCtrl(self.frame, -1, "",
  25                                  style = wx.TE_MULTILINE | wx.TE_READONLY)
  26         self.text.AppendText(st)
  27         self.frame.SetSize((450, 300))
  28         self.frame.Show(True)
  29         EVT_CLOSE(self.frame, self.OnCloseWindow)
  30
  31
  32     def OnCloseWindow(self, event):
  33         if self.frame is not None:
  34             self.frame.Destroy()
  35         self.frame = None
  36         self.text  = None
  37
  38
  39     # These methods provide the file-like output behaviour.
  40     def write(self, text):
  41         """
  42         Create the output window if needed and write the string to it.
  43         If not called in the context of the gui thread then uses
  44         CallAfter to do the work there.
  45         """
  46         if self.frame is None:
  47             if not wx.Thread_IsMain():
  48                 wx.CallAfter(self.CreateOutputWindow, text)
  49             else:
  50                 self.CreateOutputWindow(text)
  51         else:
  52             if not wx.Thread_IsMain():
  53                 wx.CallAfter(self.text.AppendText, text)
  54             else:
  55                 self.text.AppendText(text)
  56
  57
  58     def close(self):
  59         if self.frame is not None:
  60             wx.CallAfter(self.frame.Close)
  61
  62
  63
  64 #----------------------------------------------------------------------
  65
  66 _defRedirect = (wx.Platform == '__WXMSW__' or wx.Platform == '__WXMAC__')
  67
  68 class App(wx.PyApp):
  69     """
  70     The ``wx.App`` class represents the application and is used to:
  71
  72       * bootstrap the wxPython system and initialize the underlying
  73         gui toolkit
  74       * set and get application-wide properties
  75       * implement the windowing system main message or event loop,
  76         and to dispatch events to window instances
  77       * etc.
  78
  79     Every application must have a ``wx.App`` instance, and all
  80     creation of UI objects should be delayed until after the
  81     ``wx.App`` object has been created in order to ensure that the gui
  82     platform and wxWidgets have been fully initialized.
  83
  84     Normally you would derive from this class and implement an
  85     ``OnInit`` method that creates a frame and then calls
  86     ``self.SetTopWindow(frame)``.
  87
  88     :see: `wx.PySimpleApp` for a simpler app class that can be used
  89     directly.
  90     """
  91
  92     outputWindowClass = PyOnDemandOutputWindow
  93
  94     def __init__(self, redirect=_defRedirect, filename=None,
  95                  useBestVisual=False, clearSigInt=True):
  96         """
  97         Construct a ``wx.App`` object.
  98
  99         :param redirect: Should ``sys.stdout`` and ``sys.stderr`` be
 100             redirected?  Defaults to True on Windows and Mac, False
 101             otherwise.  If `filename` is None then output will be
 102             redirected to a window that pops up as needed.  (You can
 103             control what kind of window is created for the output by
 104             resetting the class variable ``outputWindowClass`` to a
 105             class of your choosing.)
 106
 107         :param filename: The name of a file to redirect output to, if
 108             redirect is True.
 109
 110         :param useBestVisual: Should the app try to use the best
 111             available visual provided by the system (only relevant on
 112             systems that have more than one visual.)  This parameter
 113             must be used instead of calling `SetUseBestVisual` later
 114             on because it must be set before the underlying GUI
 115             toolkit is initialized.
 116
 117         :param clearSigInt: Should SIGINT be cleared?  This allows the
 118             app to terminate upon a Ctrl-C in the console like other
 119             GUI apps will.
 120
 121         :note: You should override OnInit to do applicaition
 122             initialization to ensure that the system, toolkit and
 123             wxWidgets are fully initialized.
 124         """
 125         wx.PyApp.__init__(self)
 126
 127         if wx.Platform == "__WXMAC__":
 128             try:
 129                 import MacOS
 130                 if not MacOS.WMAvailable():
 131                     print """\
 132 This program needs access to the screen. Please run with 'pythonw',
 133 not 'python', and only when you are logged in on the main display of
 134 your Mac."""
 135                     _sys.exit(1)
 136             except SystemExit:
 137                 raise
 138             except:
 139                 pass
 140
 141         # This has to be done before OnInit
 142         self.SetUseBestVisual(useBestVisual)
 143
 144         # Set the default handler for SIGINT.  This fixes a problem
 145         # where if Ctrl-C is pressed in the console that started this
 146         # app then it will not appear to do anything, (not even send
 147         # KeyboardInterrupt???)  but will later segfault on exit.  By
 148         # setting the default handler then the app will exit, as
 149         # expected (depending on platform.)
 150         if clearSigInt:
 151             try:
 152                 import signal
 153                 signal.signal(signal.SIGINT, signal.SIG_DFL)
 154             except:
 155                 pass
 156
 157         # Save and redirect the stdio to a window?
 158         self.stdioWin = None
 159         self.saveStdio = (_sys.stdout, _sys.stderr)
 160         if redirect:
 161             self.RedirectStdio(filename)
 162
 163         # This finishes the initialization of wxWindows and then calls
 164         # the OnInit that should be present in the derived class
 165         self._BootstrapApp()
 166
 167
 168     def __del__(self):
 169         try:
 170             self.RestoreStdio()  # Just in case the MainLoop was overridden
 171         except:
 172             pass
 173
 174
 175     def SetTopWindow(self, frame):
 176         """Set the \"main\" top level window"""
 177         if self.stdioWin:
 178             self.stdioWin.SetParent(frame)
 179         wx.PyApp.SetTopWindow(self, frame)
 180
 181
 182     def MainLoop(self):
 183         """Execute the main GUI event loop"""
 184         wx.PyApp.MainLoop(self)
 185         self.RestoreStdio()
 186
 187
 188     def RedirectStdio(self, filename=None):
 189         """Redirect sys.stdout and sys.stderr to a file or a popup window."""
 190         if filename:
 191             _sys.stdout = _sys.stderr = open(filename, 'a')
 192         else:
 193             self.stdioWin = self.outputWindowClass()
 194             _sys.stdout = _sys.stderr = self.stdioWin
 195
 196
 197     def RestoreStdio(self):
 198         _sys.stdout, _sys.stderr = self.saveStdio
 199
 200
 201
 202 # change from wx.PyApp_XX to wx.App_XX
 203 App_GetMacSupportPCMenuShortcuts = _core_.PyApp_GetMacSupportPCMenuShortcuts
 204 App_GetMacAboutMenuItemId        = _core_.PyApp_GetMacAboutMenuItemId
 205 App_GetMacPreferencesMenuItemId  = _core_.PyApp_GetMacPreferencesMenuItemId
 206 App_GetMacExitMenuItemId         = _core_.PyApp_GetMacExitMenuItemId
 207 App_GetMacHelpMenuTitleName      = _core_.PyApp_GetMacHelpMenuTitleName
 208 App_SetMacSupportPCMenuShortcuts = _core_.PyApp_SetMacSupportPCMenuShortcuts
 209 App_SetMacAboutMenuItemId        = _core_.PyApp_SetMacAboutMenuItemId
 210 App_SetMacPreferencesMenuItemId  = _core_.PyApp_SetMacPreferencesMenuItemId
 211 App_SetMacExitMenuItemId         = _core_.PyApp_SetMacExitMenuItemId
 212 App_SetMacHelpMenuTitleName      = _core_.PyApp_SetMacHelpMenuTitleName
 213 App_GetComCtl32Version           = _core_.PyApp_GetComCtl32Version
 214
 215 #----------------------------------------------------------------------------
 216
 217 class PySimpleApp(wx.App):
 218     """
 219     A simple application class.  You can just create one of these and
 220     then then make your top level windows later, and not have to worry
 221     about OnInit.  For example::
 222
 223         app = wx.PySimpleApp()
 224         frame = wx.Frame(None, title='Hello World')
 225         frame.Show()
 226         app.MainLoop()
 227
 228     :see: `wx.App`
 229     """
 230
 231     def __init__(self, redirect=False, filename=None,
 232                  useBestVisual=False, clearSigInt=True):
 233         """
 234         :see: `wx.App.__init__`
 235         """
 236         wx.App.__init__(self, redirect, filename, useBestVisual, clearSigInt)
 237
 238     def OnInit(self):
 239         wx.InitAllImageHandlers()
 240         return True
 241
 242
 243
 244 # Is anybody using this one?
 245 class PyWidgetTester(wx.App):
 246     def __init__(self, size = (250, 100)):
 247         self.size = size
 248         wx.App.__init__(self, 0)
 249
 250     def OnInit(self):
 251         self.frame = wx.Frame(None, -1, "Widget Tester", pos=(0,0), size=self.size)
 252         self.SetTopWindow(self.frame)
 253         return True
 254
 255     def SetWidget(self, widgetClass, *args, **kwargs):
 256         w = widgetClass(self.frame, *args, **kwargs)
 257         self.frame.Show(True)
 258
 259 #----------------------------------------------------------------------------
 260 # DO NOT hold any other references to this object.  This is how we
 261 # know when to cleanup system resources that wxWidgets is holding.  When
 262 # the sys module is unloaded, the refcount on sys.__wxPythonCleanup
 263 # goes to zero and it calls the wx.App_CleanUp function.
 264
 265 class __wxPyCleanup:
 266     def __init__(self):
 267         self.cleanup = _core_.App_CleanUp
 268     def __del__(self):
 269         self.cleanup()
 270
 271 _sys.__wxPythonCleanup = __wxPyCleanup()
 272
 273 ## # another possible solution, but it gets called too early...
 274 ## import atexit
 275 ## atexit.register(_core_.wxApp_CleanUp)
 276
 277
 278 #----------------------------------------------------------------------------