wxPython/demo/run.py

   1 #!/usr/bin/env python
   2 #----------------------------------------------------------------------------
   3 # Name:         run.py
   4 # Purpose:      Simple framework for running individual demos
   5 #
   6 # Author:       Robin Dunn
   7 #
   8 # Created:      6-March-2000
   9 # RCS-ID:       $Id$
  10 # Copyright:    (c) 2000 by Total Control Software
  11 # Licence:      wxWindows license
  12 #----------------------------------------------------------------------------
  13
  14 """
  15 This program will load and run one of the individual demos in this
  16 directory within its own frame window.  Just specify the module name
  17 on the command line.
  18 """
  19
  20 import wx                  # This module uses the new wx namespace
  21 import sys, os
  22
  23 # stuff for debugging
  24 print "wx.VERSION_STRING = ", wx.VERSION_STRING
  25 print "pid:", os.getpid()
  26 ##raw_input("Press a key...")
  27
  28 assertMode = wx.PYAPP_ASSERT_DIALOG
  29 ##assertMode = wx.PYAPP_ASSERT_EXCEPTION
  30
  31
  32 #----------------------------------------------------------------------------
  33
  34 class Log:
  35     def WriteText(self, text):
  36         if text[-1:] == '\n':
  37             text = text[:-1]
  38         wx.LogMessage(text)
  39     write = WriteText
  40
  41
  42 class RunDemoApp(wx.App):
  43     def __init__(self, name, module, useShell):
  44         self.name = name
  45         self.demoModule = module
  46         self.useShell = useShell
  47         wx.App.__init__(self, 0)
  48
  49
  50     def OnInit(self):
  51         wx.InitAllImageHandlers()
  52         wx.Log_SetActiveTarget(wx.LogStderr())
  53
  54         self.SetAssertMode(assertMode)
  55
  56         frame = wx.Frame(None, -1, "RunDemo: " + self.name, pos=(50,50), size=(200,100),
  57                         style=wx.NO_FULL_REPAINT_ON_RESIZE|wx.DEFAULT_FRAME_STYLE)
  58         frame.CreateStatusBar()
  59
  60         menuBar = wx.MenuBar()
  61         menu = wx.Menu()
  62         item = menu.Append(-1, "E&xit\tAlt-X", "Exit demo")
  63         self.Bind(wx.EVT_MENU, self.OnButton, item)
  64         menuBar.Append(menu, "&File")
  65
  66         ns = {}
  67         ns['app'] = self
  68         ns['module'] = self.demoModule
  69         ns['frame'] = frame
  70
  71         frame.SetMenuBar(menuBar)
  72         frame.Show(True)
  73         frame.Bind(wx.EVT_CLOSE, self.OnCloseFrame)
  74
  75         win = self.demoModule.runTest(frame, frame, Log())
  76
  77         # a window will be returned if the demo does not create
  78         # its own top-level window
  79         if win:
  80             # so set the frame to a good size for showing stuff
  81             frame.SetSize((640, 480))
  82             win.SetFocus()
  83             self.window = win
  84             ns['win'] = win
  85             frect = frame.GetRect()
  86
  87         else:
  88             # otherwise the demo made its own frame, so just put a
  89             # button in this one
  90             if hasattr(frame, 'otherWin'):
  91                 ns['win'] = frame.otherWin
  92                 frect = frame.otherWin.GetRect()
  93                 p = wx.Panel(frame, -1)
  94                 b = wx.Button(p, -1, " Exit ", (10,10))
  95                 p.Fit()
  96                 #frame.SetClientSize(p.GetSize())
  97                 frame.SetSize((200, 100))
  98                 frame.Bind(wx.EVT_BUTTON, self.OnButton, b)
  99             else:
 100                 # It was probably a dialog or something that is already
 101                 # gone, so we're done.
 102                 frame.Destroy()
 103                 return True
 104
 105         self.SetTopWindow(frame)
 106         self.frame = frame
 107         #wx.Log_SetActiveTarget(wx.LogStderr())
 108         #wx.Log_SetTraceMask(wx.TraceMessages)
 109
 110         if self.useShell:
 111             # Make a PyShell window, and position it below our test window
 112             from wx import py
 113             shell = py.shell.ShellFrame(None, locals=ns)
 114             frect.OffsetXY(0, frect.height)
 115             frect.height = 400
 116             shell.SetRect(frect)
 117             shell.Show()
 118
 119             # Hook the close event of the test window so that we close
 120             # the shell at the same time
 121             def CloseShell(evt):
 122                 if shell:
 123                     shell.Close()
 124                 evt.Skip()
 125             frame.Bind(wx.EVT_CLOSE, CloseShell)
 126
 127         return True
 128
 129
 130     def OnButton(self, evt):
 131         self.frame.Close(True)
 132
 133
 134     def OnCloseFrame(self, evt):
 135         if hasattr(self, "window") and hasattr(self.window, "ShutdownDemo"):
 136             self.window.ShutdownDemo()
 137         evt.Skip()
 138
 139
 140 #----------------------------------------------------------------------------
 141
 142
 143 def main(argv):
 144     useShell = False
 145     for x in range(len(sys.argv)):
 146         if sys.argv[x] in ['--shell', '-shell', '-s']:
 147             useShell = True
 148             del sys.argv[x]
 149             break
 150
 151     if len(argv) < 2:
 152         print "Please specify a demo module name on the command-line"
 153         raise SystemExit
 154
 155     name, ext  = os.path.splitext(argv[1])
 156     module = __import__(name)
 157
 158
 159     app = RunDemoApp(name, module, useShell)
 160     app.MainLoop()
 161
 162
 163
 164 if __name__ == "__main__":
 165     main(sys.argv)
 166
 167