X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/1e4a197e4c60e461b8068b0619692ea083e30b8b..6d99eb3e379fc5cdb835f9a87f92b612ec58dd1c:/wxPython/demo/GridEnterHandler.py

diff --git a/wxPython/demo/GridEnterHandler.py b/wxPython/demo/GridEnterHandler.py
index d6989fbcfe..638eac61b5 100644
--- a/wxPython/demo/GridEnterHandler.py
+++ b/wxPython/demo/GridEnterHandler.py
@@ -1,11 +1,12 @@
-from wxPython.wx import *
-from wxPython.grid import *
+
+import  wx
+import  wx.grid as  gridlib
 
 #---------------------------------------------------------------------------
 
-class NewEnterHandlingGrid(wxGrid):
+class NewEnterHandlingGrid(gridlib.Grid):
     def __init__(self, parent, log):
-        wxGrid.__init__(self, parent, -1)
+        gridlib.Grid.__init__(self, parent, -1)
         self.log = log
 
         self.CreateGrid(20, 6)
@@ -15,11 +16,11 @@ class NewEnterHandlingGrid(wxGrid):
         self.SetColSize(0, 150)
         self.SetColSize(5, 150)
 
-        EVT_KEY_DOWN(self, self.OnKeyDown)
+        self.Bind(wx.EVT_KEY_DOWN, self.OnKeyDown)
 
 
     def OnKeyDown(self, evt):
-        if evt.KeyCode() != WXK_RETURN:
+        if evt.GetKeyCode() != wx.WXK_RETURN:
             evt.Skip()
             return
 
@@ -29,8 +30,10 @@ class NewEnterHandlingGrid(wxGrid):
 
         self.DisableCellEditControl()
         success = self.MoveCursorRight(evt.ShiftDown())
+
         if not success:
             newRow = self.GetGridCursorRow() + 1
+
             if newRow < self.GetTable().GetNumberRows():
                 self.SetGridCursor(newRow, 0)
                 self.MakeCellVisible(newRow, 0)
@@ -42,9 +45,9 @@ class NewEnterHandlingGrid(wxGrid):
 
 #---------------------------------------------------------------------------
 
-class TestFrame(wxFrame):
+class TestFrame(wx.Frame):
     def __init__(self, parent, log):
-        wxFrame.__init__(self, parent, -1, "Simple Grid Demo", size=(640,480))
+        wx.Frame.__init__(self, parent, -1, "Simple Grid Demo", size=(640,480))
         grid = NewEnterHandlingGrid(self, log)
 
 
@@ -53,7 +56,7 @@ class TestFrame(wxFrame):
 
 if __name__ == '__main__':
     import sys
-    app = wxPySimpleApp()
+    app = wx.PySimpleApp()
     frame = TestFrame(None, sys.stdout)
     frame.Show(True)
     app.MainLoop()