corrected code to not suppose that Write() always writes all the data it was given

[wxWidgets.git] / wxPython / demo / Menu.py

diff --git a/wxPython/demo/Menu.py b/wxPython/demo/Menu.py
index 45ca029fba88dcc92ffb6592645f4af41b808c0f..431ab66812115914a1a484c11f0b56cf684d22fe 100644 (file)
--- a/wxPython/demo/Menu.py
+++ b/wxPython/demo/Menu.py
@@ -135,11 +135,12 @@ check the source for this sample to see how to implement them.
     # Methods
 
     def OnMenuHighlight(self, event):
     # Methods
 
     def OnMenuHighlight(self, event):

-        # Show how to get menu item imfo from this event handler
+        # Show how to get menu item info from this event handler

         id = event.GetMenuId()
         item = self.GetMenuBar().FindItemById(id)
         id = event.GetMenuId()
         item = self.GetMenuBar().FindItemById(id)

-        text = item.GetText()
-        help = item.GetHelp()
+        if item:
+            text = item.GetText()
+            help = item.GetHelp()

 
         # but in this case just call Skip so the default is done
         event.Skip() 
 
         # but in this case just call Skip so the default is done
         event.Skip() 


@@ -297,5 +298,5 @@ and GTK+ currently.
 if __name__ == '__main__':
     import sys,os
     import run
 if __name__ == '__main__':
     import sys,os
     import run

-    run.main(['', os.path.basename(sys.argv[0])])
+    run.main(['', os.path.basename(sys.argv[0])] + sys.argv[1:])