| 50 | |
| 51 | |
| 52 | class DroneStackSplit(wx.Dialog): |
| 53 | |
| 54 | def __init__(self, parent, value): |
| 55 | super().__init__(parent, title="Split Drone Stack", style=wx.DEFAULT_DIALOG_STYLE) |
| 56 | self.SetMinSize((346, 156)) |
| 57 | |
| 58 | bSizer1 = wx.BoxSizer(wx.VERTICAL) |
| 59 | |
| 60 | bSizer2 = wx.BoxSizer(wx.VERTICAL) |
| 61 | text = wx.StaticText(self, wx.ID_ANY, "New Amount:") |
| 62 | bSizer2.Add(text, 0) |
| 63 | |
| 64 | bSizer1.Add(bSizer2, 0, wx.ALL, 10) |
| 65 | |
| 66 | self.input = wx.TextCtrl(self, wx.ID_ANY, style=wx.TE_PROCESS_ENTER) |
| 67 | self.input.SetValue(str(value)) |
| 68 | |
| 69 | bSizer1.Add(self.input, 0, wx.LEFT | wx.RIGHT | wx.EXPAND, 15) |
| 70 | |
| 71 | bSizer3 = wx.BoxSizer(wx.VERTICAL) |
| 72 | bSizer3.Add(wx.StaticLine(self, wx.ID_ANY), 0, wx.BOTTOM | wx.EXPAND, 15) |
| 73 | |
| 74 | bSizer3.Add(self.CreateStdDialogButtonSizer(wx.OK | wx.CANCEL), 0, wx.EXPAND) |
| 75 | bSizer1.Add(bSizer3, 0, wx.ALL | wx.EXPAND, 10) |
| 76 | |
| 77 | self.input.Bind(wx.EVT_CHAR, self.onChar) |
| 78 | self.input.Bind(wx.EVT_TEXT_ENTER, self.processEnter) |
| 79 | self.SetSizer(bSizer1) |
| 80 | self.Fit() |
| 81 | self.CenterOnParent() |
| 82 | self.input.SetFocus() |
| 83 | self.input.SelectAll() |
| 84 | |
| 85 | def processEnter(self, evt): |
| 86 | self.EndModal(wx.ID_OK) |
| 87 | |
| 88 | # checks to make sure it's valid number |
| 89 | @staticmethod |
| 90 | def onChar(event): |
| 91 | key = event.GetKeyCode() |
| 92 | |
| 93 | acceptable_characters = "1234567890" |
| 94 | acceptable_keycode = [3, 22, 13, 8, 127] # modifiers like delete, copy, paste |
| 95 | if key in acceptable_keycode or key >= 255 or (key < 255 and chr(key) in acceptable_characters): |
| 96 | event.Skip() |
| 97 | return |
| 98 | else: |
| 99 | return False |